ClearUserWatchlistJobTest.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
 
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' ) ) );
    }
 
}