Add tests

Signed-off-by: Lukas Reschke <lukas@statuscode.ch>

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/lib/Security/CSRF/TokenStorage/SessionStorageTest.php b/tests/lib/Security/CSRF/TokenStorage/SessionStorageTest.php
index 550fa49e1b2..d1e76684507 100644
--- a/tests/lib/Security/CSRF/TokenStorage/SessionStorageTest.php
+++ b/tests/lib/Security/CSRF/TokenStorage/SessionStorageTest.php
@@ -21,6 +21,8 @@
 
 namespace Test\Security\CSRF\TokenStorage;
 
+use OCP\ISession;
+
 class SessionStorageTest extends \Test\TestCase {
 	/** @var \OCP\ISession */
 	private $session;
@@ -106,4 +108,15 @@ class SessionStorageTest extends \Test\TestCase {
 			->willReturn(false);
 		$this->assertSame(false, $this->sessionStorage->hasToken());
 	}
+
+	public function testSetSession() {
+		$session = $this->createMock(ISession::class);
+		$session
+			->expects($this->once())
+			->method('get')
+			->with('requesttoken')
+			->willReturn('MyToken');
+		$this->sessionStorage->setSession($session);
+		$this->assertSame('MyToken', $this->sessionStorage->getToken());
+	}
 }