diff options
Diffstat (limited to 'apps/encryption/tests/lib/crypto/encryptionTest.php')
| -rw-r--r-- | apps/encryption/tests/lib/crypto/encryptionTest.php | 425 |
1 files changed, 0 insertions, 425 deletions
diff --git a/apps/encryption/tests/lib/crypto/encryptionTest.php b/apps/encryption/tests/lib/crypto/encryptionTest.php deleted file mode 100644 index 8a228c2c215..00000000000 --- a/apps/encryption/tests/lib/crypto/encryptionTest.php +++ /dev/null @@ -1,425 +0,0 @@ -<?php -/** - * @author Björn Schießle <schiessle@owncloud.com> - * @author Joas Schilling <nickvergessen@owncloud.com> - * - * @copyright Copyright (c) 2016, ownCloud, Inc. - * @license AGPL-3.0 - * - * This code is free software: you can redistribute it and/or modify - * it under the terms of the GNU Affero General Public License, version 3, - * as published by the Free Software Foundation. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU Affero General Public License for more details. - * - * You should have received a copy of the GNU Affero General Public License, version 3, - * along with this program. If not, see <http://www.gnu.org/licenses/> - * - */ - -namespace OCA\Encryption\Tests\lib\Crypto; - -use OCA\Encryption\Exceptions\PublicKeyMissingException; -use Test\TestCase; -use OCA\Encryption\Crypto\Encryption; - -class EncryptionTest extends TestCase { - - /** @var Encryption */ - private $instance; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $keyManagerMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $encryptAllMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $decryptAllMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $sessionMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $cryptMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $utilMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $loggerMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $l10nMock; - - /** @var \PHPUnit_Framework_MockObject_MockObject */ - private $storageMock; - - public function setUp() { - parent::setUp(); - - $this->storageMock = $this->getMockBuilder('OCP\Files\Storage') - ->disableOriginalConstructor()->getMock(); - $this->cryptMock = $this->getMockBuilder('OCA\Encryption\Crypto\Crypt') - ->disableOriginalConstructor() - ->getMock(); - $this->utilMock = $this->getMockBuilder('OCA\Encryption\Util') - ->disableOriginalConstructor() - ->getMock(); - $this->keyManagerMock = $this->getMockBuilder('OCA\Encryption\KeyManager') - ->disableOriginalConstructor() - ->getMock(); - $this->sessionMock = $this->getMockBuilder('OCA\Encryption\Session') - ->disableOriginalConstructor() - ->getMock(); - $this->encryptAllMock = $this->getMockBuilder('OCA\Encryption\Crypto\EncryptAll') - ->disableOriginalConstructor() - ->getMock(); - $this->decryptAllMock = $this->getMockBuilder('OCA\Encryption\Crypto\DecryptAll') - ->disableOriginalConstructor() - ->getMock(); - $this->loggerMock = $this->getMockBuilder('OCP\ILogger') - ->disableOriginalConstructor() - ->getMock(); - $this->l10nMock = $this->getMockBuilder('OCP\IL10N') - ->disableOriginalConstructor() - ->getMock(); - $this->l10nMock->expects($this->any()) - ->method('t') - ->with($this->anything()) - ->willReturnArgument(0); - - $this->instance = new Encryption( - $this->cryptMock, - $this->keyManagerMock, - $this->utilMock, - $this->sessionMock, - $this->encryptAllMock, - $this->decryptAllMock, - $this->loggerMock, - $this->l10nMock - ); - - } - - /** - * test if public key from one of the recipients is missing - */ - public function testEndUser1() { - $this->instance->begin('/foo/bar', 'user1', 'r', array(), array('users' => array('user1', 'user2', 'user3'))); - $this->endTest(); - } - - /** - * test if public key from owner is missing - * - * @expectedException \OCA\Encryption\Exceptions\PublicKeyMissingException - */ - public function testEndUser2() { - $this->instance->begin('/foo/bar', 'user2', 'r', array(), array('users' => array('user1', 'user2', 'user3'))); - $this->endTest(); - } - - /** - * common part of testEndUser1 and testEndUser2 - * - * @throws PublicKeyMissingException - */ - public function endTest() { - // prepare internal variables - self::invokePrivate($this->instance, 'isWriteOperation', [true]); - self::invokePrivate($this->instance, 'writeCache', ['']); - - $this->keyManagerMock->expects($this->any()) - ->method('getPublicKey') - ->will($this->returnCallback([$this, 'getPublicKeyCallback'])); - $this->keyManagerMock->expects($this->any()) - ->method('addSystemKeys') - ->will($this->returnCallback([$this, 'addSystemKeysCallback'])); - $this->cryptMock->expects($this->any()) - ->method('multiKeyEncrypt') - ->willReturn(true); - $this->cryptMock->expects($this->any()) - ->method('setAllFileKeys') - ->willReturn(true); - - $this->instance->end('/foo/bar'); - } - - - public function getPublicKeyCallback($uid) { - if ($uid === 'user2') { - throw new PublicKeyMissingException($uid); - } - return $uid; - } - - public function addSystemKeysCallback($accessList, $publicKeys) { - $this->assertSame(2, count($publicKeys)); - $this->assertArrayHasKey('user1', $publicKeys); - $this->assertArrayHasKey('user3', $publicKeys); - return $publicKeys; - } - - /** - * @dataProvider dataProviderForTestGetPathToRealFile - */ - public function testGetPathToRealFile($path, $expected) { - $this->assertSame($expected, - self::invokePrivate($this->instance, 'getPathToRealFile', array($path)) - ); - } - - public function dataProviderForTestGetPathToRealFile() { - return array( - array('/user/files/foo/bar.txt', '/user/files/foo/bar.txt'), - array('/user/files/foo.txt', '/user/files/foo.txt'), - array('/user/files_versions/foo.txt.v543534', '/user/files/foo.txt'), - array('/user/files_versions/foo/bar.txt.v5454', '/user/files/foo/bar.txt'), - ); - } - - /** - * @dataProvider dataTestBegin - */ - public function testBegin($mode, $header, $legacyCipher, $defaultCipher, $fileKey, $expected) { - - $this->sessionMock->expects($this->once()) - ->method('decryptAllModeActivated') - ->willReturn(false); - - $this->sessionMock->expects($this->never())->method('getDecryptAllUid'); - $this->sessionMock->expects($this->never())->method('getDecryptAllKey'); - $this->keyManagerMock->expects($this->never())->method('getEncryptedFileKey'); - $this->keyManagerMock->expects($this->never())->method('getShareKey'); - $this->cryptMock->expects($this->never())->method('multiKeyDecrypt'); - - $this->cryptMock->expects($this->any()) - ->method('getCipher') - ->willReturn($defaultCipher); - $this->cryptMock->expects($this->any()) - ->method('getLegacyCipher') - ->willReturn($legacyCipher); - if (empty($fileKey)) { - $this->cryptMock->expects($this->once()) - ->method('generateFileKey') - ->willReturn('fileKey'); - } else { - $this->cryptMock->expects($this->never()) - ->method('generateFileKey'); - } - - $this->keyManagerMock->expects($this->once()) - ->method('getFileKey') - ->willReturn($fileKey); - - $result = $this->instance->begin('/user/files/foo.txt', 'user', $mode, $header, []); - - $this->assertArrayHasKey('cipher', $result); - $this->assertSame($expected, $result['cipher']); - if ($mode === 'w') { - $this->assertTrue(self::invokePrivate($this->instance, 'isWriteOperation')); - } else { - $this->assertFalse(self::invokePrivate($this->instance, 'isWriteOperation')); - } - } - - public function dataTestBegin() { - return array( - array('w', ['cipher' => 'myCipher'], 'legacyCipher', 'defaultCipher', 'fileKey', 'defaultCipher'), - array('r', ['cipher' => 'myCipher'], 'legacyCipher', 'defaultCipher', 'fileKey', 'myCipher'), - array('w', [], 'legacyCipher', 'defaultCipher', '', 'defaultCipher'), - array('r', [], 'legacyCipher', 'defaultCipher', 'file_key', 'legacyCipher'), - ); - } - - - /** - * test begin() if decryptAll mode was activated - */ - public function testBeginDecryptAll() { - - $path = '/user/files/foo.txt'; - $recoveryKeyId = 'recoveryKeyId'; - $recoveryShareKey = 'recoveryShareKey'; - $decryptAllKey = 'decryptAllKey'; - $fileKey = 'fileKey'; - - $this->sessionMock->expects($this->once()) - ->method('decryptAllModeActivated') - ->willReturn(true); - $this->sessionMock->expects($this->once()) - ->method('getDecryptAllUid') - ->willReturn($recoveryKeyId); - $this->sessionMock->expects($this->once()) - ->method('getDecryptAllKey') - ->willReturn($decryptAllKey); - - $this->keyManagerMock->expects($this->once()) - ->method('getEncryptedFileKey') - ->willReturn('encryptedFileKey'); - $this->keyManagerMock->expects($this->once()) - ->method('getShareKey') - ->with($path, $recoveryKeyId) - ->willReturn($recoveryShareKey); - $this->cryptMock->expects($this->once()) - ->method('multiKeyDecrypt') - ->with('encryptedFileKey', $recoveryShareKey, $decryptAllKey) - ->willReturn($fileKey); - - $this->keyManagerMock->expects($this->never())->method('getFileKey'); - - $this->instance->begin($path, 'user', 'r', [], []); - - $this->assertSame($fileKey, - $this->invokePrivate($this->instance, 'fileKey') - ); - } - - /** - * @dataProvider dataTestUpdate - * - * @param string $fileKey - * @param boolean $expected - */ - public function testUpdate($fileKey, $expected) { - $this->keyManagerMock->expects($this->once()) - ->method('getFileKey')->willReturn($fileKey); - - $this->keyManagerMock->expects($this->any()) - ->method('getPublicKey')->willReturn('publicKey'); - - $this->keyManagerMock->expects($this->any()) - ->method('addSystemKeys') - ->willReturnCallback(function($accessList, $publicKeys) { - return $publicKeys; - }); - - $this->keyManagerMock->expects($this->never())->method('getVersion'); - $this->keyManagerMock->expects($this->never())->method('setVersion'); - - $this->assertSame($expected, - $this->instance->update('path', 'user1', ['users' => ['user1']]) - ); - } - - public function dataTestUpdate() { - return array( - array('', false), - array('fileKey', true) - ); - } - - public function testUpdateNoUsers() { - - $this->invokePrivate($this->instance, 'rememberVersion', [['path' => 2]]); - - $this->keyManagerMock->expects($this->never())->method('getFileKey'); - $this->keyManagerMock->expects($this->never())->method('getPublicKey'); - $this->keyManagerMock->expects($this->never())->method('addSystemKeys'); - $this->keyManagerMock->expects($this->once())->method('setVersion') - ->willReturnCallback(function($path, $version, $view) { - $this->assertSame('path', $path); - $this->assertSame(2, $version); - $this->assertTrue($view instanceof \OC\Files\View); - }); - $this->instance->update('path', 'user1', []); - } - - /** - * Test case if the public key is missing. ownCloud should still encrypt - * the file for the remaining users - */ - public function testUpdateMissingPublicKey() { - $this->keyManagerMock->expects($this->once()) - ->method('getFileKey')->willReturn('fileKey'); - - $this->keyManagerMock->expects($this->any()) - ->method('getPublicKey')->willReturnCallback( - function($user) { - throw new PublicKeyMissingException($user); - } - ); - - $this->keyManagerMock->expects($this->any()) - ->method('addSystemKeys') - ->willReturnCallback(function($accessList, $publicKeys) { - return $publicKeys; - }); - - $this->cryptMock->expects($this->once())->method('multiKeyEncrypt') - ->willReturnCallback( - function($fileKey, $publicKeys) { - $this->assertEmpty($publicKeys); - $this->assertSame('fileKey', $fileKey); - } - ); - - $this->keyManagerMock->expects($this->never())->method('getVersion'); - $this->keyManagerMock->expects($this->never())->method('setVersion'); - - $this->assertTrue( - $this->instance->update('path', 'user1', ['users' => ['user1']]) - ); - } - - /** - * by default the encryption module should encrypt regular files, files in - * files_versions and files in files_trashbin - * - * @dataProvider dataTestShouldEncrypt - */ - public function testShouldEncrypt($path, $shouldEncryptHomeStorage, $isHomeStorage, $expected) { - $this->utilMock->expects($this->once())->method('shouldEncryptHomeStorage') - ->willReturn($shouldEncryptHomeStorage); - - if ($shouldEncryptHomeStorage === false) { - $this->storageMock->expects($this->once())->method('instanceOfStorage') - ->with('\OCP\Files\IHomeStorage')->willReturn($isHomeStorage); - $this->utilMock->expects($this->once())->method('getStorage')->with($path) - ->willReturn($this->storageMock); - } - - $this->assertSame($expected, - $this->instance->shouldEncrypt($path) - ); - } - - public function dataTestShouldEncrypt() { - return array( - array('/user1/files/foo.txt', true, true, true), - array('/user1/files_versions/foo.txt', true, true, true), - array('/user1/files_trashbin/foo.txt', true, true, true), - array('/user1/some_folder/foo.txt', true, true, false), - array('/user1/foo.txt', true, true, false), - array('/user1/files', true, true, false), - array('/user1/files_trashbin', true, true, false), - array('/user1/files_versions', true, true, false), - // test if shouldEncryptHomeStorage is set to false - array('/user1/files/foo.txt', false, true, false), - array('/user1/files_versions/foo.txt', false, false, true), - ); - } - - /** - * @expectedException \OC\Encryption\Exceptions\DecryptionFailedException - * @expectedExceptionMessage Can not decrypt this file, probably this is a shared file. Please ask the file owner to reshare the file with you. - */ - public function testDecrypt() { - $this->instance->decrypt('abc'); - } - - public function testPrepareDecryptAll() { - $input = $this->getMock('Symfony\Component\Console\Input\InputInterface'); - $output = $this->getMock('Symfony\Component\Console\Output\OutputInterface'); - - $this->decryptAllMock->expects($this->once())->method('prepare') - ->with($input, $output, 'user'); - - $this->instance->prepareDecryptAll($input, $output, 'user'); - } - -} |