458 lines
12 KiB
PHP
458 lines
12 KiB
PHP
<?php
|
||
/**
|
||
* @author Arthur Schiwon <blizzz@arthur-schiwon.de>
|
||
* @author Frédéric Fortier <frederic.fortier@oronospolytechnique.com>
|
||
* @author Joas Schilling <nickvergessen@owncloud.com>
|
||
* @author Lukas Reschke <lukas@statuscode.ch>
|
||
* @author Morris Jobke <hey@morrisjobke.de>
|
||
* @author Thomas Müller <thomas.mueller@tmit.eu>
|
||
*
|
||
* @copyright Copyright (c) 2016, ownCloud, Inc.
|
||
* @license AGPL-3.0
|
||
*
|
||
* This code is free software: you can redistribute it and/or modify
|
||
* it under the terms of the GNU Affero General Public License, version 3,
|
||
* as published by the Free Software Foundation.
|
||
*
|
||
* This program is distributed in the hope that it will be useful,
|
||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
* GNU Affero General Public License for more details.
|
||
*
|
||
* You should have received a copy of the GNU Affero General Public License, version 3,
|
||
* along with this program. If not, see <http://www.gnu.org/licenses/>
|
||
*
|
||
*/
|
||
|
||
namespace OCA\User_LDAP\Tests;
|
||
|
||
use OCA\User_LDAP\Group_LDAP as GroupLDAP;
|
||
use OCA\User_LDAP\Access;
|
||
use OCA\User_LDAP\Connection;
|
||
|
||
/**
|
||
* Class GroupLDAPTest
|
||
*
|
||
* @group DB
|
||
*
|
||
* @package OCA\User_LDAP\Tests
|
||
*/
|
||
class Group_LDAPTest extends \Test\TestCase {
|
||
private function getAccessMock() {
|
||
static $conMethods;
|
||
static $accMethods;
|
||
|
||
if(is_null($conMethods) || is_null($accMethods)) {
|
||
$conMethods = get_class_methods('\OCA\User_LDAP\Connection');
|
||
$accMethods = get_class_methods('\OCA\User_LDAP\Access');
|
||
}
|
||
$lw = $this->getMock('\OCA\User_LDAP\ILDAPWrapper');
|
||
$connector = $this->getMock('\OCA\User_LDAP\Connection',
|
||
$conMethods,
|
||
array($lw, null, null));
|
||
$um = $this->getMockBuilder('\OCA\User_LDAP\User\Manager')
|
||
->disableOriginalConstructor()
|
||
->getMock();
|
||
$access = $this->getMock('\OCA\User_LDAP\Access',
|
||
$accMethods,
|
||
array($connector, $lw, $um));
|
||
|
||
$access->expects($this->any())
|
||
->method('getConnection')
|
||
->will($this->returnValue($connector));
|
||
|
||
return $access;
|
||
}
|
||
|
||
private function enableGroups($access) {
|
||
$access->connection->expects($this->any())
|
||
->method('__get')
|
||
->will($this->returnCallback(function($name) {
|
||
if($name === 'ldapDynamicGroupMemberURL') {
|
||
return '';
|
||
}
|
||
return 1;
|
||
}));
|
||
}
|
||
|
||
public function testCountEmptySearchString() {
|
||
$access = $this->getAccessMock();
|
||
|
||
$this->enableGroups($access);
|
||
|
||
$access->expects($this->any())
|
||
->method('groupname2dn')
|
||
->will($this->returnValue('cn=group,dc=foo,dc=bar'));
|
||
|
||
$access->expects($this->any())
|
||
->method('readAttribute')
|
||
->will($this->returnValue(array('u11', 'u22', 'u33', 'u34')));
|
||
|
||
// for primary groups
|
||
$access->expects($this->once())
|
||
->method('countUsers')
|
||
->will($this->returnValue(2));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$users = $groupBackend->countUsersInGroup('group');
|
||
|
||
$this->assertSame(6, $users);
|
||
}
|
||
|
||
public function testCountWithSearchString() {
|
||
$access = $this->getAccessMock();
|
||
|
||
$this->enableGroups($access);
|
||
|
||
$access->expects($this->any())
|
||
->method('groupname2dn')
|
||
->will($this->returnValue('cn=group,dc=foo,dc=bar'));
|
||
|
||
$access->expects($this->any())
|
||
->method('fetchListOfUsers')
|
||
->will($this->returnValue(array()));
|
||
|
||
$access->expects($this->any())
|
||
->method('readAttribute')
|
||
->will($this->returnCallback(function($name) {
|
||
//the search operation will call readAttribute, thus we need
|
||
//to anaylze the "dn". All other times we just need to return
|
||
//something that is neither null or false, but once an array
|
||
//with the users in the group – so we do so all other times for
|
||
//simplicicity.
|
||
if(strpos($name, 'u') === 0) {
|
||
return strpos($name, '3');
|
||
}
|
||
return array('u11', 'u22', 'u33', 'u34');
|
||
}));
|
||
|
||
$access->expects($this->any())
|
||
->method('dn2username')
|
||
->will($this->returnCallback(function() {
|
||
return 'foobar' . \OCP\Util::generateRandomBytes(7);
|
||
}));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$users = $groupBackend->countUsersInGroup('group', '3');
|
||
|
||
$this->assertSame(2, $users);
|
||
}
|
||
|
||
public function testPrimaryGroupID2NameSuccess() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$userDN = 'cn=alice,cn=foo,dc=barfoo,dc=bar';
|
||
|
||
$access->expects($this->once())
|
||
->method('getSID')
|
||
->with($userDN)
|
||
->will($this->returnValue('S-1-5-21-249921958-728525901-1594176202'));
|
||
|
||
$access->expects($this->once())
|
||
->method('searchGroups')
|
||
->will($this->returnValue([['dn' => ['cn=foo,dc=barfoo,dc=bar']]]));
|
||
|
||
$access->expects($this->once())
|
||
->method('dn2groupname')
|
||
->with('cn=foo,dc=barfoo,dc=bar')
|
||
->will($this->returnValue('MyGroup'));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
|
||
$group = $groupBackend->primaryGroupID2Name('3117', $userDN);
|
||
|
||
$this->assertSame('MyGroup', $group);
|
||
}
|
||
|
||
public function testPrimaryGroupID2NameNoSID() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$userDN = 'cn=alice,cn=foo,dc=barfoo,dc=bar';
|
||
|
||
$access->expects($this->once())
|
||
->method('getSID')
|
||
->with($userDN)
|
||
->will($this->returnValue(false));
|
||
|
||
$access->expects($this->never())
|
||
->method('searchGroups');
|
||
|
||
$access->expects($this->never())
|
||
->method('dn2groupname');
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
|
||
$group = $groupBackend->primaryGroupID2Name('3117', $userDN);
|
||
|
||
$this->assertSame(false, $group);
|
||
}
|
||
|
||
public function testPrimaryGroupID2NameNoGroup() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$userDN = 'cn=alice,cn=foo,dc=barfoo,dc=bar';
|
||
|
||
$access->expects($this->once())
|
||
->method('getSID')
|
||
->with($userDN)
|
||
->will($this->returnValue('S-1-5-21-249921958-728525901-1594176202'));
|
||
|
||
$access->expects($this->once())
|
||
->method('searchGroups')
|
||
->will($this->returnValue(array()));
|
||
|
||
$access->expects($this->never())
|
||
->method('dn2groupname');
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
|
||
$group = $groupBackend->primaryGroupID2Name('3117', $userDN);
|
||
|
||
$this->assertSame(false, $group);
|
||
}
|
||
|
||
public function testPrimaryGroupID2NameNoName() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$userDN = 'cn=alice,cn=foo,dc=barfoo,dc=bar';
|
||
|
||
$access->expects($this->once())
|
||
->method('getSID')
|
||
->with($userDN)
|
||
->will($this->returnValue('S-1-5-21-249921958-728525901-1594176202'));
|
||
|
||
$access->expects($this->once())
|
||
->method('searchGroups')
|
||
->will($this->returnValue([['dn' => ['cn=foo,dc=barfoo,dc=bar']]]));
|
||
|
||
$access->expects($this->once())
|
||
->method('dn2groupname')
|
||
->will($this->returnValue(false));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
|
||
$group = $groupBackend->primaryGroupID2Name('3117', $userDN);
|
||
|
||
$this->assertSame(false, $group);
|
||
}
|
||
|
||
public function testGetEntryGroupIDValue() {
|
||
//tests getEntryGroupID via getGroupPrimaryGroupID
|
||
//which is basically identical to getUserPrimaryGroupIDs
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$dn = 'cn=foobar,cn=foo,dc=barfoo,dc=bar';
|
||
$attr = 'primaryGroupToken';
|
||
|
||
$access->expects($this->once())
|
||
->method('readAttribute')
|
||
->with($dn, $attr)
|
||
->will($this->returnValue(array('3117')));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
|
||
$gid = $groupBackend->getGroupPrimaryGroupID($dn);
|
||
|
||
$this->assertSame('3117', $gid);
|
||
}
|
||
|
||
public function testGetEntryGroupIDNoValue() {
|
||
//tests getEntryGroupID via getGroupPrimaryGroupID
|
||
//which is basically identical to getUserPrimaryGroupIDs
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$dn = 'cn=foobar,cn=foo,dc=barfoo,dc=bar';
|
||
$attr = 'primaryGroupToken';
|
||
|
||
$access->expects($this->once())
|
||
->method('readAttribute')
|
||
->with($dn, $attr)
|
||
->will($this->returnValue(false));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
|
||
$gid = $groupBackend->getGroupPrimaryGroupID($dn);
|
||
|
||
$this->assertSame(false, $gid);
|
||
}
|
||
|
||
/**
|
||
* tests whether Group Backend behaves correctly when cache with uid and gid
|
||
* is hit
|
||
*/
|
||
public function testInGroupHitsUidGidCache() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$uid = 'someUser';
|
||
$gid = 'someGroup';
|
||
$cacheKey = 'inGroup'.$uid.':'.$gid;
|
||
|
||
$access->connection->expects($this->once())
|
||
->method('getFromCache')
|
||
->with($cacheKey)
|
||
->will($this->returnValue(true));
|
||
|
||
$access->expects($this->never())
|
||
->method('username2dn');
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$groupBackend->inGroup($uid, $gid);
|
||
}
|
||
|
||
public function testGetGroupsWithOffset() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$access->expects($this->once())
|
||
->method('ownCloudGroupNames')
|
||
->will($this->returnValue(array('group1', 'group2')));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$groups = $groupBackend->getGroups('', 2, 2);
|
||
|
||
$this->assertSame(2, count($groups));
|
||
}
|
||
|
||
/**
|
||
* tests that a user listing is complete, if all it's members have the group
|
||
* as their primary.
|
||
*/
|
||
public function testUsersInGroupPrimaryMembersOnly() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$access->connection->expects($this->any())
|
||
->method('getFromCache')
|
||
->will($this->returnValue(null));
|
||
|
||
$access->expects($this->any())
|
||
->method('readAttribute')
|
||
->will($this->returnCallback(function($dn, $attr) {
|
||
if($attr === 'primaryGroupToken') {
|
||
return array(1337);
|
||
}
|
||
return array();
|
||
}));
|
||
|
||
$access->expects($this->any())
|
||
->method('groupname2dn')
|
||
->will($this->returnValue('cn=foobar,dc=foo,dc=bar'));
|
||
|
||
$access->expects($this->once())
|
||
->method('ownCloudUserNames')
|
||
->will($this->returnValue(array('lisa', 'bart', 'kira', 'brad')));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$users = $groupBackend->usersInGroup('foobar');
|
||
|
||
$this->assertSame(4, count($users));
|
||
}
|
||
|
||
/**
|
||
* tests that a user counting is complete, if all it's members have the group
|
||
* as their primary.
|
||
*/
|
||
public function testCountUsersInGroupPrimaryMembersOnly() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$access->connection->expects($this->any())
|
||
->method('getFromCache')
|
||
->will($this->returnValue(null));
|
||
|
||
$access->expects($this->any())
|
||
->method('readAttribute')
|
||
->will($this->returnCallback(function($dn, $attr) {
|
||
if($attr === 'primaryGroupToken') {
|
||
return array(1337);
|
||
}
|
||
return array();
|
||
}));
|
||
|
||
$access->expects($this->any())
|
||
->method('groupname2dn')
|
||
->will($this->returnValue('cn=foobar,dc=foo,dc=bar'));
|
||
|
||
$access->expects($this->once())
|
||
->method('countUsers')
|
||
->will($this->returnValue(4));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$users = $groupBackend->countUsersInGroup('foobar');
|
||
|
||
$this->assertSame(4, $users);
|
||
}
|
||
|
||
public function testGetUserGroupsMemberOf() {
|
||
$access = $this->getAccessMock();
|
||
$this->enableGroups($access);
|
||
|
||
$dn = 'cn=userX,dc=foobar';
|
||
|
||
$access->connection->hasPrimaryGroups = false;
|
||
|
||
$access->expects($this->any())
|
||
->method('username2dn')
|
||
->will($this->returnValue($dn));
|
||
|
||
$access->expects($this->exactly(3))
|
||
->method('readAttribute')
|
||
->will($this->onConsecutiveCalls(['cn=groupA,dc=foobar', 'cn=groupB,dc=foobar'], [], []));
|
||
|
||
$access->expects($this->exactly(2))
|
||
->method('dn2groupname')
|
||
->will($this->returnArgument(0));
|
||
|
||
$access->expects($this->exactly(3))
|
||
->method('groupsMatchFilter')
|
||
->will($this->returnArgument(0));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$groups = $groupBackend->getUserGroups('userX');
|
||
|
||
$this->assertSame(2, count($groups));
|
||
}
|
||
|
||
public function testGetUserGroupsMemberOfDisabled() {
|
||
$access = $this->getAccessMock();
|
||
|
||
$access->connection->expects($this->any())
|
||
->method('__get')
|
||
->will($this->returnCallback(function($name) {
|
||
if($name === 'useMemberOfToDetectMembership') {
|
||
return 0;
|
||
} else if($name === 'ldapDynamicGroupMemberURL') {
|
||
return '';
|
||
}
|
||
return 1;
|
||
}));
|
||
|
||
$dn = 'cn=userX,dc=foobar';
|
||
|
||
$access->connection->hasPrimaryGroups = false;
|
||
|
||
$access->expects($this->once())
|
||
->method('username2dn')
|
||
->will($this->returnValue($dn));
|
||
|
||
$access->expects($this->never())
|
||
->method('readAttribute')
|
||
->with($dn, 'memberOf');
|
||
|
||
$access->expects($this->once())
|
||
->method('ownCloudGroupNames')
|
||
->will($this->returnValue([]));
|
||
|
||
$groupBackend = new GroupLDAP($access);
|
||
$groupBackend->getUserGroups('userX');
|
||
}
|
||
|
||
}
|