123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120 |
- <?php
- namespace Test\Traits;
- use OC\Encryption\EncryptionWrapper;
- use OC\Files\SetupManager;
- use OC\Memcache\ArrayCache;
- use OCA\Encryption\AppInfo\Application;
- use OCA\Encryption\KeyManager;
- use OCA\Encryption\Users\Setup;
- use OCP\Encryption\IManager;
- use OCP\IUserManager;
- use Psr\Log\LoggerInterface;
- trait EncryptionTrait {
-
- abstract protected function registerStorageWrapper($name, $wrapper);
-
- abstract protected static function markTestSkipped(string $message = ''): void;
- abstract protected static function assertTrue($condition, string $message = ''): void;
- private $encryptionWasEnabled;
- private $originalEncryptionModule;
-
- private $userManager;
-
- private $setupManager;
-
- private $config;
-
- private $encryptionApp;
- protected function loginWithEncryption($user = '') {
- \OC_Util::tearDownFS();
- \OC_User::setUserId('');
-
- \OC::$server->getUserSession()->setUser(null);
- $this->setupManager->tearDown();
- \OC_User::setUserId($user);
- $this->postLogin();
- \OC_Util::setupFS($user);
- if ($this->userManager->userExists($user)) {
- \OC::$server->getUserFolder($user);
- }
- }
- protected function setupForUser($name, $password) {
- $this->setupManager->tearDown();
- $this->setupManager->setupForUser($this->userManager->get($name));
- $container = $this->encryptionApp->getContainer();
-
- $keyManager = $container->query(KeyManager::class);
-
- $userSetup = $container->query(Setup::class);
- $userSetup->setupUser($name, $password);
- $encryptionManager = $container->query(IManager::class);
- $this->encryptionApp->setUp($encryptionManager);
- $keyManager->init($name, $password);
- }
- protected function postLogin() {
- $encryptionWrapper = new EncryptionWrapper(
- new ArrayCache(),
- \OC::$server->getEncryptionManager(),
- \OC::$server->get(LoggerInterface::class)
- );
- $this->registerStorageWrapper('oc_encryption', [$encryptionWrapper, 'wrapStorage']);
- }
- protected function setUpEncryptionTrait() {
- $isReady = \OC::$server->getEncryptionManager()->isReady();
- if (!$isReady) {
- $this->markTestSkipped('Encryption not ready');
- }
- $this->userManager = \OC::$server->get(IUserManager::class);
- $this->setupManager = \OC::$server->get(SetupManager::class);
- \OC_App::loadApp('encryption');
- $this->encryptionApp = new Application([], $isReady);
- $this->config = \OC::$server->getConfig();
- $this->encryptionWasEnabled = $this->config->getAppValue('core', 'encryption_enabled', 'no');
- $this->originalEncryptionModule = $this->config->getAppValue('core', 'default_encryption_module');
- $this->config->setAppValue('core', 'default_encryption_module', \OCA\Encryption\Crypto\Encryption::ID);
- $this->config->setAppValue('core', 'encryption_enabled', 'yes');
- $this->assertTrue(\OC::$server->getEncryptionManager()->isEnabled());
- }
- protected function tearDownEncryptionTrait() {
- if ($this->config) {
- $this->config->setAppValue('core', 'encryption_enabled', $this->encryptionWasEnabled);
- $this->config->setAppValue('core', 'default_encryption_module', $this->originalEncryptionModule);
- $this->config->deleteAppValue('encryption', 'useMasterKey');
- }
- }
- }
|