959bc315f2
The name change happened some time ago, and I think its about time to start using the name name! (Done with a find and replace) My personal motivation for doing this is that I have started trying out vscode as an IDE for mediawiki development, and right now it doesn't appear to handle php aliases very well or at all. Change-Id: I412235d91ae26e4c1c6a62e0dbb7e7cf3c5ed4a6
111 lines
3.5 KiB
PHP
111 lines
3.5 KiB
PHP
<?php
|
|
|
|
use MediaWiki\MediaWikiServices;
|
|
|
|
/**
|
|
* @covers ClearUserWatchlistJob
|
|
*
|
|
* @group JobQueue
|
|
* @group Database
|
|
*
|
|
* @license GPL-2.0-or-later
|
|
* @author Addshore
|
|
*/
|
|
class ClearUserWatchlistJobTest extends MediaWikiIntegrationTestCase {
|
|
|
|
protected function setUp() : void {
|
|
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( [
|
|
'userId' => $user->getId(), 'maxWatchlistId' => $maxId,
|
|
] )
|
|
);
|
|
|
|
$this->assertSame( 1, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
|
|
$this->assertEquals( 6, $watchedItemStore->countWatchedItems( $user ) );
|
|
$this->runJobs( 1 );
|
|
$this->assertSame( 1, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
|
|
$this->assertEquals( 4, $watchedItemStore->countWatchedItems( $user ) );
|
|
$this->runJobs( 1 );
|
|
$this->assertSame( 1, JobQueueGroup::singleton()->getQueueSizes()['clearUserWatchlist'] );
|
|
$this->assertEquals( 2, $watchedItemStore->countWatchedItems( $user ) );
|
|
$this->runJobs( 1 );
|
|
$this->assertSame( 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' ) ) );
|
|
}
|
|
|
|
public function testRunWithWatchlistExpiry() {
|
|
// Set up.
|
|
$this->setMwGlobals( 'wgWatchlistExpiry', true );
|
|
$user = $this->getUser();
|
|
$watchedItemStore = $this->getWatchedItemStore();
|
|
|
|
// Add two watched items, one with an expiry.
|
|
$watchedItemStore->addWatch( $user, new TitleValue( 0, __METHOD__ . 'no expiry' ) );
|
|
$watchedItemStore->addWatch( $user, new TitleValue( 0, __METHOD__ . 'has expiry' ), '1 week' );
|
|
|
|
// Get the IDs of these items.
|
|
$itemIds = $this->db->selectFieldValues(
|
|
[ 'watchlist' ],
|
|
'wl_id',
|
|
[ 'wl_user' => $user->getId() ],
|
|
__METHOD__
|
|
);
|
|
|
|
// Clear the watchlist by running the job.
|
|
$job = new ClearUserWatchlistJob( [
|
|
'userId' => $user->getId(),
|
|
'maxWatchlistId' => max( $itemIds ),
|
|
] );
|
|
JobQueueGroup::singleton()->push( $job );
|
|
$this->runJobs( 1 );
|
|
|
|
// Confirm that there are now no expiry records.
|
|
$watchedCount = $this->db->selectRowCount(
|
|
'watchlist_expiry',
|
|
'*',
|
|
[ 'we_item' => $itemIds ],
|
|
__METHOD__
|
|
);
|
|
$this->assertSame( 0, $watchedCount );
|
|
}
|
|
}
|