nextcloud/apps/user_ldap/tests/WizardTest.php

449 lines
12 KiB
PHP

<?php
/**
* @copyright Copyright (c) 2016, ownCloud, Inc.
*
* @author Arthur Schiwon <blizzz@arthur-schiwon.de>
* @author Joas Schilling <coding@schilljs.com>
* @author Morris Jobke <hey@morrisjobke.de>
* @author Thomas Müller <thomas.mueller@tmit.eu>
* @author Viktor Szépe <viktor@szepe.net>
*
* @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\ILDAPWrapper;
use \OCA\User_LDAP\Wizard;
/**
* Class Test_Wizard
*
* @group DB
*
* @package OCA\User_LDAP\Tests
*/
class WizardTest extends \Test\TestCase {
protected function setUp() {
parent::setUp();
//we need to make sure the consts are defined, otherwise tests will fail
//on systems without php5_ldap
$ldapConsts = array('LDAP_OPT_PROTOCOL_VERSION',
'LDAP_OPT_REFERRALS', 'LDAP_OPT_NETWORK_TIMEOUT');
foreach($ldapConsts as $const) {
if(!defined($const)) {
define($const, 42);
}
}
}
private function getWizardAndMocks() {
static $confMethods;
static $connMethods;
static $accMethods;
if(is_null($confMethods)) {
$confMethods = get_class_methods('\OCA\User_LDAP\Configuration');
$connMethods = get_class_methods('\OCA\User_LDAP\Connection');
$accMethods = get_class_methods('\OCA\User_LDAP\Access');
}
$lw = $this->createMock(ILDAPWrapper::class);
$conf = $this->getMock('\OCA\User_LDAP\Configuration',
$confMethods,
array($lw, null, null));
$connector = $this->getMock('\OCA\User_LDAP\Connection',
$connMethods, array($lw, null, null));
$um = $this->getMockBuilder('\OCA\User_LDAP\User\Manager')
->disableOriginalConstructor()
->getMock();
$helper = new \OCA\User_LDAP\Helper();
$access = $this->getMock('\OCA\User_LDAP\Access',
$accMethods, array($connector, $lw, $um, $helper));
return array(new Wizard($conf, $lw, $access), $conf, $lw, $access);
}
private function prepareLdapWrapperForConnections(&$ldap) {
$ldap->expects($this->once())
->method('connect')
//dummy value, usually invalid
->will($this->returnValue(true));
$ldap->expects($this->exactly(3))
->method('setOption')
->will($this->returnValue(true));
$ldap->expects($this->once())
->method('bind')
->will($this->returnValue(true));
}
public function testCumulativeSearchOnAttributeLimited() {
list($wizard, $configuration, $ldap) = $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function($name) {
if($name === 'ldapBase') {
return array('base');
}
return null;
}));
$this->prepareLdapWrapperForConnections($ldap);
$ldap->expects($this->any())
->method('isResource')
->will($this->returnValue(true));
$ldap->expects($this->exactly(2))
->method('search')
//dummy value, usually invalid
->will($this->returnValue(true));
$ldap->expects($this->exactly(2))
->method('countEntries')
//an is_resource check will follow, so we need to return a dummy resource
->will($this->returnValue(23));
//5 DNs per filter means 2x firstEntry and 8x nextEntry
$ldap->expects($this->exactly(2))
->method('firstEntry')
//dummy value, usually invalid
->will($this->returnValue(true));
$ldap->expects($this->exactly(8))
->method('nextEntry')
//dummy value, usually invalid
->will($this->returnValue(true));
$ldap->expects($this->exactly(10))
->method('getAttributes')
//dummy value, usually invalid
->will($this->returnValue(array('cn' => array('foo'), 'count' => 1)));
global $uidnumber;
$uidnumber = 1;
$ldap->expects($this->exactly(10))
->method('getDN')
//dummy value, usually invalid
->will($this->returnCallback(function($a, $b) {
global $uidnumber;
return $uidnumber++;
}));
// The following expectations are the real test
$filters = array('f1', 'f2', '*');
$wizard->cumulativeSearchOnAttribute($filters, 'cn', 5);
unset($uidnumber);
}
public function testCumulativeSearchOnAttributeUnlimited() {
list($wizard, $configuration, $ldap) = $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function($name) {
if($name === 'ldapBase') {
return array('base');
}
return null;
}));
$this->prepareLdapWrapperForConnections($ldap);
$ldap->expects($this->any())
->method('isResource')
->will($this->returnCallback(function($r) {
if($r === true) {
return true;
}
if($r % 24 === 0) {
global $uidnumber;
$uidnumber++;
return false;
}
return true;
}));
$ldap->expects($this->exactly(2))
->method('search')
//dummy value, usually invalid
->will($this->returnValue(true));
$ldap->expects($this->exactly(2))
->method('countEntries')
//an is_resource check will follow, so we need to return a dummy resource
->will($this->returnValue(23));
//5 DNs per filter means 2x firstEntry and 8x nextEntry
$ldap->expects($this->exactly(2))
->method('firstEntry')
//dummy value, usually invalid
->will($this->returnCallback(function($r) {
global $uidnumber;
return $uidnumber;
}));
$ldap->expects($this->exactly(46))
->method('nextEntry')
//dummy value, usually invalid
->will($this->returnCallback(function($r) {
global $uidnumber;
return $uidnumber;
}));
$ldap->expects($this->exactly(46))
->method('getAttributes')
//dummy value, usually invalid
->will($this->returnValue(array('cn' => array('foo'), 'count' => 1)));
global $uidnumber;
$uidnumber = 1;
$ldap->expects($this->exactly(46))
->method('getDN')
//dummy value, usually invalid
->will($this->returnCallback(function($a, $b) {
global $uidnumber;
return $uidnumber++;
}));
// The following expectations are the real test
$filters = array('f1', 'f2', '*');
$wizard->cumulativeSearchOnAttribute($filters, 'cn', 0);
unset($uidnumber);
}
public function testDetectEmailAttributeAlreadySet() {
list($wizard, $configuration, $ldap, $access)
= $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function ($name) {
if($name === 'ldapEmailAttribute') {
return 'myEmailAttribute';
} else {
//for requirement checks
return 'let me pass';
}
}));
$access->expects($this->once())
->method('countUsers')
->will($this->returnValue(42));
$wizard->detectEmailAttribute();
}
public function testDetectEmailAttributeOverrideSet() {
list($wizard, $configuration, $ldap, $access)
= $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function ($name) {
if($name === 'ldapEmailAttribute') {
return 'myEmailAttribute';
} else {
//for requirement checks
return 'let me pass';
}
}));
$access->expects($this->exactly(3))
->method('combineFilterWithAnd')
->will($this->returnCallback(function ($filterParts) {
return str_replace('=*', '', array_pop($filterParts));
}));
$access->expects($this->exactly(3))
->method('countUsers')
->will($this->returnCallback(function ($filter) {
if($filter === 'myEmailAttribute') {
return 0;
} else if($filter === 'mail') {
return 3;
} else if($filter === 'mailPrimaryAddress') {
return 17;
}
throw new \Exception('Untested filter: ' . $filter);
}));
$result = $wizard->detectEmailAttribute()->getResultArray();
$this->assertSame('mailPrimaryAddress',
$result['changes']['ldap_email_attr']);
}
public function testDetectEmailAttributeFind() {
list($wizard, $configuration, $ldap, $access)
= $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function ($name) {
if($name === 'ldapEmailAttribute') {
return '';
} else {
//for requirement checks
return 'let me pass';
}
}));
$access->expects($this->exactly(2))
->method('combineFilterWithAnd')
->will($this->returnCallback(function ($filterParts) {
return str_replace('=*', '', array_pop($filterParts));
}));
$access->expects($this->exactly(2))
->method('countUsers')
->will($this->returnCallback(function ($filter) {
if($filter === 'myEmailAttribute') {
return 0;
} else if($filter === 'mail') {
return 3;
} else if($filter === 'mailPrimaryAddress') {
return 17;
}
throw new \Exception('Untested filter: ' . $filter);
}));
$result = $wizard->detectEmailAttribute()->getResultArray();
$this->assertSame('mailPrimaryAddress',
$result['changes']['ldap_email_attr']);
}
public function testDetectEmailAttributeFindNothing() {
list($wizard, $configuration, $ldap, $access)
= $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function ($name) {
if($name === 'ldapEmailAttribute') {
return 'myEmailAttribute';
} else {
//for requirement checks
return 'let me pass';
}
}));
$access->expects($this->exactly(3))
->method('combineFilterWithAnd')
->will($this->returnCallback(function ($filterParts) {
return str_replace('=*', '', array_pop($filterParts));
}));
$access->expects($this->exactly(3))
->method('countUsers')
->will($this->returnCallback(function ($filter) {
if($filter === 'myEmailAttribute') {
return 0;
} else if($filter === 'mail') {
return 0;
} else if($filter === 'mailPrimaryAddress') {
return 0;
}
throw new \Exception('Untested filter: ' . $filter);
}));
$result = $wizard->detectEmailAttribute();
$this->assertSame(false, $result->hasChanges());
}
public function testCumulativeSearchOnAttributeSkipReadDN() {
// tests that there is no infinite loop, when skipping already processed
// DNs (they can be returned multiple times for multiple filters )
list($wizard, $configuration, $ldap) = $this->getWizardAndMocks();
$configuration->expects($this->any())
->method('__get')
->will($this->returnCallback(function($name) {
if($name === 'ldapBase') {
return array('base');
}
return null;
}));
$this->prepareLdapWrapperForConnections($ldap);
$ldap->expects($this->any())
->method('isResource')
->will($this->returnCallback(function($res) {
return (bool)$res;
}));
$ldap->expects($this->any())
->method('search')
//dummy value, usually invalid
->will($this->returnValue(true));
$ldap->expects($this->any())
->method('countEntries')
//an is_resource check will follow, so we need to return a dummy resource
->will($this->returnValue(7));
//5 DNs per filter means 2x firstEntry and 8x nextEntry
$ldap->expects($this->any())
->method('firstEntry')
//dummy value, usually invalid
->will($this->returnValue(1));
global $mark;
$mark = false;
// entries return order: 1, 2, 3, 4, 4, 5, 6
$ldap->expects($this->any())
->method('nextEntry')
//dummy value, usually invalid
->will($this->returnCallback(function($a, $prev){
$current = $prev + 1;
if($current === 7) {
return false;
}
global $mark;
if($prev === 4 && !$mark) {
$mark = true;
return 4;
}
return $current;
}));
$ldap->expects($this->any())
->method('getAttributes')
//dummy value, usually invalid
->will($this->returnCallback(function($a, $entry) {
return array('cn' => array($entry), 'count' => 1);
}));
$ldap->expects($this->any())
->method('getDN')
//dummy value, usually invalid
->will($this->returnCallback(function($a, $b) {
return $b;
}));
// The following expectations are the real test
$filters = array('f1', 'f2', '*');
$resultArray = $wizard->cumulativeSearchOnAttribute($filters, 'cn', 0);
$this->assertSame(6, count($resultArray));
unset($mark);
}
}