123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272 |
- <?php
- /**
- * @author Björn Schießle <schiessle@owncloud.com>
- * @author Clark Tomlinson <fallen013@gmail.com>
- * @author Morris Jobke <hey@morrisjobke.de>
- *
- * @copyright Copyright (c) 2015, 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\Hooks;
-
-
- use OCA\Encryption\Crypto\Crypt;
- use OCA\Encryption\Hooks\UserHooks;
- use Test\TestCase;
-
- class UserHooksTest extends TestCase {
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $utilMock;
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $recoveryMock;
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $sessionMock;
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $keyManagerMock;
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $userSetupMock;
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $userSessionMock;
- /**
- * @var \PHPUnit_Framework_MockObject_MockObject
- */
- private $cryptMock;
- /**
- * @var UserHooks
- */
- private $instance;
-
- private $params = ['uid' => 'testUser', 'password' => 'password'];
-
- public function testLogin() {
- $this->userSetupMock->expects($this->exactly(2))
- ->method('setupUser')
- ->willReturnOnConsecutiveCalls(true, false);
-
- $this->keyManagerMock->expects($this->once())
- ->method('init')
- ->with('testUser', 'password');
-
- $this->assertNull($this->instance->login($this->params));
- $this->assertFalse($this->instance->login($this->params));
- }
-
- public function testLogout() {
- $this->sessionMock->expects($this->once())
- ->method('clear');
-
- $this->assertNull($this->instance->logout());
- }
-
- public function testPostCreateUser() {
- $this->userSetupMock->expects($this->once())
- ->method('setupUser');
-
- $this->assertNull($this->instance->postCreateUser($this->params));
- }
-
- public function testPostDeleteUser() {
- $this->keyManagerMock->expects($this->once())
- ->method('deletePublicKey')
- ->with('testUser');
-
- $this->assertNull($this->instance->postDeleteUser($this->params));
- }
-
- public function testPreSetPassphrase() {
- $this->userSessionMock->expects($this->once())
- ->method('canChangePassword');
-
- $this->assertNull($this->instance->preSetPassphrase($this->params));
- }
-
- public function testSetPassphrase() {
- $this->sessionMock->expects($this->exactly(4))
- ->method('getPrivateKey')
- ->willReturnOnConsecutiveCalls(true, false);
-
- $this->cryptMock->expects($this->exactly(4))
- ->method('symmetricEncryptFileContent')
- ->willReturn(true);
-
- $this->cryptMock->expects($this->any())
- ->method('generateHeader')
- ->willReturn(Crypt::HEADER_START . ':Cipher:test:' . Crypt::HEADER_END);
-
- $this->keyManagerMock->expects($this->exactly(4))
- ->method('setPrivateKey')
- ->willReturnCallback(function ($user, $key) {
- $header = substr($key, 0, strlen(Crypt::HEADER_START));
- $this->assertSame(
- Crypt::HEADER_START,
- $header, 'every encrypted file should start with a header');
- });
-
- $this->assertNull($this->instance->setPassphrase($this->params));
- $this->params['recoveryPassword'] = 'password';
-
- $this->recoveryMock->expects($this->exactly(3))
- ->method('isRecoveryEnabledForUser')
- ->with('testUser')
- ->willReturnOnConsecutiveCalls(true, false);
-
-
- // Test first if statement
- $this->assertNull($this->instance->setPassphrase($this->params));
-
- // Test Second if conditional
- $this->keyManagerMock->expects($this->exactly(2))
- ->method('userHasKeys')
- ->with('testUser')
- ->willReturn(true);
-
- $this->assertNull($this->instance->setPassphrase($this->params));
-
- // Test third and final if condition
- $this->utilMock->expects($this->once())
- ->method('userHasFiles')
- ->with('testUser')
- ->willReturn(false);
-
- $this->cryptMock->expects($this->once())
- ->method('createKeyPair');
-
- $this->keyManagerMock->expects($this->once())
- ->method('setPrivateKey');
-
- $this->recoveryMock->expects($this->once())
- ->method('recoverUsersFiles')
- ->with('password', 'testUser');
-
- $this->assertNull($this->instance->setPassphrase($this->params));
- }
-
- public function testSetPasswordNoUser() {
- $this->sessionMock->expects($this->once())
- ->method('getPrivateKey')
- ->willReturn(true);
-
- $userSessionMock = $this->getMockBuilder('OCP\IUserSession')
- ->disableOriginalConstructor()
- ->getMock();
-
- $userSessionMock->expects($this->any())->method('getUser')->will($this->returnValue(null));
-
- $this->recoveryMock->expects($this->once())
- ->method('isRecoveryEnabledForUser')
- ->with('testUser')
- ->willReturn(false);
-
- $userHooks = new UserHooks($this->keyManagerMock,
- $this->loggerMock,
- $this->userSetupMock,
- $userSessionMock,
- $this->utilMock,
- $this->sessionMock,
- $this->cryptMock,
- $this->recoveryMock
- );
-
- $this->assertNull($userHooks->setPassphrase($this->params));
- }
-
- public function testPostPasswordReset() {
- $this->keyManagerMock->expects($this->once())
- ->method('replaceUserKeys')
- ->with('testUser');
-
- $this->userSetupMock->expects($this->once())
- ->method('setupServerSide')
- ->with('testUser', 'password');
-
- $this->assertNull($this->instance->postPasswordReset($this->params));
- }
-
- protected function setUp() {
- parent::setUp();
- $this->loggerMock = $this->getMock('OCP\ILogger');
- $this->keyManagerMock = $this->getMockBuilder('OCA\Encryption\KeyManager')
- ->disableOriginalConstructor()
- ->getMock();
- $this->userSetupMock = $this->getMockBuilder('OCA\Encryption\Users\Setup')
- ->disableOriginalConstructor()
- ->getMock();
-
- $this->userSessionMock = $this->getMockBuilder('OCP\IUserSession')
- ->disableOriginalConstructor()
- ->setMethods([
- 'isLoggedIn',
- 'getUID',
- 'login',
- 'logout',
- 'setUser',
- 'getUser',
- 'canChangePassword'
- ])
- ->getMock();
-
- $this->userSessionMock->expects($this->any())->method('getUID')->will($this->returnValue('testUser'));
-
- $this->userSessionMock->expects($this->any())
- ->method($this->anything())
- ->will($this->returnSelf());
-
- $utilMock = $this->getMockBuilder('OCA\Encryption\Util')
- ->disableOriginalConstructor()
- ->getMock();
-
- $sessionMock = $this->getMockBuilder('OCA\Encryption\Session')
- ->disableOriginalConstructor()
- ->getMock();
-
- $this->cryptMock = $this->getMockBuilder('OCA\Encryption\Crypto\Crypt')
- ->disableOriginalConstructor()
- ->getMock();
- $recoveryMock = $this->getMockBuilder('OCA\Encryption\Recovery')
- ->disableOriginalConstructor()
- ->getMock();
-
- $this->sessionMock = $sessionMock;
- $this->recoveryMock = $recoveryMock;
- $this->utilMock = $utilMock;
- $this->instance = new UserHooks($this->keyManagerMock,
- $this->loggerMock,
- $this->userSetupMock,
- $this->userSessionMock,
- $this->utilMock,
- $this->sessionMock,
- $this->cryptMock,
- $this->recoveryMock
- );
-
- }
-
- }
|