382 lines
12 KiB
PHP
382 lines
12 KiB
PHP
|
<?php
|
||
|
|
/**
|
||
|
|
* @author Björn Schießle <schiessle@owncloud.com>
|
||
|
|
*
|
||
|
|
* @copyright Copyright (c) 2015, 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 Tests\Core\Command\Encryption;
|
||
|
|
|
||
|
|
|
||
|
|
use OC\Core\Command\Encryption\ChangeKeyStorageRoot;
|
||
|
|
use OC\Encryption\Util;
|
||
|
|
use OC\Files\View;
|
||
|
|
use OCP\IConfig;
|
||
|
|
use OCP\IUserManager;
|
||
|
|
use Symfony\Component\Console\Helper\QuestionHelper;
|
||
|
|
use Symfony\Component\Console\Input\InputInterface;
|
||
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
||
|
|
use Test\TestCase;
|
||
|
|
|
||
|
|
class ChangeKeyStorageRootTest extends TestCase {
|
||
|
|
|
||
|
|
/** @var ChangeKeyStorageRoot */
|
||
|
|
protected $changeKeyStorageRoot;
|
||
|
|
|
||
|
|
/** @var View | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $view;
|
||
|
|
|
||
|
|
/** @var IUserManager | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $userManager;
|
||
|
|
|
||
|
|
/** @var IConfig | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $config;
|
||
|
|
|
||
|
|
/** @var Util | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $util;
|
||
|
|
|
||
|
|
/** @var QuestionHelper | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $questionHelper;
|
||
|
|
|
||
|
|
/** @var InputInterface | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $inputInterface;
|
||
|
|
|
||
|
|
/** @var OutputInterface | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $outputInterface;
|
||
|
|
|
||
|
|
/** @var \OCP\UserInterface | \PHPUnit_Framework_MockObject_MockObject */
|
||
|
|
protected $userInterface;
|
||
|
|
|
||
|
|
public function setUp() {
|
||
|
|
parent::setUp();
|
||
|
|
|
||
|
|
$this->view = $this->getMock('\OC\Files\View');
|
||
|
|
$this->userManager = $this->getMock('\OCP\IUserManager');
|
||
|
|
$this->config = $this->getMock('\OCP\IConfig');
|
||
|
|
$this->util = $this->getMockBuilder('OC\Encryption\Util')->disableOriginalConstructor()->getMock();
|
||
|
|
$this->questionHelper = $this->getMock('Symfony\Component\Console\Helper\QuestionHelper');
|
||
|
|
$this->inputInterface = $this->getMock('Symfony\Component\Console\Input\InputInterface');
|
||
|
|
$this->outputInterface = $this->getMock('Symfony\Component\Console\Output\OutputInterface');
|
||
|
|
$this->userInterface = $this->getMock('\OCP\UserInterface');
|
||
|
|
|
||
|
|
$outputFormaterInterface = $this->getMock('Symfony\Component\Console\Formatter\OutputFormatterInterface');
|
||
|
|
$this->outputInterface->expects($this->any())->method('getFormatter')
|
||
|
|
->willReturn($outputFormaterInterface);
|
||
|
|
|
||
|
|
$this->changeKeyStorageRoot = new ChangeKeyStorageRoot(
|
||
|
|
$this->view,
|
||
|
|
$this->userManager,
|
||
|
|
$this->config,
|
||
|
|
$this->util,
|
||
|
|
$this->questionHelper
|
||
|
|
);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dataProvider dataTestExecute
|
||
|
|
*/
|
||
|
|
public function testExecute($newRoot, $answer, $successMoveKey) {
|
||
|
|
|
||
|
|
$changeKeyStorageRoot = $this->getMockBuilder('OC\Core\Command\Encryption\ChangeKeyStorageRoot')
|
||
|
|
->setConstructorArgs(
|
||
|
|
[
|
||
|
|
$this->view,
|
||
|
|
$this->userManager,
|
||
|
|
$this->config,
|
||
|
|
$this->util,
|
||
|
|
$this->questionHelper
|
||
|
|
]
|
||
|
|
)->setMethods(['moveAllKeys'])->getMock();
|
||
|
|
|
||
|
|
$this->util->expects($this->once())->method('getKeyStorageRoot')
|
||
|
|
->willReturn('');
|
||
|
|
$this->inputInterface->expects($this->once())->method('getArgument')
|
||
|
|
->with('newRoot')->willReturn($newRoot);
|
||
|
|
|
||
|
|
if ($answer === true || $newRoot !== null) {
|
||
|
|
$changeKeyStorageRoot->expects($this->once())->method('moveAllKeys')
|
||
|
|
->willReturn($successMoveKey);
|
||
|
|
} else {
|
||
|
|
$changeKeyStorageRoot->expects($this->never())->method('moveAllKeys');
|
||
|
|
}
|
||
|
|
|
||
|
|
if ($successMoveKey === true) {
|
||
|
|
$this->util->expects($this->once())->method('setKeyStorageRoot');
|
||
|
|
} else {
|
||
|
|
$this->util->expects($this->never())->method('setKeyStorageRoot');
|
||
|
|
}
|
||
|
|
|
||
|
|
if ($newRoot === null) {
|
||
|
|
$this->questionHelper->expects($this->once())->method('ask')->willReturn($answer);
|
||
|
|
} else {
|
||
|
|
$this->questionHelper->expects($this->never())->method('ask');
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->invokePrivate(
|
||
|
|
$changeKeyStorageRoot,
|
||
|
|
'execute',
|
||
|
|
[$this->inputInterface, $this->outputInterface]
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function dataTestExecute() {
|
||
|
|
return [
|
||
|
|
[null, true, true],
|
||
|
|
[null, true, false],
|
||
|
|
[null, false, null],
|
||
|
|
['/newRoot', null, true],
|
||
|
|
['/newRoot', null, false]
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testMoveAllKeys() {
|
||
|
|
|
||
|
|
/** @var \OC\Core\Command\Encryption\ChangeKeyStorageRoot $changeKeyStorageRoot */
|
||
|
|
$changeKeyStorageRoot = $this->getMockBuilder('OC\Core\Command\Encryption\ChangeKeyStorageRoot')
|
||
|
|
->setConstructorArgs(
|
||
|
|
[
|
||
|
|
$this->view,
|
||
|
|
$this->userManager,
|
||
|
|
$this->config,
|
||
|
|
$this->util,
|
||
|
|
$this->questionHelper
|
||
|
|
]
|
||
|
|
)->setMethods(['prepareNewRoot', 'moveSystemKeys', 'moveUserKeys'])->getMock();
|
||
|
|
|
||
|
|
$changeKeyStorageRoot->expects($this->at(0))->method('prepareNewRoot')->with('newRoot');
|
||
|
|
$changeKeyStorageRoot->expects($this->at(1))->method('moveSystemKeys')->with('oldRoot', 'newRoot');
|
||
|
|
$changeKeyStorageRoot->expects($this->at(2))->method('moveUserKeys')->with('oldRoot', 'newRoot', $this->outputInterface);
|
||
|
|
|
||
|
|
$this->invokePrivate($changeKeyStorageRoot, 'moveAllKeys', ['oldRoot', 'newRoot', $this->outputInterface]);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testPrepareNewRoot() {
|
||
|
|
$this->view->expects($this->once())->method('is_dir')->with('newRoot')
|
||
|
|
->willReturn(true);
|
||
|
|
|
||
|
|
$this->view->expects($this->once())->method('file_put_contents')
|
||
|
|
->with('newRoot/' . \OC\Encryption\Keys\Storage::KEY_STORAGE_MARKER,
|
||
|
|
'ownCloud will detect this folder as key storage root only if this file exists');
|
||
|
|
|
||
|
|
$this->invokePrivate($this->changeKeyStorageRoot, 'prepareNewRoot', ['newRoot']);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dataProvider dataTestPrepareNewRootException
|
||
|
|
* @expectedException \Exception
|
||
|
|
*
|
||
|
|
* @param bool $dirExists
|
||
|
|
* @param bool $couldCreateFile
|
||
|
|
*/
|
||
|
|
public function testPrepareNewRootException($dirExists, $couldCreateFile) {
|
||
|
|
$this->view->expects($this->once())->method('is_dir')->with('newRoot')
|
||
|
|
->willReturn($dirExists);
|
||
|
|
$this->view->expects($this->any())->method('file_put_contents')->willReturn($couldCreateFile);
|
||
|
|
|
||
|
|
$this->invokePrivate($this->changeKeyStorageRoot, 'prepareNewRoot', ['newRoot']);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function dataTestPrepareNewRootException() {
|
||
|
|
return [
|
||
|
|
[true, false],
|
||
|
|
[false, true]
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dataProvider dataTestMoveSystemKeys
|
||
|
|
*
|
||
|
|
* @param bool $dirExists
|
||
|
|
* @param bool $targetExists
|
||
|
|
* @param bool $executeRename
|
||
|
|
*/
|
||
|
|
public function testMoveSystemKeys($dirExists, $targetExists, $executeRename) {
|
||
|
|
|
||
|
|
$changeKeyStorageRoot = $this->getMockBuilder('OC\Core\Command\Encryption\ChangeKeyStorageRoot')
|
||
|
|
->setConstructorArgs(
|
||
|
|
[
|
||
|
|
$this->view,
|
||
|
|
$this->userManager,
|
||
|
|
$this->config,
|
||
|
|
$this->util,
|
||
|
|
$this->questionHelper
|
||
|
|
]
|
||
|
|
)->setMethods(['targetExists'])->getMock();
|
||
|
|
|
||
|
|
$this->view->expects($this->once())->method('is_dir')
|
||
|
|
->with('oldRoot/files_encryption')->willReturn($dirExists);
|
||
|
|
$changeKeyStorageRoot->expects($this->any())->method('targetExists')
|
||
|
|
->with('newRoot/files_encryption')->willReturn($targetExists);
|
||
|
|
|
||
|
|
if ($executeRename) {
|
||
|
|
$this->view->expects($this->once())->method('rename')
|
||
|
|
->with('oldRoot/files_encryption', 'newRoot/files_encryption');
|
||
|
|
} else {
|
||
|
|
$this->view->expects($this->never())->method('rename');
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->invokePrivate($changeKeyStorageRoot, 'moveSystemKeys', ['oldRoot', 'newRoot']);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
public function dataTestMoveSystemKeys() {
|
||
|
|
return [
|
||
|
|
[true, false, true],
|
||
|
|
[false, true, false],
|
||
|
|
[true, true, false],
|
||
|
|
[false, false, false]
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
public function testMoveUserKeys() {
|
||
|
|
|
||
|
|
$changeKeyStorageRoot = $this->getMockBuilder('OC\Core\Command\Encryption\ChangeKeyStorageRoot')
|
||
|
|
->setConstructorArgs(
|
||
|
|
[
|
||
|
|
$this->view,
|
||
|
|
$this->userManager,
|
||
|
|
$this->config,
|
||
|
|
$this->util,
|
||
|
|
$this->questionHelper
|
||
|
|
]
|
||
|
|
)->setMethods(['setupUserFS', 'moveUserEncryptionFolder'])->getMock();
|
||
|
|
|
||
|
|
$this->userManager->expects($this->once())->method('getBackends')
|
||
|
|
->willReturn([$this->userInterface]);
|
||
|
|
$this->userInterface->expects($this->once())->method('getUsers')
|
||
|
|
->willReturn(['user1', 'user2']);
|
||
|
|
$changeKeyStorageRoot->expects($this->exactly(2))->method('setupUserFS');
|
||
|
|
$changeKeyStorageRoot->expects($this->exactly(2))->method('moveUserEncryptionFolder');
|
||
|
|
|
||
|
|
$this->invokePrivate($changeKeyStorageRoot, 'moveUserKeys', ['oldRoot', 'newRoot', $this->outputInterface]);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dataProvider dataTestMoveUserEncryptionFolder
|
||
|
|
*
|
||
|
|
* @param bool $userExists
|
||
|
|
* @param bool $isDir
|
||
|
|
* @param bool $targetExists
|
||
|
|
* @param bool $shouldRename
|
||
|
|
*/
|
||
|
|
public function testMoveUserEncryptionFolder($userExists, $isDir, $targetExists, $shouldRename) {
|
||
|
|
|
||
|
|
$changeKeyStorageRoot = $this->getMockBuilder('OC\Core\Command\Encryption\ChangeKeyStorageRoot')
|
||
|
|
->setConstructorArgs(
|
||
|
|
[
|
||
|
|
$this->view,
|
||
|
|
$this->userManager,
|
||
|
|
$this->config,
|
||
|
|
$this->util,
|
||
|
|
$this->questionHelper
|
||
|
|
]
|
||
|
|
)->setMethods(['targetExists', 'prepareParentFolder'])->getMock();
|
||
|
|
|
||
|
|
$this->userManager->expects($this->once())->method('userExists')
|
||
|
|
->willReturn($userExists);
|
||
|
|
$this->view->expects($this->any())->method('is_dir')
|
||
|
|
->willReturn($isDir);
|
||
|
|
$changeKeyStorageRoot->expects($this->any())->method('targetExists')
|
||
|
|
->willReturn($targetExists);
|
||
|
|
|
||
|
|
if ($shouldRename) {
|
||
|
|
$changeKeyStorageRoot->expects($this->once())->method('prepareParentFolder')
|
||
|
|
->with('newRoot/user1');
|
||
|
|
$this->view->expects($this->once())->method('rename')
|
||
|
|
->with('oldRoot/user1/files_encryption', 'newRoot/user1/files_encryption');
|
||
|
|
} else {
|
||
|
|
$changeKeyStorageRoot->expects($this->never())->method('prepareParentFolder');
|
||
|
|
$this->view->expects($this->never())->method('rename');
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->invokePrivate($changeKeyStorageRoot, 'moveUserEncryptionFolder', ['user1', 'oldRoot', 'newRoot']);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
public function dataTestMoveUserEncryptionFolder() {
|
||
|
|
return [
|
||
|
|
[true, true, false, true],
|
||
|
|
[true, false, true, false],
|
||
|
|
[false, true, true, false],
|
||
|
|
[false, false, true, false],
|
||
|
|
[false, true, false, false],
|
||
|
|
[false, true, true, false],
|
||
|
|
[false, false, false, false]
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @dataProvider dataTestPrepareParentFolder
|
||
|
|
*/
|
||
|
|
public function testPrepareParentFolder($path, $pathExists) {
|
||
|
|
$this->view->expects($this->any())->method('file_exists')
|
||
|
|
->willReturnCallback(
|
||
|
|
function($fileExistsPath) use ($path, $pathExists) {
|
||
|
|
if ($path === $fileExistsPath) {
|
||
|
|
return $pathExists;
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
);
|
||
|
|
|
||
|
|
if ($pathExists === false) {
|
||
|
|
$subDirs = explode('/', ltrim($path, '/'));
|
||
|
|
$this->view->expects($this->exactly(count($subDirs)))->method('mkdir');
|
||
|
|
} else {
|
||
|
|
$this->view->expects($this->never())->method('mkdir');
|
||
|
|
}
|
||
|
|
|
||
|
|
$this->invokePrivate(
|
||
|
|
$this->changeKeyStorageRoot,
|
||
|
|
'prepareParentFolder',
|
||
|
|
[$path]
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
public function dataTestPrepareParentFolder() {
|
||
|
|
return [
|
||
|
|
['/user/folder/sub_folder/keystorage', true],
|
||
|
|
['/user/folder/sub_folder/keystorage', false]
|
||
|
|
];
|
||
|
|
}
|
||
|
|
|
||
|
|
public function testTargetExists() {
|
||
|
|
$this->view->expects($this->once())->method('file_exists')->with('path')
|
||
|
|
->willReturn(false);
|
||
|
|
|
||
|
|
$this->assertFalse(
|
||
|
|
$this->invokePrivate($this->changeKeyStorageRoot, 'targetExists', ['path'])
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @expectedException \Exception
|
||
|
|
*/
|
||
|
|
public function testTargetExistsException() {
|
||
|
|
$this->view->expects($this->once())->method('file_exists')->with('path')
|
||
|
|
->willReturn(true);
|
||
|
|
|
||
|
|
$this->invokePrivate($this->changeKeyStorageRoot, 'targetExists', ['path']);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|