CryptoWrappingTest.php 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. <?php
  2. /**
  3. * @author Joas Schilling <nickvergessen@owncloud.com>
  4. *
  5. * @copyright Copyright (c) 2015, ownCloud, Inc.
  6. * @license AGPL-3.0
  7. *
  8. * This code is free software: you can redistribute it and/or modify
  9. * it under the terms of the GNU Affero General Public License, version 3,
  10. * as published by the Free Software Foundation.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU Affero General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Affero General Public License, version 3,
  18. * along with this program. If not, see <http://www.gnu.org/licenses/>
  19. *
  20. */
  21. namespace Test\Session;
  22. use OC\Session\CryptoSessionData;
  23. use OCP\ISession;
  24. use Test\TestCase;
  25. class CryptoWrappingTest extends TestCase {
  26. /** @var \PHPUnit\Framework\MockObject\MockObject|\OCP\Security\ICrypto */
  27. protected $crypto;
  28. /** @var \PHPUnit\Framework\MockObject\MockObject|\OCP\ISession */
  29. protected $wrappedSession;
  30. /** @var \OC\Session\CryptoSessionData */
  31. protected $instance;
  32. protected function setUp(): void {
  33. parent::setUp();
  34. $this->wrappedSession = $this->getMockBuilder(ISession::class)
  35. ->disableOriginalConstructor()
  36. ->getMock();
  37. $this->crypto = $this->getMockBuilder('OCP\Security\ICrypto')
  38. ->disableOriginalConstructor()
  39. ->getMock();
  40. $this->crypto->expects($this->any())
  41. ->method('encrypt')
  42. ->willReturnCallback(function ($input) {
  43. return $input;
  44. });
  45. $this->crypto->expects($this->any())
  46. ->method('decrypt')
  47. ->willReturnCallback(function ($input) {
  48. if ($input === '') {
  49. return '';
  50. }
  51. return substr($input, 1, -1);
  52. });
  53. $this->instance = new CryptoSessionData($this->wrappedSession, $this->crypto, 'PASS');
  54. }
  55. public function testUnwrappingGet() {
  56. $unencryptedValue = 'foobar';
  57. $encryptedValue = $this->crypto->encrypt($unencryptedValue);
  58. $this->wrappedSession->expects($this->once())
  59. ->method('get')
  60. ->with('encrypted_session_data')
  61. ->willReturnCallback(function () use ($encryptedValue) {
  62. return $encryptedValue;
  63. });
  64. $this->assertSame($unencryptedValue, $this->wrappedSession->get('encrypted_session_data'));
  65. }
  66. }