1 files changed, 95 insertions, 0 deletions
diff --git a/tests/lib/AppFramework/Middleware/SessionMiddlewareTest.php b/tests/lib/AppFramework/Middleware/SessionMiddlewareTest.php
new file mode 100644
index 00000000000..17fcc1904c1
--- /dev/null
+++ b/tests/lib/AppFramework/Middleware/SessionMiddlewareTest.php
@@ -0,0 +1,95 @@
+<?php
+/**
+ * ownCloud - App Framework
+ *
+ * This file is licensed under the Affero General Public License version 3 or
+ * later. See the COPYING file.
+ *
+ * @author Thomas Müller <deepdiver@owncloud.com>
+ * @copyright Thomas Müller 2014
+ */
+
+
+namespace Test\AppFramework\Middleware;
+
+use OC\AppFramework\Http\Request;
+use OC\AppFramework\Middleware\SessionMiddleware;
+use OC\AppFramework\Utility\ControllerMethodReflector;
+use OCP\AppFramework\Http\Response;
+
+
+class SessionMiddlewareTest extends \Test\TestCase {
+
+	/**
+	 * @var ControllerMethodReflector
+	 */
+	private $reflector;
+
+	/**
+	 * @var Request
+	 */
+	private $request;
+
+	protected function setUp() {
+		parent::setUp();
+
+		$this->request = new Request(
+			[],
+			$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(),
+			$this->getMock('\OCP\IConfig')
+		);
+		$this->reflector = new ControllerMethodReflector();
+	}
+
+	/**
+	 * @UseSession
+	 */
+	public function testSessionNotClosedOnBeforeController() {
+		$session = $this->getSessionMock(0);
+
+		$this->reflector->reflect($this, __FUNCTION__);
+		$middleware = new SessionMiddleware($this->request, $this->reflector, $session);
+		$middleware->beforeController($this, __FUNCTION__);
+	}
+
+	/**
+	 * @UseSession
+	 */
+	public function testSessionClosedOnAfterController() {
+		$session = $this->getSessionMock(1);
+
+		$this->reflector->reflect($this, __FUNCTION__);
+		$middleware = new SessionMiddleware($this->request, $this->reflector, $session);
+		$middleware->afterController($this, __FUNCTION__, new Response());
+	}
+
+	public function testSessionClosedOnBeforeController() {
+		$session = $this->getSessionMock(1);
+
+		$this->reflector->reflect($this, __FUNCTION__);
+		$middleware = new SessionMiddleware($this->request, $this->reflector, $session);
+		$middleware->beforeController($this, __FUNCTION__);
+	}
+
+	public function testSessionNotClosedOnAfterController() {
+		$session = $this->getSessionMock(0);
+
+		$this->reflector->reflect($this, __FUNCTION__);
+		$middleware = new SessionMiddleware($this->request, $this->reflector, $session);
+		$middleware->afterController($this, __FUNCTION__, new Response());
+	}
+
+	/**
+	 * @return mixed
+	 */
+	private function getSessionMock($expectedCloseCount) {
+		$session = $this->getMockBuilder('\OC\Session\Memory')
+			->disableOriginalConstructor()
+			->getMock();
+
+		$session->expects($this->exactly($expectedCloseCount))
+			->method('close');
+		return $session;
+	}
+
+}