258 lines
9.4 KiB
PHP
258 lines
9.4 KiB
PHP
<?php
|
|
|
|
namespace Test\Encryption;
|
|
|
|
use OC\Encryption\Manager;
|
|
use OC\Encryption\Util;
|
|
use OC\Files\View;
|
|
use OC\Memcache\ArrayCache;
|
|
use OCP\Encryption\IEncryptionModule;
|
|
use OCP\IConfig;
|
|
use OCP\IL10N;
|
|
use OCP\ILogger;
|
|
use Test\TestCase;
|
|
|
|
class ManagerTest extends TestCase {
|
|
|
|
/** @var Manager */
|
|
private $manager;
|
|
|
|
/** @var IConfig|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $config;
|
|
|
|
/** @var ILogger|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $logger;
|
|
|
|
/** @var IL10N|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $l10n;
|
|
|
|
/** @var View|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $view;
|
|
|
|
/** @var Util|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $util;
|
|
|
|
/** @var ArrayCache|\PHPUnit_Framework_MockObject_MockObject */
|
|
private $arrayCache;
|
|
|
|
protected function setUp(): void {
|
|
parent::setUp();
|
|
$this->config = $this->createMock(IConfig::class);
|
|
$this->logger = $this->createMock(ILogger::class);
|
|
$this->l10n = $this->createMock(IL10N::class);
|
|
$this->view = $this->createMock(View::class);
|
|
$this->util = $this->createMock(Util::class);
|
|
$this->arrayCache = $this->createMock(ArrayCache::class);
|
|
$this->manager = new Manager($this->config, $this->logger, $this->l10n, $this->view, $this->util, $this->arrayCache);
|
|
}
|
|
|
|
public function testManagerIsDisabled() {
|
|
$this->assertFalse($this->manager->isEnabled());
|
|
}
|
|
|
|
public function testManagerIsDisabledIfEnabledButNoModules() {
|
|
$this->config->expects($this->any())->method('getAppValue')->willReturn(true);
|
|
$this->assertFalse($this->manager->isEnabled());
|
|
}
|
|
|
|
public function testManagerIsDisabledIfDisabledButModules() {
|
|
$this->config->expects($this->any())->method('getAppValue')->willReturn(false);
|
|
$em = $this->createMock(IEncryptionModule::class);
|
|
$em->expects($this->any())->method('getId')->willReturn('id');
|
|
$em->expects($this->any())->method('getDisplayName')->willReturn('TestDummyModule0');
|
|
$this->manager->registerEncryptionModule('id', 'TestDummyModule0', function() use ($em) {return $em;});
|
|
$this->assertFalse($this->manager->isEnabled());
|
|
}
|
|
|
|
public function testManagerIsEnabled() {
|
|
$this->config->expects($this->any())->method('getSystemValue')->willReturn(true);
|
|
$this->config->expects($this->any())->method('getAppValue')->willReturn('yes');
|
|
$this->assertTrue($this->manager->isEnabled());
|
|
}
|
|
|
|
public function testModuleRegistration() {
|
|
$this->config->expects($this->any())->method('getAppValue')->willReturn('yes');
|
|
|
|
$this->addNewEncryptionModule($this->manager, 0);
|
|
$this->assertCount(1, $this->manager->getEncryptionModules());
|
|
|
|
return $this->manager;
|
|
}
|
|
|
|
/**
|
|
* @depends testModuleRegistration
|
|
*/
|
|
public function testModuleReRegistration($manager) {
|
|
$this->expectException(\OC\Encryption\Exceptions\ModuleAlreadyExistsException::class);
|
|
$this->expectExceptionMessage('Id "ID0" already used by encryption module "TestDummyModule0"');
|
|
|
|
$this->addNewEncryptionModule($manager, 0);
|
|
}
|
|
|
|
public function testModuleUnRegistration() {
|
|
$this->config->expects($this->any())->method('getAppValue')->willReturn(true);
|
|
$this->addNewEncryptionModule($this->manager, 0);
|
|
$this->assertCount(1, $this->manager->getEncryptionModules());
|
|
|
|
$this->manager->unregisterEncryptionModule('ID0');
|
|
$this->assertEmpty($this->manager->getEncryptionModules());
|
|
|
|
}
|
|
|
|
|
|
public function testGetEncryptionModuleUnknown() {
|
|
$this->expectException(\OC\Encryption\Exceptions\ModuleDoesNotExistsException::class);
|
|
$this->expectExceptionMessage('Module with ID: unknown does not exist.');
|
|
|
|
$this->config->expects($this->any())->method('getAppValue')->willReturn(true);
|
|
$this->addNewEncryptionModule($this->manager, 0);
|
|
$this->assertCount(1, $this->manager->getEncryptionModules());
|
|
$this->manager->getEncryptionModule('unknown');
|
|
}
|
|
|
|
public function testGetEncryptionModuleEmpty() {
|
|
global $defaultId;
|
|
$defaultId = null;
|
|
|
|
$this->config->expects($this->any())
|
|
->method('getAppValue')
|
|
->with('core', 'default_encryption_module')
|
|
->willReturnCallback(function() { global $defaultId; return $defaultId; });
|
|
|
|
$this->addNewEncryptionModule($this->manager, 0);
|
|
$this->assertCount(1, $this->manager->getEncryptionModules());
|
|
$this->addNewEncryptionModule($this->manager, 1);
|
|
$this->assertCount(2, $this->manager->getEncryptionModules());
|
|
|
|
// Should return the default module
|
|
$defaultId = 'ID0';
|
|
$this->assertEquals('ID0', $this->manager->getEncryptionModule()->getId());
|
|
$defaultId = 'ID1';
|
|
$this->assertEquals('ID1', $this->manager->getEncryptionModule()->getId());
|
|
}
|
|
|
|
public function testGetEncryptionModule() {
|
|
global $defaultId;
|
|
$defaultId = null;
|
|
|
|
$this->config->expects($this->any())
|
|
->method('getAppValue')
|
|
->with('core', 'default_encryption_module')
|
|
->willReturnCallback(function() { global $defaultId; return $defaultId; });
|
|
|
|
$this->addNewEncryptionModule($this->manager, 0);
|
|
$defaultId = 'ID0';
|
|
$this->assertCount(1, $this->manager->getEncryptionModules());
|
|
|
|
$en0 = $this->manager->getEncryptionModule('ID0');
|
|
$this->assertEquals('ID0', $en0->getId());
|
|
|
|
$en0 = self::invokePrivate($this->manager, 'getDefaultEncryptionModule');
|
|
$this->assertEquals('ID0', $en0->getId());
|
|
|
|
$this->assertEquals('ID0', $this->manager->getDefaultEncryptionModuleId());
|
|
}
|
|
|
|
public function testSetDefaultEncryptionModule() {
|
|
global $defaultId;
|
|
$defaultId = null;
|
|
|
|
$this->config->expects($this->any())
|
|
->method('getAppValue')
|
|
->with('core', 'default_encryption_module')
|
|
->willReturnCallback(function() { global $defaultId; return $defaultId; });
|
|
|
|
$this->addNewEncryptionModule($this->manager, 0);
|
|
$this->assertCount(1, $this->manager->getEncryptionModules());
|
|
$this->addNewEncryptionModule($this->manager, 1);
|
|
$this->assertCount(2, $this->manager->getEncryptionModules());
|
|
|
|
// Default module is the first we set
|
|
$defaultId = 'ID0';
|
|
$this->assertEquals('ID0', $this->manager->getDefaultEncryptionModuleId());
|
|
|
|
// Set to an existing module
|
|
$this->config->expects($this->once())
|
|
->method('setAppValue')
|
|
->with('core', 'default_encryption_module', 'ID1');
|
|
$this->assertTrue($this->manager->setDefaultEncryptionModule('ID1'));
|
|
$defaultId = 'ID1';
|
|
$this->assertEquals('ID1', $this->manager->getDefaultEncryptionModuleId());
|
|
|
|
// Set to an unexisting module
|
|
$this->assertFalse($this->manager->setDefaultEncryptionModule('ID2'));
|
|
$this->assertEquals('ID1', $this->manager->getDefaultEncryptionModuleId());
|
|
}
|
|
|
|
// /**
|
|
// * @expectedException \OC\Encryption\Exceptions\ModuleAlreadyExistsException
|
|
// * @expectedExceptionMessage Id "0" already used by encryption module "TestDummyModule0"
|
|
// */
|
|
// public function testModuleRegistration() {
|
|
// $config = $this->createMock(IConfig::class);
|
|
// $config->expects($this->any())->method('getSystemValue')->willReturn(true);
|
|
// $em = $this->createMock(IEncryptionModule::class);
|
|
// $em->expects($this->any())->method('getId')->willReturn(0);
|
|
// $em->expects($this->any())->method('getDisplayName')->willReturn('TestDummyModule0');
|
|
// $m = new Manager($config);
|
|
// $m->registerEncryptionModule($em);
|
|
// $this->assertTrue($m->isEnabled());
|
|
// $m->registerEncryptionModule($em);
|
|
// }
|
|
//
|
|
// public function testModuleUnRegistration() {
|
|
// $config = $this->createMock(IConfig::class);
|
|
// $config->expects($this->any())->method('getSystemValue')->willReturn(true);
|
|
// $em = $this->createMock(IEncryptionModule::class);
|
|
// $em->expects($this->any())->method('getId')->willReturn(0);
|
|
// $em->expects($this->any())->method('getDisplayName')->willReturn('TestDummyModule0');
|
|
// $m = new Manager($config);
|
|
// $m->registerEncryptionModule($em);
|
|
// $this->assertTrue($m->isEnabled());
|
|
// $m->unregisterEncryptionModule($em);
|
|
// $this->assertFalse($m->isEnabled());
|
|
// }
|
|
//
|
|
// /**
|
|
// * @expectedException \OC\Encryption\Exceptions\ModuleDoesNotExistsException
|
|
// * @expectedExceptionMessage Module with ID: unknown does not exist.
|
|
// */
|
|
// public function testGetEncryptionModuleUnknown() {
|
|
// $config = $this->createMock(IConfig::class);
|
|
// $config->expects($this->any())->method('getSystemValue')->willReturn(true);
|
|
// $em = $this->createMock(IEncryptionModule::class);
|
|
// $em->expects($this->any())->method('getId')->willReturn(0);
|
|
// $em->expects($this->any())->method('getDisplayName')->willReturn('TestDummyModule0');
|
|
// $m = new Manager($config);
|
|
// $m->registerEncryptionModule($em);
|
|
// $this->assertTrue($m->isEnabled());
|
|
// $m->getEncryptionModule('unknown');
|
|
// }
|
|
//
|
|
// public function testGetEncryptionModule() {
|
|
// $config = $this->createMock(IConfig::class);
|
|
// $config->expects($this->any())->method('getSystemValue')->willReturn(true);
|
|
// $em = $this->createMock(IEncryptionModule::class);
|
|
// $em->expects($this->any())->method('getId')->willReturn(0);
|
|
// $em->expects($this->any())->method('getDisplayName')->willReturn('TestDummyModule0');
|
|
// $m = new Manager($config);
|
|
// $m->registerEncryptionModule($em);
|
|
// $this->assertTrue($m->isEnabled());
|
|
// $en0 = $m->getEncryptionModule(0);
|
|
// $this->assertEquals(0, $en0->getId());
|
|
// }
|
|
|
|
protected function addNewEncryptionModule(Manager $manager, $id) {
|
|
$encryptionModule = $this->createMock(IEncryptionModule::class);
|
|
$encryptionModule->expects($this->any())
|
|
->method('getId')
|
|
->willReturn('ID' . $id);
|
|
$encryptionModule->expects($this->any())
|
|
->method('getDisplayName')
|
|
->willReturn('TestDummyModule' . $id);
|
|
/** @var \OCP\Encryption\IEncryptionModule $encryptionModule */
|
|
$manager->registerEncryptionModule('ID' . $id, 'TestDummyModule' . $id, function() use ($encryptionModule) {
|
|
return $encryptionModule;
|
|
});
|
|
}
|
|
}
|