111 lines
3.8 KiB
PHP
111 lines
3.8 KiB
PHP
<?php
|
|
|
|
namespace MediaWiki\Tests\Maintenance;
|
|
|
|
use DeleteOldRevisions;
|
|
use WikiPage;
|
|
|
|
/**
|
|
* @covers \DeleteOldRevisions
|
|
* @group Database
|
|
* @author Dreamy Jazz
|
|
*/
|
|
class DeleteOldRevisionsTest extends MaintenanceBaseTestCase {
|
|
public function getMaintenanceClass() {
|
|
return DeleteOldRevisions::class;
|
|
}
|
|
|
|
private function expectTheExpectedOutputRegex(
|
|
bool $shouldDelete, int $specifiedPageId, int $expectedOldRevisionsCount
|
|
) {
|
|
$expectedOutputRegex = '/Delete old revisions';
|
|
// Only expect the page IDs to be outputted if we specified some.
|
|
if ( $specifiedPageId ) {
|
|
$expectedOutputRegex .= '[\s\S]*Limiting to page IDs.*' . $specifiedPageId;
|
|
} else {
|
|
$expectedOutputRegex .= '[\s\S]*(?!Limiting to page IDs)';
|
|
}
|
|
$expectedOutputRegex .= '[\s\S]*Searching for active revisions.*done[\s\S]*' .
|
|
'Searching for inactive revisions.*done[\s\S]*' .
|
|
"$expectedOldRevisionsCount old revisions found";
|
|
if ( $shouldDelete ) {
|
|
$expectedOutputRegex .= '[\s\S]*Deleting/';
|
|
} else {
|
|
$expectedOutputRegex .= '[\s\S]*(?!Deleting)/';
|
|
}
|
|
$this->expectOutputRegex( $expectedOutputRegex );
|
|
}
|
|
|
|
/** @dataProvider provideExecuteForNoPages */
|
|
public function testExecuteForNoPages( $deleteOptionProvided ) {
|
|
if ( $deleteOptionProvided ) {
|
|
$this->maintenance->setOption( 'delete', 1 );
|
|
}
|
|
$this->maintenance->execute();
|
|
$this->expectTheExpectedOutputRegex( false, 0, 0 );
|
|
}
|
|
|
|
public static function provideExecuteForNoPages() {
|
|
return [
|
|
'Delete option provided' => [ true ],
|
|
'Delete option not provided' => [ false ],
|
|
];
|
|
}
|
|
|
|
private function verifyRevisionCountForPage( WikiPage $page, int $expectedCount ) {
|
|
$this->newSelectQueryBuilder()
|
|
->select( 'COUNT(*)' )
|
|
->from( 'revision' )
|
|
->where( [ 'rev_page' => $page->getId() ] )
|
|
->assertFieldValue( $expectedCount );
|
|
}
|
|
|
|
/** @dataProvider provideExecute */
|
|
public function testExecute( $shouldDelete, $shouldSpecifyPageId, $expectedOldRevisionsCount ) {
|
|
$firstExistingTestPage = $this->getExistingTestPage();
|
|
$secondExistingTestPage = $this->getExistingTestPage();
|
|
$thirdExistingTestPage = $this->getExistingTestPage();
|
|
// Make an edit to both the first and second existing test pages, so that they have old revisions
|
|
$this->editPage( $firstExistingTestPage, 'abcdef' );
|
|
$this->editPage( $secondExistingTestPage, 'abcef' );
|
|
// Check that the revision count for each page is correct for the test.
|
|
$this->verifyRevisionCountForPage( $thirdExistingTestPage, 1 );
|
|
$this->verifyRevisionCountForPage( $secondExistingTestPage, 2 );
|
|
$this->verifyRevisionCountForPage( $firstExistingTestPage, 2 );
|
|
// Set the options for the maintenance script
|
|
if ( $shouldDelete ) {
|
|
$this->maintenance->setOption( 'delete', 1 );
|
|
}
|
|
if ( $shouldSpecifyPageId ) {
|
|
$this->maintenance->setArg( 'page_id', $firstExistingTestPage->getId() );
|
|
}
|
|
// Run the maintenance script and then verify the number of revisions for each test page is as expected
|
|
$this->maintenance->execute();
|
|
$this->verifyRevisionCountForPage( $thirdExistingTestPage, 1 );
|
|
$this->verifyRevisionCountForPage(
|
|
$secondExistingTestPage,
|
|
// Should not have been touched, unless --delete was specified and --page_id was not specified
|
|
$shouldDelete && !$shouldSpecifyPageId ? 1 : 2
|
|
);
|
|
$this->verifyRevisionCountForPage(
|
|
$firstExistingTestPage,
|
|
// Should not be touched, unless --delete is specified.
|
|
$shouldDelete ? 1 : 2
|
|
);
|
|
$this->expectTheExpectedOutputRegex(
|
|
$shouldDelete,
|
|
$shouldSpecifyPageId ? $firstExistingTestPage->getId() : 0,
|
|
$expectedOldRevisionsCount
|
|
);
|
|
}
|
|
|
|
public static function provideExecute() {
|
|
return [
|
|
'No options provided' => [ false, false, 2 ],
|
|
'Not deleting, but specified page ID' => [ false, true, 1 ],
|
|
'Deleting without specifying page IDs' => [ true, false, 2 ],
|
|
'Deleting while specifying page ID' => [ true, true, 1 ],
|
|
];
|
|
}
|
|
}
|