79 lines
2.5 KiB
PHP
79 lines
2.5 KiB
PHP
<?php
|
|
use MediaWiki\MediaWikiServices;
|
|
|
|
/**
|
|
* @covers ClearUserWatchlistJob
|
|
*
|
|
* @group JobQueue
|
|
* @group Database
|
|
*
|
|
* @license GPL-2.0-or-later
|
|
* @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' ) ) );
|
|
}
|
|
|
|
}
|