1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84
|
<?php
namespace MediaWiki\Tests\Maintenance;
use MediaWiki\Deferred\DeferredUpdates;
use MediaWiki\MainConfigNames;
use MediaWiki\Title\Title;
use UpdateArticleCount;
/**
* @covers \UpdateArticleCount
* @group Database
* @author Dreamy Jazz
*/
class UpdateArticleCountTest extends MaintenanceBaseTestCase {
protected function getMaintenanceClass() {
return UpdateArticleCount::class;
}
/**
* @param int $count
* @return Title[]
*/
private function getExistingTestPages( int $count ): array {
$returnArray = [];
for ( $i = 0; $i < $count; $i++ ) {
$returnArray[] = $this->getExistingTestPage()->getTitle();
}
return $returnArray;
}
private function getExistingTestPagesWithLinks( int $count ) {
$titles = $this->getExistingTestPages( $count );
// Replace the content of the existing pages with a wikilink.
foreach ( $titles as $title ) {
$this->editPage( $title, '[[Test]]' );
}
}
/** @dataProvider provideExecute */
public function testExecute(
$options, $existingPagesCount, $existingPagesWithLinksCount, $articleCountMethod,
$expectedArticleCountInDb, $expectedOutputString
) {
$this->overrideConfigValue( MainConfigNames::ArticleCountMethod, $articleCountMethod );
// Get some testing pages
$this->getExistingTestPages( $existingPagesCount );
$this->getExistingTestPagesWithLinks( $existingPagesWithLinksCount );
// Call ::doUpdates to ensure that the site_stats table is properly updated before we set the value to 0.
DeferredUpdates::doUpdates();
// Set the site_stats ss_good_articles to 0 to test that it changes after the maintenance script runs.
$this->getDb()->newUpdateQueryBuilder()
->update( 'site_stats' )
->set( [ 'ss_good_articles' => 0 ] )
->where( [ 'ss_row_id' => 1 ] )
->execute();
// Run the maintenance script
foreach ( $options as $name => $value ) {
$this->maintenance->setOption( $name, $value );
}
$this->maintenance->execute();
// Verify that the output is as expected and that the ss_good_articles column has the correct value.
$this->expectOutputString( $expectedOutputString );
$this->assertSelect(
'site_stats', 'ss_good_articles', [ 'ss_row_id' => 1 ],
[ [ $expectedArticleCountInDb ] ],
);
}
public static function provideExecute() {
return [
'Dry-run, use-master specified, 2 pages without links, 1 with links, wgArticleCountMethod as all' => [
[ 'use-master' => 1 ], 2, 1, 'all', 0,
"Counting articles...found 3.\nTo update the site statistics table, run the script " .
"with the --update option.\n"
],
'1 page without links, 4 pages with links, wgArticleCountMethod as link' => [
[ 'update' => 1 ], 1, 4, 'link', 4,
"Counting articles...found 4.\nUpdating site statistics table...done.\n"
],
];
}
}
|