155 lines
3.7 KiB
PHP
155 lines
3.7 KiB
PHP
<?php
|
|
|
|
namespace Test\Encryption;
|
|
|
|
use OC\Encryption\Util;
|
|
use Test\TestCase;
|
|
|
|
class UtilTest extends TestCase {
|
|
|
|
/**
|
|
* block size will always be 8192 for a PHP stream
|
|
* @see https://bugs.php.net/bug.php?id=21641
|
|
* @var integer
|
|
*/
|
|
protected $headerSize = 8192;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject */
|
|
protected $view;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject */
|
|
protected $userManager;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject */
|
|
protected $groupManager;
|
|
|
|
/** @var \PHPUnit_Framework_MockObject_MockObject */
|
|
private $config;
|
|
|
|
/** @var \OC\Encryption\Util */
|
|
private $util;
|
|
|
|
public function setUp() {
|
|
parent::setUp();
|
|
$this->view = $this->getMockBuilder('OC\Files\View')
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$this->userManager = $this->getMockBuilder('OC\User\Manager')
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$this->groupManager = $this->getMockBuilder('OC\Group\Manager')
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$this->config = $this->getMockBuilder('OCP\IConfig')
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$this->util = new Util(
|
|
$this->view,
|
|
$this->userManager,
|
|
$this->groupManager,
|
|
$this->config
|
|
);
|
|
|
|
}
|
|
|
|
/**
|
|
* @dataProvider providesHeadersForEncryptionModule
|
|
*/
|
|
public function testGetEncryptionModuleId($expected, $header) {
|
|
$id = $this->util->getEncryptionModuleId($header);
|
|
$this->assertEquals($expected, $id);
|
|
}
|
|
|
|
public function providesHeadersForEncryptionModule() {
|
|
return [
|
|
['', []],
|
|
['', ['1']],
|
|
[2, ['oc_encryption_module' => 2]],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @dataProvider providesHeaders
|
|
*/
|
|
public function testReadHeader($header, $expected, $moduleId) {
|
|
$expected['oc_encryption_module'] = $moduleId;
|
|
$result = $this->util->readHeader($header);
|
|
$this->assertSameSize($expected, $result);
|
|
foreach ($expected as $key => $value) {
|
|
$this->assertArrayHasKey($key, $result);
|
|
$this->assertSame($value, $result[$key]);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @dataProvider providesHeaders
|
|
*/
|
|
public function testCreateHeader($expected, $header, $moduleId) {
|
|
|
|
$em = $this->getMock('\OCP\Encryption\IEncryptionModule');
|
|
$em->expects($this->any())->method('getId')->willReturn($moduleId);
|
|
|
|
$result = $this->util->createHeader($header, $em);
|
|
$this->assertEquals($expected, $result);
|
|
}
|
|
|
|
public function providesHeaders() {
|
|
return [
|
|
[str_pad('HBEGIN:oc_encryption_module:0:HEND', $this->headerSize, '-', STR_PAD_RIGHT)
|
|
, [], '0'],
|
|
[str_pad('HBEGIN:oc_encryption_module:0:custom_header:foo:HEND', $this->headerSize, '-', STR_PAD_RIGHT)
|
|
, ['custom_header' => 'foo'], '0'],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @expectedException \OC\Encryption\Exceptions\EncryptionHeaderKeyExistsException
|
|
*/
|
|
public function testCreateHeaderFailed() {
|
|
|
|
$header = array('header1' => 1, 'header2' => 2, 'oc_encryption_module' => 'foo');
|
|
|
|
$em = $this->getMock('\OCP\Encryption\IEncryptionModule');
|
|
$em->expects($this->any())->method('getId')->willReturn('moduleId');
|
|
|
|
$this->util->createHeader($header, $em);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider providePathsForTestIsExcluded
|
|
*/
|
|
public function testIsExcluded($path, $expected) {
|
|
$this->userManager
|
|
->expects($this->any())
|
|
->method('userExists')
|
|
->will($this->returnCallback(array($this, 'isExcludedCallback')));
|
|
|
|
$this->assertSame($expected,
|
|
$this->util->isExcluded($path)
|
|
);
|
|
}
|
|
|
|
public function providePathsForTestIsExcluded() {
|
|
return array(
|
|
array('files_encryption/foo.txt', true),
|
|
array('test/foo.txt', false),
|
|
array('/user1/files_encryption/foo.txt', true),
|
|
array('/user1/files/foo.txt', false),
|
|
|
|
);
|
|
}
|
|
|
|
public function isExcludedCallback() {
|
|
$args = func_get_args();
|
|
if ($args[0] === 'user1') {
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
}
|