d032bb52cd
Anything that wants to be "central" right now has to depend on CentralAuth, and then either can't work without CentralAuth or has to branch all over the place based on whether CentralAuth is present. Most of the time all it really needs is a mapping from local users to central user IDs and back or the ability to query whether the local user is attached on some other wiki, so let's make an interface for that in core. See I52aa0460 for an example implementation (CentralAuth), and Ibd192e29 for an example use (OAuth). Bug: T111302 Change-Id: I49568358ec35fdfd0b9e53e441adabded5c7b80f
149 lines
4.4 KiB
PHP
149 lines
4.4 KiB
PHP
<?php
|
|
|
|
/**
|
|
* @covers LocalIdLookup
|
|
* @group Database
|
|
*/
|
|
class LocalIdLookupTest extends MediaWikiTestCase {
|
|
private $localUsers = array();
|
|
|
|
protected function setUp() {
|
|
global $wgGroupPermissions;
|
|
|
|
parent::setUp();
|
|
|
|
$this->stashMwGlobals( array( 'wgGroupPermissions' ) );
|
|
$wgGroupPermissions['local-id-lookup-test']['hideuser'] = true;
|
|
}
|
|
|
|
public function addDBData() {
|
|
for ( $i = 1; $i <= 4; $i++ ) {
|
|
$user = User::newFromName( "UTLocalIdLookup$i" );
|
|
if ( $user->getId() == 0 ) {
|
|
$user->addToDatabase();
|
|
}
|
|
$this->localUsers["UTLocalIdLookup$i"] = $user->getId();
|
|
}
|
|
|
|
User::newFromName( 'UTLocalIdLookup1' )->addGroup( 'local-id-lookup-test' );
|
|
|
|
$block = new Block( array(
|
|
'address' => 'UTLocalIdLookup3',
|
|
'by' => User::idFromName( 'UTSysop' ),
|
|
'reason' => __METHOD__,
|
|
'expiry' => '1 day',
|
|
'hideName' => false,
|
|
) );
|
|
$block->insert();
|
|
|
|
$block = new Block( array(
|
|
'address' => 'UTLocalIdLookup4',
|
|
'by' => User::idFromName( 'UTSysop' ),
|
|
'reason' => __METHOD__,
|
|
'expiry' => '1 day',
|
|
'hideName' => true,
|
|
) );
|
|
$block->insert();
|
|
}
|
|
|
|
public function testLookupCentralIds() {
|
|
$lookup = new LocalIdLookup();
|
|
$user1 = User::newFromName( 'UTLocalIdLookup1' );
|
|
$user2 = User::newFromName( 'UTLocalIdLookup2' );
|
|
|
|
$this->assertTrue( $user1->isAllowed( 'hideuser' ), 'sanity check' );
|
|
$this->assertFalse( $user2->isAllowed( 'hideuser' ), 'sanity check' );
|
|
|
|
$this->assertSame( array(), $lookup->lookupCentralIds( array() ) );
|
|
|
|
$expect = array_flip( $this->localUsers );
|
|
$expect[123] = 'X';
|
|
ksort( $expect );
|
|
|
|
$expect2 = $expect;
|
|
$expect2[$this->localUsers['UTLocalIdLookup4']] = '';
|
|
|
|
$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, $user1 ) );
|
|
$this->assertSame( $expect2, $lookup->lookupCentralIds( $arg, $user2 ) );
|
|
}
|
|
|
|
public function testLookupUserNames() {
|
|
$lookup = new LocalIdLookup();
|
|
$user1 = User::newFromName( 'UTLocalIdLookup1' );
|
|
$user2 = User::newFromName( 'UTLocalIdLookup2' );
|
|
|
|
$this->assertTrue( $user1->isAllowed( 'hideuser' ), 'sanity check' );
|
|
$this->assertFalse( $user2->isAllowed( 'hideuser' ), 'sanity check' );
|
|
|
|
$this->assertSame( array(), $lookup->lookupUserNames( array() ) );
|
|
|
|
$expect = $this->localUsers;
|
|
$expect['UTDoesNotExist'] = 'X';
|
|
ksort( $expect );
|
|
|
|
$expect2 = $expect;
|
|
$expect2['UTLocalIdLookup4'] = '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, $user1 ) );
|
|
$this->assertSame( $expect2, $lookup->lookupUserNames( $arg, $user2 ) );
|
|
}
|
|
|
|
public function testIsAttached() {
|
|
$lookup = new LocalIdLookup();
|
|
$user1 = User::newFromName( 'UTLocalIdLookup1' );
|
|
$user2 = User::newFromName( 'DoesNotExist' );
|
|
|
|
$this->assertTrue( $lookup->isAttached( $user1 ) );
|
|
$this->assertFalse( $lookup->isAttached( $user2 ) );
|
|
|
|
$wiki = wfWikiId();
|
|
$this->assertTrue( $lookup->isAttached( $user1, $wiki ) );
|
|
$this->assertFalse( $lookup->isAttached( $user2, $wiki ) );
|
|
|
|
$wiki = 'not-' . wfWikiId();
|
|
$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 ) {
|
|
global $wgDBName;
|
|
$this->setMwGlobals( array(
|
|
'wgSharedDB' => $sharedDB ? $wgDBName : null,
|
|
'wgSharedTables' => $sharedTable ? array( 'user' ) : array(),
|
|
'wgLocalDatabases' => $localDBSet ? array( 'shared' ) : array(),
|
|
) );
|
|
|
|
$lookup = new LocalIdLookup();
|
|
$this->assertSame(
|
|
$sharedDB && $sharedTable && $localDBSet,
|
|
$lookup->isAttached( User::newFromName( 'UTLocalIdLookup1' ), 'shared' )
|
|
);
|
|
}
|
|
|
|
public static function provideIsAttachedShared() {
|
|
$ret = array();
|
|
for ( $i = 0; $i < 7; $i++ ) {
|
|
$ret[] = array(
|
|
(bool)( $i & 1 ),
|
|
(bool)( $i & 2 ),
|
|
(bool)( $i & 4 ),
|
|
);
|
|
}
|
|
return $ret;
|
|
}
|
|
|
|
}
|