CSRFTokenControllerTest.php 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. <?php
  2. /**
  3. * @copyright 2017 Christoph Wurst <christoph@winzerhof-wurst.at>
  4. *
  5. * @author 2017 Christoph Wurst <christoph@winzerhof-wurst.at>
  6. *
  7. * @license GNU AGPL version 3 or any later version
  8. *
  9. * This program is free software: you can redistribute it and/or modify
  10. * it under the terms of the GNU Affero General Public License as
  11. * published by the Free Software Foundation, either version 3 of the
  12. * License, or (at your option) any later version.
  13. *
  14. * This program is distributed in the hope that it will be useful,
  15. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  17. * GNU Affero General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU Affero General Public License
  20. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  21. *
  22. */
  23. namespace Tests\Core\Controller;
  24. use OC\Core\Controller\CSRFTokenController;
  25. use OC\Security\CSRF\CsrfToken;
  26. use OC\Security\CSRF\CsrfTokenManager;
  27. use OCP\AppFramework\Http;
  28. use OCP\AppFramework\Http\JSONResponse;
  29. use OCP\IRequest;
  30. use PHPUnit_Framework_MockObject_MockObject;
  31. use Test\TestCase;
  32. class CSRFTokenControllerTest extends TestCase {
  33. /** @var CSRFTokenController */
  34. private $controller;
  35. /** @var IRequest|PHPUnit_Framework_MockObject_MockObject */
  36. private $request;
  37. /** @var CsrfTokenManager|PHPUnit_Framework_MockObject_MockObject */
  38. private $tokenManager;
  39. protected function setUp() {
  40. parent::setUp();
  41. $this->request = $this->createMock(IRequest::class);
  42. $this->tokenManager = $this->createMock(CsrfTokenManager::class);
  43. $this->controller = new CSRFTokenController('core', $this->request,
  44. $this->tokenManager);
  45. }
  46. public function testGetToken() {
  47. $token = $this->createMock(CsrfToken::class);
  48. $this->tokenManager->method('getToken')->willReturn($token);
  49. $token->method('getEncryptedValue')->willReturn('toktok123');
  50. $response = $this->controller->index();
  51. $this->assertInstanceOf(JSONResponse::class, $response);
  52. $this->assertSame(Http::STATUS_OK, $response->getStatus());
  53. $this->assertEquals([
  54. 'token' => 'toktok123'
  55. ], $response->getData());
  56. }
  57. }