Page MenuHomeWickedGov Phorge
Files F5988715

LocalIdLookupTest.php
No OneTemporary
Actions

Size
6 KB
Referenced Files
None
Subscribers
None

LocalIdLookupTest.php
View Options

<?php
use MediaWiki\Config\HashConfig;
use MediaWiki\MainConfigNames;
use MediaWiki\Tests\Unit\Permissions\MockAuthorityTrait;
use MediaWiki\User\CentralId\CentralIdLookup;
use MediaWiki\User\CentralId\LocalIdLookup;
use MediaWiki\User\UserIdentity;
use MediaWiki\User\UserIdentityValue;
use MediaWiki\WikiMap\WikiMap;
use Wikimedia\ArrayUtils\ArrayUtils;
use Wikimedia\Rdbms\IDBAccessObject;
/**
* @covers \MediaWiki\User\CentralId\LocalIdLookup
* @group Database
*/
class LocalIdLookupTest extends MediaWikiIntegrationTestCase {
use MockAuthorityTrait;
/** @var UserIdentity[] */
private $localUsers = [];
public function addDBData() {
for ( $i = 1; $i <= 4; $i++ ) {
$this->localUsers[] = $this->getMutableTestUser()->getUserIdentity();
}
$sysop = static::getTestSysop()->getUserIdentity();
$blockStore = $this->getServiceContainer()->getDatabaseBlockStore();
$blockStore->insertBlockWithParams( [
'targetUser' => $this->localUsers[2],
'by' => $sysop,
'reason' => __METHOD__,
'expiry' => '1 day',
'hideName' => false,
] );
$blockStore->insertBlockWithParams( [
'targetUser' => $this->localUsers[3],
'by' => $sysop,
'reason' => __METHOD__,
'expiry' => '1 day',
'hideName' => true,
] );
}
private function newLookup( array $configOverride = [] ) {
$lookup = new LocalIdLookup(
new HashConfig( $configOverride + [
MainConfigNames::SharedDB => null,
MainConfigNames::SharedTables => [],
MainConfigNames::LocalDatabases => [],
] ),
$this->getServiceContainer()->getConnectionProvider(),
$this->getServiceContainer()->getHideUserUtils()
);
$lookup->init(
'test',
$this->getServiceContainer()->getUserIdentityLookup(),
$this->getServiceContainer()->getUserFactory()
);
return $lookup;
}
public function testLookupCentralIds() {
$lookup = $this->newLookup();
$permitted = $this->mockAnonAuthorityWithPermissions( [ 'hideuser' ] );
$nonPermitted = $this->mockAnonAuthorityWithoutPermissions( [ 'hideuser' ] );
$this->assertSame( [], $lookup->lookupCentralIds( [] ) );
$expect = [];
foreach ( $this->localUsers as $localUser ) {
$expect[$localUser->getId()] = $localUser->getName();
}
$expect[12345] = 'X';
ksort( $expect );
$expect2 = $expect;
$expect2[$this->localUsers[3]->getId()] = '';
$arg = array_fill_keys( array_keys( $expect ), 'X' );
$this->assertSame( $expect2, $lookup->lookupCentralIds( $arg ) );
$this->assertSame( $expect, $lookup->lookupCentralIds( $arg, CentralIdLookup::AUDIENCE_RAW ) );
$this->assertSame( $expect, $lookup->lookupCentralIds( $arg, $permitted ) );
$this->assertSame( $expect2, $lookup->lookupCentralIds( $arg, $nonPermitted ) );
}
public function testLookupUserNames() {
$lookup = $this->newLookup();
$permitted = $this->mockAnonAuthorityWithPermissions( [ 'hideuser' ] );
$nonPermitted = $this->mockAnonAuthorityWithoutPermissions( [ 'hideuser' ] );
$this->assertSame( [], $lookup->lookupUserNames( [] ) );
$expect = [];
foreach ( $this->localUsers as $localUser ) {
$expect[$localUser->getName()] = $localUser->getId();
}
$expect['UTDoesNotExist'] = 'X';
ksort( $expect );
$expect2 = $expect;
$expect2[$this->localUsers[3]->getName()] = 'X';
$arg = array_fill_keys( array_keys( $expect ), 'X' );
$this->assertSame( $expect2, $lookup->lookupUserNames( $arg ) );
$this->assertSame( $expect, $lookup->lookupUserNames( $arg, CentralIdLookup::AUDIENCE_RAW ) );
$this->assertSame( $expect, $lookup->lookupUserNames( $arg, $permitted ) );
$this->assertSame( $expect2, $lookup->lookupUserNames( $arg, $nonPermitted ) );
}
/** @dataProvider provideLookupAttachedUserNames */
public function testLookupAttachedUserNames( $isShared ) {
$lookup = $this->newLookup( [
MainConfigNames::SharedDB => $isShared ? "dummy" : null,
MainConfigNames::SharedTables => $isShared ? [ 'user' ] : [],
MainConfigNames::LocalDatabases => $isShared ? [ 'shared' ] : [],
] );
$nameToId = [];
foreach ( $this->localUsers as $user ) {
$nameToId[$user->getName()] = 0;
if ( $isShared ) {
$expected[$user->getName()] = $user->getId();
} else {
$expected[$user->getName()] = 0;
}
}
$result = $lookup->lookupAttachedUserNames(
$nameToId,
CentralIdLookup::AUDIENCE_RAW,
IDBAccessObject::READ_NORMAL,
'shared'
);
$this->assertSame( $expected, $result );
}
public static function provideLookupAttachedUserNames() {
return [ [ false, true ] ];
}
public function testIsAttached() {
$lookup = $this->newLookup();
$user1 = UserIdentityValue::newRegistered( 42, 'Test' );
$user2 = UserIdentityValue::newAnonymous( 'DoesNotExist' );
$this->assertTrue( $lookup->isAttached( $user1 ) );
$this->assertFalse( $lookup->isAttached( $user2 ) );
$wiki = UserIdentityValue::LOCAL;
$this->assertTrue( $lookup->isAttached( $user1, $wiki ) );
$this->assertFalse( $lookup->isAttached( $user2, $wiki ) );
$wiki = 'some_other_wiki';
$this->assertFalse( $lookup->isAttached( $user1, $wiki ) );
$this->assertFalse( $lookup->isAttached( $user2, $wiki ) );
}
/**
* @dataProvider provideIsAttachedShared
* @param bool $sharedDB $wgSharedDB is set
* @param bool $sharedTable $wgSharedTables contains 'user'
* @param bool $localDBSet $wgLocalDatabases contains the shared DB
*/
public function testIsAttachedShared( $sharedDB, $sharedTable, $localDBSet ) {
$lookup = $this->newLookup( [
MainConfigNames::SharedDB => $sharedDB ? "dummy" : null,
MainConfigNames::SharedTables => $sharedTable ? [ 'user' ] : [],
MainConfigNames::LocalDatabases => $localDBSet ? [ 'shared' ] : [],
] );
$this->assertSame(
$sharedDB && $sharedTable && $localDBSet,
$lookup->isAttached( UserIdentityValue::newRegistered( 42, 'Test' ), 'shared' )
);
}
public static function provideIsAttachedShared() {
$bool = [ false, true ];
return ArrayUtils::cartesianProduct( $bool, $bool, $bool );
}
public function testGetScope() {
$lookup = $this->newLookup();
$wikiId = WikiMap::getCurrentWikiId();
$this->assertSame( "test:$wikiId", $lookup->getScope() );
$lookup = $this->newLookup( [
MainConfigNames::SharedDB => 'shared',
MainConfigNames::SharedTables => [ 'user' ],
MainConfigNames::LocalDatabases => [ 'shared' ],
] );
$this->assertSame( 'test:shared', $lookup->getScope() );
}
}

File Metadata

Mime Type
text/x-php
Expires
Sun, Oct 4, 00:53 (3 w, 2 d ago)
Storage Engine
local-disk
Storage Format
Raw Data
Storage Handle
9e/d9/c14242e76e0a2515d794433ee15a
Default Alt Text
LocalIdLookupTest.php (6 KB)

Event Timeline