tests/phpunit/maintenance/PurgeExpiredUserrightsTest.php

   1 <?php
   2
   3 use MediaWiki\Tests\Maintenance\MaintenanceBaseTestCase;
   4 use MediaWiki\User\UserGroupManager;
   5
   6 /**
   7  * @covers \PurgeExpiredUserrights
   8  * @author Dreamy Jazz
   9  */
  10 class PurgeExpiredUserrightsTest extends MaintenanceBaseTestCase {
  11         public function getMaintenanceClass() {
  12                 return PurgeExpiredUserrights::class;
  13         }
  14
  15         /** @dataProvider provideExecute */
  16         public function testExecute( $mockPurgeExpiredReturnValue, $expectedOutputRegex ) {
  17                 // Mock the UserGroupManager to expect that ::purgeExpired is called once
  18                 // and also then return a fake value.
  19                 $mockUserGroupManager = $this->createMock( UserGroupManager::class );
  20                 $mockUserGroupManager->expects( $this->once() )
  21                         ->method( 'purgeExpired' )
  22                         ->willReturn( $mockPurgeExpiredReturnValue );
  23                 $this->setService( 'UserGroupManager', $mockUserGroupManager );
  24                 $this->maintenance->execute();
  25                 $this->expectOutputRegex( $expectedOutputRegex );
  26         }
  27
  28         public static function provideExecute() {
  29                 return [
  30                         '::purgeExpired returns false' => [ false, '/Purging expired user rights[\s\S]*Purging failed/' ],
  31                         '::purgeExpired returns 0' => [ 0, '/Purging expired user rights[\s\S]*0 rows purged/' ],
  32                         '::purgeExpired returns 3' => [ 3, '/Purging expired user rights[\s\S]*3 rows purged/' ],
  33                 ];
  34         }
  35 }