tests/phpunit/includes/user/LocalIdLookupTest.php

   1 <?php
   2
   3 use MediaWiki\Block\DatabaseBlock;
   4
   5 /**
   6  * @covers LocalIdLookup
   7  * @group Database
   8  */
   9 class LocalIdLookupTest extends MediaWikiTestCase {
  10         private $localUsers = [];
  11
  12         protected function setUp() {
  13                 parent::setUp();
  14
  15                 $this->setGroupPermissions( 'local-id-lookup-test', 'hideuser', true );
  16                 $this->overrideMwServices();
  17         }
  18
  19         public function addDBData() {
  20                 for ( $i = 1; $i <= 4; $i++ ) {
  21                         $this->localUsers[] = $this->getMutableTestUser()->getUser();
  22                 }
  23
  24                 $sysop = static::getTestSysop()->getUser();
  25
  26                 $block = new DatabaseBlock( [
  27                         'address' => $this->localUsers[2]->getName(),
  28                         'by' => $sysop->getId(),
  29                         'reason' => __METHOD__,
  30                         'expiry' => '1 day',
  31                         'hideName' => false,
  32                 ] );
  33                 $block->insert();
  34
  35                 $block = new DatabaseBlock( [
  36                         'address' => $this->localUsers[3]->getName(),
  37                         'by' => $sysop->getId(),
  38                         'reason' => __METHOD__,
  39                         'expiry' => '1 day',
  40                         'hideName' => true,
  41                 ] );
  42                 $block->insert();
  43         }
  44
  45         public function getLookupUser() {
  46                 return static::getTestUser( [ 'local-id-lookup-test' ] )->getUser();
  47         }
  48
  49         public function testLookupCentralIds() {
  50                 $lookup = new LocalIdLookup();
  51
  52                 $user1 = $this->getLookupUser();
  53                 $user2 = User::newFromName( 'UTLocalIdLookup2' );
  54
  55                 $this->assertTrue( $user1->isAllowed( 'hideuser' ), 'sanity check' );
  56                 $this->assertFalse( $user2->isAllowed( 'hideuser' ), 'sanity check' );
  57
  58                 $this->assertSame( [], $lookup->lookupCentralIds( [] ) );
  59
  60                 $expect = [];
  61                 foreach ( $this->localUsers as $localUser ) {
  62                         $expect[$localUser->getId()] = $localUser->getName();
  63                 }
  64                 $expect[12345] = 'X';
  65                 ksort( $expect );
  66
  67                 $expect2 = $expect;
  68                 $expect2[$this->localUsers[3]->getId()] = '';
  69
  70                 $arg = array_fill_keys( array_keys( $expect ), 'X' );
  71
  72                 $this->assertSame( $expect2, $lookup->lookupCentralIds( $arg ) );
  73                 $this->assertSame( $expect, $lookup->lookupCentralIds( $arg, CentralIdLookup::AUDIENCE_RAW ) );
  74                 $this->assertSame( $expect, $lookup->lookupCentralIds( $arg, $user1 ) );
  75                 $this->assertSame( $expect2, $lookup->lookupCentralIds( $arg, $user2 ) );
  76         }
  77
  78         public function testLookupUserNames() {
  79                 $lookup = new LocalIdLookup();
  80                 $user1 = $this->getLookupUser();
  81                 $user2 = User::newFromName( 'UTLocalIdLookup2' );
  82
  83                 $this->assertTrue( $user1->isAllowed( 'hideuser' ), 'sanity check' );
  84                 $this->assertFalse( $user2->isAllowed( 'hideuser' ), 'sanity check' );
  85
  86                 $this->assertSame( [], $lookup->lookupUserNames( [] ) );
  87
  88                 $expect = [];
  89                 foreach ( $this->localUsers as $localUser ) {
  90                         $expect[$localUser->getName()] = $localUser->getId();
  91                 }
  92                 $expect['UTDoesNotExist'] = 'X';
  93                 ksort( $expect );
  94
  95                 $expect2 = $expect;
  96                 $expect2[$this->localUsers[3]->getName()] = 'X';
  97
  98                 $arg = array_fill_keys( array_keys( $expect ), 'X' );
  99
 100                 $this->assertSame( $expect2, $lookup->lookupUserNames( $arg ) );
 101                 $this->assertSame( $expect, $lookup->lookupUserNames( $arg, CentralIdLookup::AUDIENCE_RAW ) );
 102                 $this->assertSame( $expect, $lookup->lookupUserNames( $arg, $user1 ) );
 103                 $this->assertSame( $expect2, $lookup->lookupUserNames( $arg, $user2 ) );
 104         }
 105
 106         public function testIsAttached() {
 107                 $lookup = new LocalIdLookup();
 108                 $user1 = $this->getLookupUser();
 109                 $user2 = User::newFromName( 'DoesNotExist' );
 110
 111                 $this->assertTrue( $lookup->isAttached( $user1 ) );
 112                 $this->assertFalse( $lookup->isAttached( $user2 ) );
 113
 114                 $wiki = wfWikiID();
 115                 $this->assertTrue( $lookup->isAttached( $user1, $wiki ) );
 116                 $this->assertFalse( $lookup->isAttached( $user2, $wiki ) );
 117
 118                 $wiki = 'not-' . wfWikiID();
 119                 $this->assertFalse( $lookup->isAttached( $user1, $wiki ) );
 120                 $this->assertFalse( $lookup->isAttached( $user2, $wiki ) );
 121         }
 122
 123         /**
 124          * @dataProvider provideIsAttachedShared
 125          * @param bool $sharedDB $wgSharedDB is set
 126          * @param bool $sharedTable $wgSharedTables contains 'user'
 127          * @param bool $localDBSet $wgLocalDatabases contains the shared DB
 128          */
 129         public function testIsAttachedShared( $sharedDB, $sharedTable, $localDBSet ) {
 130                 $this->setMwGlobals( [
 131                         'wgSharedDB' => $sharedDB ? "dummy" : null,
 132                         'wgSharedTables' => $sharedTable ? [ 'user' ] : [],
 133                         'wgLocalDatabases' => $localDBSet ? [ 'shared' ] : [],
 134                 ] );
 135
 136                 $lookup = new LocalIdLookup();
 137                 $this->assertSame(
 138                         $sharedDB && $sharedTable && $localDBSet,
 139                         $lookup->isAttached( $this->getLookupUser(), 'shared' )
 140                 );
 141         }
 142
 143         public static function provideIsAttachedShared() {
 144                 $ret = [];
 145                 for ( $i = 0; $i < 7; $i++ ) {
 146                         $ret[] = [
 147                                 (bool)( $i & 1 ),
 148                                 (bool)( $i & 2 ),
 149                                 (bool)( $i & 4 ),
 150                         ];
 151                 }
 152                 return $ret;
 153         }
 154
 155 }