tests/phpunit/includes/deferred/SiteStatsUpdateTest.php

   1 <?php
   2
   3 use MediaWiki\Deferred\DeferredUpdates;
   4 use MediaWiki\Deferred\SiteStatsUpdate;
   5 use MediaWiki\SiteStats\SiteStats;
   6 use MediaWiki\SiteStats\SiteStatsInit;
   7 use Wikimedia\TestingAccessWrapper;
   8
   9 /**
  10  * @group Database
  11  * @covers \MediaWiki\Deferred\SiteStatsUpdate
  12  * @covers \MediaWiki\SiteStats\SiteStats
  13  * @covers \MediaWiki\SiteStats\SiteStatsInit
  14  */
  15 class SiteStatsUpdateTest extends MediaWikiIntegrationTestCase {
  16         public function testFactoryAndMerge() {
  17                 $update1 = SiteStatsUpdate::factory( [ 'pages' => 1, 'users' => 2 ] );
  18                 $update2 = SiteStatsUpdate::factory( [ 'users' => 1, 'images' => 1 ] );
  19
  20                 $update1->merge( $update2 );
  21                 $wrapped = TestingAccessWrapper::newFromObject( $update1 );
  22
  23                 $this->assertSame( 1, $wrapped->pages );
  24                 $this->assertEquals( 3, $wrapped->users );
  25                 $this->assertSame( 1, $wrapped->images );
  26                 $this->assertSame( 0, $wrapped->edits );
  27                 $this->assertSame( 0, $wrapped->articles );
  28         }
  29
  30         public function testDoUpdate() {
  31                 $dbw = $this->getDb();
  32                 $statsInit = new SiteStatsInit( $dbw );
  33                 $statsInit->refresh();
  34
  35                 $ei = SiteStats::edits(); // trigger load
  36                 $pi = SiteStats::pages();
  37                 $ui = SiteStats::users();
  38                 $fi = SiteStats::images();
  39                 $ai = SiteStats::articles();
  40
  41                 $this->assertSame( 0, DeferredUpdates::pendingUpdatesCount() );
  42
  43                 $dbw->begin( __METHOD__ ); // block opportunistic updates
  44
  45                 DeferredUpdates::addUpdate(
  46                         SiteStatsUpdate::factory( [ 'pages' => 2, 'images' => 1, 'edits' => 2 ] )
  47                 );
  48                 $this->assertSame( 1, DeferredUpdates::pendingUpdatesCount() );
  49
  50                 // Still the same
  51                 SiteStats::unload();
  52                 $this->assertEquals( $pi, SiteStats::pages(), 'page count' );
  53                 $this->assertEquals( $ei, SiteStats::edits(), 'edit count' );
  54                 $this->assertEquals( $ui, SiteStats::users(), 'user count' );
  55                 $this->assertEquals( $fi, SiteStats::images(), 'file count' );
  56                 $this->assertEquals( $ai, SiteStats::articles(), 'article count' );
  57                 $this->assertSame( 1, DeferredUpdates::pendingUpdatesCount() );
  58
  59                 // This also notifies DeferredUpdates to do an opportunistic run
  60                 $dbw->commit( __METHOD__ );
  61                 $this->runDeferredUpdates();
  62                 $this->assertSame( 0, DeferredUpdates::pendingUpdatesCount() );
  63
  64                 SiteStats::unload();
  65                 $this->assertEquals( $pi + 2, SiteStats::pages(), 'page count' );
  66                 $this->assertEquals( $ei + 2, SiteStats::edits(), 'edit count' );
  67                 $this->assertEquals( $ui, SiteStats::users(), 'user count' );
  68                 $this->assertEquals( $fi + 1, SiteStats::images(), 'file count' );
  69                 $this->assertEquals( $ai, SiteStats::articles(), 'article count' );
  70
  71                 $statsInit = new SiteStatsInit();
  72                 $statsInit->refresh();
  73         }
  74 }