Code Coverage for /workspace/src/extensions/GrowthExperiments/includes/NewcomerTasks/LinkRecommendationFilter.php

1	<?php
2
3	namespace GrowthExperiments\NewcomerTasks;
4
5	use GrowthExperiments\NewcomerTasks\AddLink\LinkRecommendationStore;
6	use GrowthExperiments\NewcomerTasks\Task\TaskSet;
7	use GrowthExperiments\NewcomerTasks\TaskType\LinkRecommendationTaskType;
8
9	/**
10	* Filter out link recommendation tasks that don't have a DB entry for recommendation data.
11	*/
12	class LinkRecommendationFilter extends AbstractTaskSetFilter implements TaskSetFilter {
13
14	/** @var LinkRecommendationStore */
15	private $linkRecommendationStore;
16
17	/**
18	* @param LinkRecommendationStore $linkRecommendationStore
19	*/
20	public function __construct( LinkRecommendationStore $linkRecommendationStore ) {
21	$this->linkRecommendationStore = $linkRecommendationStore;
22	}
23
24	/** @inheritDoc */
25	public function filter( TaskSet $taskSet, int $maxLength = PHP_INT_MAX ): TaskSet {
26	$invalidTasks = [];
27	$validTasks = [];
28	foreach ( $taskSet as $task ) {
29	if ( count( $validTasks ) >= $maxLength ) {
30	break;
31	}
32	if ( $task->getTaskType() instanceof LinkRecommendationTaskType &&
33	!$this->linkRecommendationStore->getByLinkTarget( $task->getTitle() ) ) {
34	$invalidTasks[] = $task;
35	} else {
36	$validTasks[] = $task;
37	}
38	}
39	return $this->copyValidAndInvalidTasksToNewTaskSet( $taskSet, $validTasks, $invalidTasks );
40	}
41	}