1 files changed, 79 insertions, 0 deletions
diff --git a/www/wiki/tests/phpunit/includes/jobqueue/jobs/ClearUserWatchlistJobTest.php b/www/wiki/tests/phpunit/includes/jobqueue/jobs/ClearUserWatchlistJobTest.php
new file mode 100644
index 00000000..6ae7d605
--- /dev/null
+++ b/www/wiki/tests/phpunit/includes/jobqueue/jobs/ClearUserWatchlistJobTest.php
@@ -0,0 +1,79 @@
+<?php
+use MediaWiki\MediaWikiServices;
+
+/**
+ * @covers ClearUserWatchlistJob
+ *
+ * @group JobQueue
+ * @group Database
+ *
+ * @license GNU GPL v2+
+ * @author Addshore
+ */
+class ClearUserWatchlistJobTest extends MediaWikiTestCase {
+
+	public function setUp() {
+		parent::setUp();
+		self::$users['ClearUserWatchlistJobTestUser']
+			= new TestUser( 'ClearUserWatchlistJobTestUser' );
+		$this->runJobs();
+		JobQueueGroup::destroySingletons();
+	}
+
+	private function getUser() {
+		return self::$users['ClearUserWatchlistJobTestUser']->getUser();
+	}
+
+	private function runJobs( $jobLimit = 9999 ) {
+		$runJobs = new RunJobs;
+		$runJobs->loadParamsAndArgs( null, [ 'quiet' => true, 'maxjobs' => $jobLimit ] );
+		$runJobs->execute();
+	}
+
+	private function getWatchedItemStore() {
+		return MediaWikiServices::getInstance()->getWatchedItemStore();
+	}
+
+	public function testRun() {
+		$user = $this->getUser();
+		$watchedItemStore = $this->getWatchedItemStore();
+
+		$watchedItemStore->addWatch( $user, new TitleValue( 0, 'A' ) );
+		$watchedItemStore->addWatch( $user, new TitleValue( 1, 'A' ) );
+		$watchedItemStore->addWatch( $user, new TitleValue( 0, 'B' ) );
+		$watchedItemStore->addWatch( $user, new TitleValue( 1, 'B' ) );
+
+		$maxId = $watchedItemStore->getMaxId();
+
+		$watchedItemStore->addWatch( $user, new TitleValue( 0, 'C' ) );
+		$watchedItemStore->addWatch( $user, new TitleValue( 1, 'C' ) );
+
+		$this->setMwGlobals( 'wgUpdateRowsPerQuery', 2 );
+
+		JobQueueGroup::singleton()->push(
+			new ClearUserWatchlistJob(
+				null,
+				[
+					'userId' => $user->getId(),
+					'maxWatchlistId' => $maxId,
+				]
+			)
+		);
+
+		$this->assertEquals( 1, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
+		$this->assertEquals( 6, $watchedItemStore->countWatchedItems( $user ) );
+		$this->runJobs( 1 );
+		$this->assertEquals( 1, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
+		$this->assertEquals( 4, $watchedItemStore->countWatchedItems( $user ) );
+		$this->runJobs( 1 );
+		$this->assertEquals( 1, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
+		$this->assertEquals( 2, $watchedItemStore->countWatchedItems( $user ) );
+		$this->runJobs( 1 );
+		$this->assertEquals( 0, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
+		$this->assertEquals( 2, $watchedItemStore->countWatchedItems( $user ) );
+
+		$this->assertTrue( $watchedItemStore->isWatched( $user, new TitleValue( 0, 'C' ) ) );
+		$this->assertTrue( $watchedItemStore->isWatched( $user, new TitleValue( 1, 'C' ) ) );
+	}
+
+}