tests/phpunit/maintenance/CheckSignaturesTest.php

   1 <?php
   2
   3 namespace MediaWiki\Tests\Maintenance;
   4
   5 use CheckSignatures;
   6 use MediaWiki\User\UserIdentity;
   7
   8 /**
   9  * @covers \CheckSignatures
  10  * @group Database
  11  * @author Dreamy Jazz
  12  */
  13 class CheckSignaturesTest extends MaintenanceBaseTestCase {
  14
  15         protected function getMaintenanceClass() {
  16                 return CheckSignatures::class;
  17         }
  18
  19         /**
  20          * @param int $count
  21          * @return UserIdentity[]
  22          */
  23         private function generateMutableTestUsers( int $count ): array {
  24                 $returnArray = [];
  25                 for ( $i = 0; $i < $count; $i++ ) {
  26                         $testUser = $this->getMutableTestUser()->getUserIdentity();
  27                         $returnArray[] = $testUser;
  28                 }
  29                 return $returnArray;
  30         }
  31
  32         /**
  33          * @param int $count
  34          * @return UserIdentity[]
  35          */
  36         private function generateUsersWithInvalidSignatures( int $count ): array {
  37                 $userOptionsManager = $this->getServiceContainer()->getUserOptionsManager();
  38                 $returnArray = $this->generateMutableTestUsers( $count );
  39                 foreach ( $returnArray as $user ) {
  40                         // Set the signature to invalid HTML
  41                         $userOptionsManager->setOption( $user, 'nickname', '<b>{{test' );
  42                         $userOptionsManager->setOption( $user, 'fancysig', true );
  43                         $userOptionsManager->saveOptions( $user );
  44                 }
  45                 return $returnArray;
  46         }
  47
  48         private function generateUsersWithValidSignatures( int $count ) {
  49                 $userOptionsManager = $this->getServiceContainer()->getUserOptionsManager();
  50                 foreach ( $this->generateMutableTestUsers( $count ) as $user ) {
  51                         // Set the signature to invalid HTML
  52                         $userOptionsManager->setOption( $user, 'nickname', '[[User:' . $user->getName() . "]]" );
  53                         $userOptionsManager->setOption( $user, 'fancysig', true );
  54                         $userOptionsManager->saveOptions( $user );
  55                 }
  56         }
  57
  58         /** @dataProvider provideExecute */
  59         public function testExecute( $validCount, $invalidCount ) {
  60                 $usersWithInvalidSignatures = $this->generateUsersWithInvalidSignatures( $invalidCount );
  61                 $this->generateUsersWithValidSignatures( $validCount );
  62                 $this->maintenance->execute();
  63                 $expectedOutputRegex = '/';
  64                 foreach ( $usersWithInvalidSignatures as $user ) {
  65                         $expectedOutputRegex .= $user->getName() . "\n";
  66                 }
  67                 $expectedOutputRegex .= ".*$invalidCount invalid signatures/";
  68                 $this->expectOutputRegex( $expectedOutputRegex );
  69         }
  70
  71         public static function provideExecute() {
  72                 return [
  73                         'All signatures valid' => [ 4, 0 ],
  74                         'Some signatures valid' => [ 2, 3 ],
  75                 ];
  76         }
  77 }