diff options
author | Joas Schilling <nickvergessen@gmx.de> | 2016-05-20 15:38:20 +0200 |
---|---|---|
committer | Thomas Müller <DeepDiver1975@users.noreply.github.com> | 2016-05-20 15:38:20 +0200 |
commit | 94ad54ec9b96d41a614fbbad4a97b34c41a6901f (patch) | |
tree | f3eb7cdda2704aaf0cd59d58efe66bcbd34cb67d /tests/lib/Traits/EncryptionTrait.php | |
parent | 2ef751b1ec28f7b5c7113af60ec8c9fa0ae1cf87 (diff) | |
download | nextcloud-server-94ad54ec9b96d41a614fbbad4a97b34c41a6901f.tar.gz nextcloud-server-94ad54ec9b96d41a614fbbad4a97b34c41a6901f.zip |
Move tests/ to PSR-4 (#24731)
* Move a-b to PSR-4
* Move c-d to PSR-4
* Move e+g to PSR-4
* Move h-l to PSR-4
* Move m-r to PSR-4
* Move s-u to PSR-4
* Move files/ to PSR-4
* Move remaining tests to PSR-4
* Remove Test\ from old autoloader
Diffstat (limited to 'tests/lib/Traits/EncryptionTrait.php')
-rw-r--r-- | tests/lib/Traits/EncryptionTrait.php | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/tests/lib/Traits/EncryptionTrait.php b/tests/lib/Traits/EncryptionTrait.php new file mode 100644 index 00000000000..5e2ca4e561f --- /dev/null +++ b/tests/lib/Traits/EncryptionTrait.php @@ -0,0 +1,104 @@ +<?php +/** + * Copyright (c) 2015 Robin Appelman <icewind@owncloud.com> + * This file is licensed under the Affero General Public License version 3 or + * later. + * See the COPYING-README file. + */ + +namespace Test\Traits; + +use OC\Encryption\EncryptionWrapper; +use OC\Memcache\ArrayCache; +use OCA\Encryption\AppInfo\Application; +use OCA\Encryption\KeyManager; +use OCA\Encryption\Users\Setup; +use OC\Files\Filesystem; + +/** + * Enables encryption + */ +trait EncryptionTrait { + // from MountProviderTrait + abstract protected function registerStorageWrapper($name, $wrapper); + + // from phpunit + abstract protected function markTestSkipped($reason = ''); + abstract protected function assertTrue($condition, $message = ''); + + private $encryptionWasEnabled; + + private $originalEncryptionModule; + + /** + * @var \OCP\IConfig + */ + private $config; + + /** + * @var \OCA\Encryption\AppInfo\Application + */ + private $encryptionApp; + + protected function loginWithEncryption($user = '') { + \OC_Util::tearDownFS(); + \OC_User::setUserId(''); + // needed for fully logout + \OC::$server->getUserSession()->setUser(null); + + Filesystem::tearDown(); + \OC_User::setUserId($user); + $this->postLogin(); + \OC_Util::setupFS($user); + if (\OC_User::userExists($user)) { + \OC::$server->getUserFolder($user); + } + } + + protected function setupForUser($name, $password) { + \OC_Util::tearDownFS(); + \OC_Util::setupFS($name); + $container = $this->encryptionApp->getContainer(); + /** @var KeyManager $keyManager */ + $keyManager = $container->query('KeyManager'); + /** @var Setup $userSetup */ + $userSetup = $container->query('UserSetup'); + $userSetup->setupUser($name, $password); + $keyManager->init($name, $password); + } + + protected function postLogin() { + $encryptionWrapper = new EncryptionWrapper( + new ArrayCache(), + \OC::$server->getEncryptionManager(), + \OC::$server->getLogger() + ); + + $this->registerStorageWrapper('oc_encryption', array($encryptionWrapper, 'wrapStorage')); + } + + protected function setUpEncryptionTrait() { + $isReady = \OC::$server->getEncryptionManager()->isReady(); + if (!$isReady) { + $this->markTestSkipped('Encryption not ready'); + } + + \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); + } + } +} |