2015-04-01 20:07:54 +03:00
|
|
|
<?php
|
|
|
|
/**
|
2016-07-21 17:49:16 +03:00
|
|
|
* @copyright Copyright (c) 2016, ownCloud, Inc.
|
|
|
|
*
|
2017-11-06 17:56:42 +03:00
|
|
|
* @author Bjoern Schiessle <bjoern@schiessle.org>
|
2016-05-26 20:56:05 +03:00
|
|
|
* @author Björn Schießle <bjoern@schiessle.org>
|
2015-04-07 18:02:49 +03:00
|
|
|
* @author Clark Tomlinson <fallen013@gmail.com>
|
2016-07-21 17:49:16 +03:00
|
|
|
* @author Joas Schilling <coding@schilljs.com>
|
2017-11-06 17:56:42 +03:00
|
|
|
* @author Morris Jobke <hey@morrisjobke.de>
|
|
|
|
* @author Roeland Jago Douma <roeland@famdouma.nl>
|
2015-04-07 18:02:49 +03:00
|
|
|
*
|
2015-04-01 20:07:54 +03:00
|
|
|
* @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\Encryption\Tests;
|
|
|
|
|
|
|
|
|
2016-09-02 11:29:05 +03:00
|
|
|
use OC\Files\View;
|
2017-10-26 14:46:16 +03:00
|
|
|
use OCA\Encryption\Crypto\Crypt;
|
2015-04-01 20:07:54 +03:00
|
|
|
use OCA\Encryption\Util;
|
2016-09-02 11:29:05 +03:00
|
|
|
use OCP\Files\Mount\IMountPoint;
|
2017-10-26 14:46:16 +03:00
|
|
|
use OCP\Files\Storage;
|
2016-09-02 11:29:05 +03:00
|
|
|
use OCP\IConfig;
|
|
|
|
use OCP\ILogger;
|
2019-08-26 16:22:00 +03:00
|
|
|
use OCP\IUser;
|
2016-09-02 11:29:05 +03:00
|
|
|
use OCP\IUserManager;
|
2017-10-26 14:46:16 +03:00
|
|
|
use OCP\IUserSession;
|
2019-08-26 16:22:00 +03:00
|
|
|
use PHPUnit\Framework\MockObject\MockObject;
|
2015-04-01 20:07:54 +03:00
|
|
|
use Test\TestCase;
|
|
|
|
|
|
|
|
class UtilTest extends TestCase {
|
|
|
|
private static $tempStorage = [];
|
2015-05-12 19:49:25 +03:00
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
/** @var \OCP\IConfig|\PHPUnit_Framework_MockObject_MockObject */
|
2015-04-01 20:07:54 +03:00
|
|
|
private $configMock;
|
2015-05-12 19:49:25 +03:00
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
/** @var \OC\Files\View|\PHPUnit_Framework_MockObject_MockObject */
|
2015-04-01 20:07:54 +03:00
|
|
|
private $filesMock;
|
2015-05-12 19:49:25 +03:00
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
/** @var \OCP\IUserManager|\PHPUnit_Framework_MockObject_MockObject */
|
2015-05-12 19:49:25 +03:00
|
|
|
private $userManagerMock;
|
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
/** @var \OCP\Files\Mount\IMountPoint|\PHPUnit_Framework_MockObject_MockObject */
|
2015-10-13 18:54:06 +03:00
|
|
|
private $mountMock;
|
|
|
|
|
2015-05-12 19:49:25 +03:00
|
|
|
/** @var Util */
|
2015-04-01 20:07:54 +03:00
|
|
|
private $instance;
|
|
|
|
|
|
|
|
public function testSetRecoveryForUser() {
|
|
|
|
$this->instance->setRecoveryForUser('1');
|
|
|
|
$this->assertArrayHasKey('recoveryEnabled', self::$tempStorage);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testIsRecoveryEnabledForUser() {
|
2015-04-29 18:18:41 +03:00
|
|
|
$this->assertTrue($this->instance->isRecoveryEnabledForUser('admin'));
|
2015-04-01 20:07:54 +03:00
|
|
|
|
|
|
|
// Assert recovery will return default value if not set
|
|
|
|
unset(self::$tempStorage['recoveryEnabled']);
|
2015-04-29 18:18:41 +03:00
|
|
|
$this->assertEquals(0, $this->instance->isRecoveryEnabledForUser('admin'));
|
2015-04-01 20:07:54 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
public function testUserHasFiles() {
|
|
|
|
$this->filesMock->expects($this->once())
|
|
|
|
->method('file_exists')
|
|
|
|
->will($this->returnValue(true));
|
|
|
|
|
|
|
|
$this->assertTrue($this->instance->userHasFiles('admin'));
|
|
|
|
}
|
|
|
|
|
|
|
|
protected function setUp() {
|
|
|
|
parent::setUp();
|
2016-09-02 11:29:05 +03:00
|
|
|
$this->mountMock = $this->createMock(IMountPoint::class);
|
|
|
|
$this->filesMock = $this->createMock(View::class);
|
|
|
|
$this->userManagerMock = $this->createMock(IUserManager::class);
|
2015-04-01 20:07:54 +03:00
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
/** @var \OCA\Encryption\Crypto\Crypt $cryptMock */
|
2017-10-26 14:46:16 +03:00
|
|
|
$cryptMock = $this->getMockBuilder(Crypt::class)
|
2015-04-01 20:07:54 +03:00
|
|
|
->disableOriginalConstructor()
|
|
|
|
->getMock();
|
2016-05-12 10:42:19 +03:00
|
|
|
/** @var \OCP\ILogger $loggerMock */
|
2016-09-02 11:29:05 +03:00
|
|
|
$loggerMock = $this->createMock(ILogger::class);
|
2015-04-01 20:07:54 +03:00
|
|
|
|
2019-08-26 16:22:00 +03:00
|
|
|
$user = $this->createMock(IUser::class);
|
|
|
|
$user->expects($this->any())
|
|
|
|
->method('getUID')
|
|
|
|
->willReturn('admin');
|
2015-04-01 20:07:54 +03:00
|
|
|
|
2019-08-26 16:22:00 +03:00
|
|
|
/** @var IUserSession|MockObject $userSessionMock */
|
|
|
|
$userSessionMock = $this->createMock(IUserSession::class);
|
2015-04-01 20:07:54 +03:00
|
|
|
$userSessionMock->expects($this->any())
|
2019-08-26 16:22:00 +03:00
|
|
|
->method('getUser')
|
|
|
|
->willReturn($user);
|
|
|
|
$userSessionMock->expects($this->any())
|
|
|
|
->method('isLoggedIn')
|
|
|
|
->willReturn(true);
|
2015-04-01 20:07:54 +03:00
|
|
|
|
2016-09-02 11:29:05 +03:00
|
|
|
$this->configMock = $this->createMock(IConfig::class);
|
2015-04-01 20:07:54 +03:00
|
|
|
|
|
|
|
$this->configMock->expects($this->any())
|
|
|
|
->method('getUserValue')
|
|
|
|
->will($this->returnCallback([$this, 'getValueTester']));
|
|
|
|
|
|
|
|
$this->configMock->expects($this->any())
|
|
|
|
->method('setUserValue')
|
|
|
|
->will($this->returnCallback([$this, 'setValueTester']));
|
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
$this->instance = new Util($this->filesMock, $cryptMock, $loggerMock, $userSessionMock, $this->configMock, $this->userManagerMock);
|
2015-04-01 20:07:54 +03:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param $userId
|
|
|
|
* @param $app
|
|
|
|
* @param $key
|
|
|
|
* @param $value
|
|
|
|
*/
|
|
|
|
public function setValueTester($userId, $app, $key, $value) {
|
|
|
|
self::$tempStorage[$key] = $value;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @param $userId
|
|
|
|
* @param $app
|
|
|
|
* @param $key
|
|
|
|
* @param $default
|
|
|
|
* @return mixed
|
|
|
|
*/
|
|
|
|
public function getValueTester($userId, $app, $key, $default) {
|
|
|
|
if (!empty(self::$tempStorage[$key])) {
|
|
|
|
return self::$tempStorage[$key];
|
|
|
|
}
|
|
|
|
return $default ?: null;
|
|
|
|
}
|
|
|
|
|
2015-09-07 12:38:44 +03:00
|
|
|
/**
|
|
|
|
* @dataProvider dataTestIsMasterKeyEnabled
|
|
|
|
*
|
|
|
|
* @param string $value
|
|
|
|
* @param bool $expect
|
|
|
|
*/
|
|
|
|
public function testIsMasterKeyEnabled($value, $expect) {
|
|
|
|
$this->configMock->expects($this->once())->method('getAppValue')
|
2017-06-02 12:05:10 +03:00
|
|
|
->with('encryption', 'useMasterKey', '1')->willReturn($value);
|
2015-09-07 12:38:44 +03:00
|
|
|
$this->assertSame($expect,
|
|
|
|
$this->instance->isMasterKeyEnabled()
|
|
|
|
);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function dataTestIsMasterKeyEnabled() {
|
|
|
|
return [
|
|
|
|
['0', false],
|
|
|
|
['1', true]
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
2015-10-13 18:54:06 +03:00
|
|
|
/**
|
|
|
|
* @dataProvider dataTestShouldEncryptHomeStorage
|
2016-05-12 10:42:19 +03:00
|
|
|
* @param string $returnValue return value from getAppValue()
|
|
|
|
* @param bool $expected
|
2015-10-13 18:54:06 +03:00
|
|
|
*/
|
|
|
|
public function testShouldEncryptHomeStorage($returnValue, $expected) {
|
|
|
|
$this->configMock->expects($this->once())->method('getAppValue')
|
|
|
|
->with('encryption', 'encryptHomeStorage', '1')
|
|
|
|
->willReturn($returnValue);
|
|
|
|
|
|
|
|
$this->assertSame($expected,
|
|
|
|
$this->instance->shouldEncryptHomeStorage());
|
|
|
|
}
|
|
|
|
|
|
|
|
public function dataTestShouldEncryptHomeStorage() {
|
|
|
|
return [
|
|
|
|
['1', true],
|
|
|
|
['0', false]
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @dataProvider dataTestSetEncryptHomeStorage
|
|
|
|
* @param $value
|
|
|
|
* @param $expected
|
|
|
|
*/
|
|
|
|
public function testSetEncryptHomeStorage($value, $expected) {
|
|
|
|
$this->configMock->expects($this->once())->method('setAppValue')
|
|
|
|
->with('encryption', 'encryptHomeStorage', $expected);
|
|
|
|
$this->instance->setEncryptHomeStorage($value);
|
|
|
|
}
|
|
|
|
|
|
|
|
public function dataTestSetEncryptHomeStorage() {
|
|
|
|
return [
|
|
|
|
[true, '1'],
|
|
|
|
[false, '0']
|
|
|
|
];
|
|
|
|
}
|
|
|
|
|
|
|
|
public function testGetStorage() {
|
2017-10-26 14:46:16 +03:00
|
|
|
$return = $this->getMockBuilder(Storage::class)
|
2016-05-12 10:42:19 +03:00
|
|
|
->disableOriginalConstructor()
|
|
|
|
->getMock();
|
|
|
|
|
2015-10-13 18:54:06 +03:00
|
|
|
$path = '/foo/bar.txt';
|
|
|
|
$this->filesMock->expects($this->once())->method('getMount')->with($path)
|
|
|
|
->willReturn($this->mountMock);
|
2016-05-12 10:42:19 +03:00
|
|
|
$this->mountMock->expects($this->once())->method('getStorage')->willReturn($return);
|
2015-10-13 18:54:06 +03:00
|
|
|
|
2016-05-12 10:42:19 +03:00
|
|
|
$this->assertEquals($return, $this->instance->getStorage($path));
|
2015-10-13 18:54:06 +03:00
|
|
|
}
|
|
|
|
|
2015-04-01 20:07:54 +03:00
|
|
|
}
|