120 lines
3.9 KiB
PHP
120 lines
3.9 KiB
PHP
<?php
|
|
|
|
namespace Test\Files\Storage\Wrapper;
|
|
|
|
use OC\Files\Storage\Temporary;
|
|
use OC\Files\View;
|
|
|
|
class Encryption extends \Test\Files\Storage\Storage {
|
|
|
|
/**
|
|
* @var Temporary
|
|
*/
|
|
private $sourceStorage;
|
|
|
|
public function setUp() {
|
|
|
|
parent::setUp();
|
|
|
|
$mockModule = $this->buildMockModule();
|
|
$encryptionManager = $this->getMockBuilder('\OC\Encryption\Manager')
|
|
->disableOriginalConstructor()
|
|
->setMethods(['getDefaultEncryptionModule', 'getEncryptionModule', 'isEnabled'])
|
|
->getMock();
|
|
$encryptionManager->expects($this->any())
|
|
->method('getDefaultEncryptionModule')
|
|
->willReturn($mockModule);
|
|
$encryptionManager->expects($this->any())
|
|
->method('getEncryptionModule')
|
|
->willReturn($mockModule);
|
|
$encryptionManager->expects($this->any())
|
|
->method('isEnabled')
|
|
->willReturn(true);
|
|
|
|
$config = $this->getMockBuilder('\OCP\IConfig')
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$util = $this->getMock('\OC\Encryption\Util', ['getUidAndFilename'], [new View(), new \OC\User\Manager(), $config]);
|
|
$util->expects($this->any())
|
|
->method('getUidAndFilename')
|
|
->willReturnCallback(function ($path) {
|
|
return ['user1', $path];
|
|
});
|
|
|
|
$file = $this->getMockBuilder('\OC\Encryption\File')
|
|
->disableOriginalConstructor()
|
|
->getMock();
|
|
|
|
$logger = $this->getMock('\OC\Log');
|
|
|
|
$this->sourceStorage = new Temporary(array());
|
|
$keyStore = $this->getMockBuilder('\OC\Encryption\Keys\Storage')
|
|
->disableOriginalConstructor()->getMock();
|
|
$this->instance = new EncryptionWrapper([
|
|
'storage' => $this->sourceStorage,
|
|
'root' => 'foo',
|
|
'mountPoint' => '/'
|
|
],
|
|
$encryptionManager, $util, $logger, $file, null, $keyStore
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @return \PHPUnit_Framework_MockObject_MockObject
|
|
*/
|
|
protected function buildMockModule() {
|
|
$encryptionModule = $this->getMockBuilder('\OCP\Encryption\IEncryptionModule')
|
|
->disableOriginalConstructor()
|
|
->setMethods(['getId', 'getDisplayName', 'begin', 'end', 'encrypt', 'decrypt', 'update', 'shouldEncrypt', 'calculateUnencryptedSize', 'getUnencryptedBlockSize'])
|
|
->getMock();
|
|
|
|
$encryptionModule->expects($this->any())->method('getId')->willReturn('UNIT_TEST_MODULE');
|
|
$encryptionModule->expects($this->any())->method('getDisplayName')->willReturn('Unit test module');
|
|
$encryptionModule->expects($this->any())->method('begin')->willReturn([]);
|
|
$encryptionModule->expects($this->any())->method('end')->willReturn('');
|
|
$encryptionModule->expects($this->any())->method('encrypt')->willReturnArgument(0);
|
|
$encryptionModule->expects($this->any())->method('decrypt')->willReturnArgument(0);
|
|
$encryptionModule->expects($this->any())->method('update')->willReturn(true);
|
|
$encryptionModule->expects($this->any())->method('shouldEncrypt')->willReturn(true);
|
|
$encryptionModule->expects($this->any())->method('calculateUnencryptedSize')->willReturn(42);
|
|
$encryptionModule->expects($this->any())->method('getUnencryptedBlockSize')->willReturn(6126);
|
|
return $encryptionModule;
|
|
}
|
|
|
|
// public function testMkDirRooted() {
|
|
// $this->instance->mkdir('bar');
|
|
// $this->assertTrue($this->sourceStorage->is_dir('foo/bar'));
|
|
// }
|
|
//
|
|
// public function testFilePutContentsRooted() {
|
|
// $this->instance->file_put_contents('bar', 'asd');
|
|
// $this->assertEquals('asd', $this->sourceStorage->file_get_contents('foo/bar'));
|
|
// }
|
|
}
|
|
|
|
//
|
|
// FIXME: this is too bad and needs adjustment
|
|
//
|
|
class EncryptionWrapper extends \OC\Files\Storage\Wrapper\Encryption {
|
|
private $keyStore;
|
|
|
|
public function __construct(
|
|
$parameters,
|
|
\OC\Encryption\Manager $encryptionManager = null,
|
|
\OC\Encryption\Util $util = null,
|
|
\OC\Log $logger = null,
|
|
\OC\Encryption\File $fileHelper = null,
|
|
$uid = null,
|
|
$keyStore = null
|
|
) {
|
|
$this->keyStore = $keyStore;
|
|
parent::__construct($parameters, $encryptionManager, $util, $logger, $fileHelper, $uid);
|
|
}
|
|
|
|
protected function getKeyStorage($encryptionModuleId) {
|
|
return $this->keyStore;
|
|
}
|
|
|
|
}
|