2 files changed, 166 insertions, 0 deletions
diff --git a/www/wiki/tests/phpunit/includes/jobqueue/jobs/CategoryMembershipChangeJobTest.php b/www/wiki/tests/phpunit/includes/jobqueue/jobs/CategoryMembershipChangeJobTest.php
new file mode 100644
index 00000000..5960a16b
--- /dev/null
+++ b/www/wiki/tests/phpunit/includes/jobqueue/jobs/CategoryMembershipChangeJobTest.php
@@ -0,0 +1,87 @@
+<?php
+
+/**
+ * @covers CategoryMembershipChangeJob
+ *
+ * @group JobQueue
+ * @group Database
+ *
+ * @license GNU GPL v2+
+ * @author Addshore
+ */
+class CategoryMembershipChangeJobTest extends MediaWikiTestCase {
+
+	const TITLE_STRING = 'UTCatChangeJobPage';
+
+	/**
+	 * @var Title
+	 */
+	private $title;
+
+	public function setUp() {
+		parent::setUp();
+		$this->setMwGlobals( 'wgRCWatchCategoryMembership', true );
+		$this->setContentLang( 'qqx' );
+	}
+
+	public function addDBDataOnce() {
+		parent::addDBDataOnce();
+		$insertResult = $this->insertPage( self::TITLE_STRING, 'UT Content' );
+		$this->title = $insertResult['title'];
+	}
+
+	private function runJobs() {
+		JobQueueGroup::destroySingletons();
+		$jobs = new RunJobs;
+		$jobs->loadParamsAndArgs( null, [ 'quiet' => true ], null );
+		$jobs->execute();
+	}
+
+	/**
+	 * @param string $text new page text
+	 *
+	 * @return int|null
+	 */
+	private function editPageText( $text ) {
+		$page = WikiPage::factory( $this->title );
+		$editResult = $page->doEditContent(
+			ContentHandler::makeContent( $text, $this->title ),
+			__METHOD__
+		);
+		/** @var Revision $revision */
+		$revision = $editResult->value['revision'];
+		$this->runJobs();
+
+		return $revision->getId();
+	}
+
+	/**
+	 * @param int $revId
+	 *
+	 * @return RecentChange|null
+	 */
+	private function getCategorizeRecentChangeForRevId( $revId ) {
+		return RecentChange::newFromConds(
+			[
+				'rc_type' => RC_CATEGORIZE,
+				'rc_this_oldid' => $revId,
+			],
+			__METHOD__
+		);
+	}
+
+	public function testRun_normalCategoryAddedAndRemoved() {
+		$addedRevId = $this->editPageText( '[[Category:Normal]]' );
+		$removedRevId = $this->editPageText( 'Blank' );
+
+		$this->assertEquals(
+			'(recentchanges-page-added-to-category: ' . self::TITLE_STRING . ')',
+			$this->getCategorizeRecentChangeForRevId( $addedRevId )->getAttribute( 'rc_comment' )
+		);
+		$this->assertEquals(
+			'(recentchanges-page-removed-from-category: ' . self::TITLE_STRING . ')',
+			$this->getCategorizeRecentChangeForRevId( $removedRevId )->getAttribute( 'rc_comment' )
+		);
+	}
+
+}
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' ) ) );
+	}
+
+}