Fix tests

Signed-off-by: Roeland Jago Douma <roeland@famdouma.nl>
This commit is contained in:
Roeland Jago Douma 2017-07-26 10:50:39 +02:00
parent 3548603a88
commit f71dc7523f
No known key found for this signature in database
GPG Key ID: F941078878347C0C
5 changed files with 81 additions and 69 deletions

View File

@ -26,6 +26,7 @@ namespace OCA\Federation\Middleware;
use OC\HintException; use OC\HintException;
use OCA\Federation\Controller\SettingsController; use OCA\Federation\Controller\SettingsController;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Http; use OCP\AppFramework\Http;
use OCP\AppFramework\Http\JSONResponse; use OCP\AppFramework\Http\JSONResponse;
use OCP\AppFramework\Middleware; use OCP\AppFramework\Middleware;
@ -52,12 +53,13 @@ class AddServerMiddleware extends Middleware {
/** /**
* Log error message and return a response which can be displayed to the user * Log error message and return a response which can be displayed to the user
* *
* @param \OCP\AppFramework\Controller $controller * @param Controller $controller
* @param string $methodName * @param string $methodName
* @param \Exception $exception * @param \Exception $exception
* @return JSONResponse * @return JSONResponse
* @throws \Exception
*/ */
public function afterException($controller, $methodName, \Exception $exception) { public function afterException(Controller $controller, $methodName, \Exception $exception) {
if (($controller instanceof SettingsController) === false) { if (($controller instanceof SettingsController) === false) {
throw $exception; throw $exception;
} }

View File

@ -24,6 +24,7 @@ namespace Test\Core\Middleware;
use OC\Core\Middleware\TwoFactorMiddleware; use OC\Core\Middleware\TwoFactorMiddleware;
use OC\AppFramework\Http\Request; use OC\AppFramework\Http\Request;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Utility\IControllerMethodReflector; use OCP\AppFramework\Utility\IControllerMethodReflector;
use OCP\IConfig; use OCP\IConfig;
use OCP\ISession; use OCP\ISession;
@ -44,6 +45,9 @@ class TwoFactorMiddlewareTest extends TestCase {
/** @var TwoFactorMiddleware */ /** @var TwoFactorMiddleware */
private $middleware; private $middleware;
/** @var Controller */
private $controller;
protected function setUp() { protected function setUp() {
parent::setUp(); parent::setUp();
@ -67,6 +71,7 @@ class TwoFactorMiddlewareTest extends TestCase {
); );
$this->middleware = new TwoFactorMiddleware($this->twoFactorManager, $this->userSession, $this->session, $this->urlGenerator, $this->reflector, $this->request); $this->middleware = new TwoFactorMiddleware($this->twoFactorManager, $this->userSession, $this->session, $this->urlGenerator, $this->reflector, $this->request);
$this->controller = $this->createMock(Controller::class);
} }
public function testBeforeControllerNotLoggedIn() { public function testBeforeControllerNotLoggedIn() {
@ -81,7 +86,7 @@ class TwoFactorMiddlewareTest extends TestCase {
$this->userSession->expects($this->never()) $this->userSession->expects($this->never())
->method('getUser'); ->method('getUser');
$this->middleware->beforeController(null, 'index'); $this->middleware->beforeController($this->controller, 'index');
} }
public function testBeforeControllerPublicPage() { public function testBeforeControllerPublicPage() {
@ -92,7 +97,7 @@ class TwoFactorMiddlewareTest extends TestCase {
$this->userSession->expects($this->never()) $this->userSession->expects($this->never())
->method('isLoggedIn'); ->method('isLoggedIn');
$this->middleware->beforeController(null, 'create'); $this->middleware->beforeController($this->controller, 'create');
} }
public function testBeforeControllerNoTwoFactorCheckNeeded() { public function testBeforeControllerNoTwoFactorCheckNeeded() {
@ -113,7 +118,7 @@ class TwoFactorMiddlewareTest extends TestCase {
->with($user) ->with($user)
->will($this->returnValue(false)); ->will($this->returnValue(false));
$this->middleware->beforeController(null, 'index'); $this->middleware->beforeController($this->controller, 'index');
} }
/** /**
@ -141,7 +146,7 @@ class TwoFactorMiddlewareTest extends TestCase {
->with($user) ->with($user)
->will($this->returnValue(true)); ->will($this->returnValue(true));
$this->middleware->beforeController(null, 'index'); $this->middleware->beforeController($this->controller, 'index');
} }
/** /**
@ -184,7 +189,7 @@ class TwoFactorMiddlewareTest extends TestCase {
->will($this->returnValue('test/url')); ->will($this->returnValue('test/url'));
$expected = new \OCP\AppFramework\Http\RedirectResponse('test/url'); $expected = new \OCP\AppFramework\Http\RedirectResponse('test/url');
$this->assertEquals($expected, $this->middleware->afterException(null, 'index', $ex)); $this->assertEquals($expected, $this->middleware->afterException($this->controller, 'index', $ex));
} }
public function testAfterException() { public function testAfterException() {
@ -196,7 +201,7 @@ class TwoFactorMiddlewareTest extends TestCase {
->will($this->returnValue('redirect/url')); ->will($this->returnValue('redirect/url'));
$expected = new \OCP\AppFramework\Http\RedirectResponse('redirect/url'); $expected = new \OCP\AppFramework\Http\RedirectResponse('redirect/url');
$this->assertEquals($expected, $this->middleware->afterException(null, 'index', $ex)); $this->assertEquals($expected, $this->middleware->afterException($this->controller, 'index', $ex));
} }
} }

View File

@ -17,26 +17,30 @@ use OC\AppFramework\Middleware\Security\CORSMiddleware;
use OC\AppFramework\Utility\ControllerMethodReflector; use OC\AppFramework\Utility\ControllerMethodReflector;
use OC\AppFramework\Middleware\Security\Exceptions\SecurityException; use OC\AppFramework\Middleware\Security\Exceptions\SecurityException;
use OC\Security\Bruteforce\Throttler; use OC\Security\Bruteforce\Throttler;
use OC\User\Session;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Http\JSONResponse; use OCP\AppFramework\Http\JSONResponse;
use OCP\AppFramework\Http\Response; use OCP\AppFramework\Http\Response;
use OCP\IConfig;
use OCP\Security\ISecureRandom;
class CORSMiddlewareTest extends \Test\TestCase { class CORSMiddlewareTest extends \Test\TestCase {
/** @var ControllerMethodReflector */
private $reflector; private $reflector;
/** @var Session|\PHPUnit_Framework_MockObject_MockObject */
private $session; private $session;
/** @var Throttler */ /** @var Throttler */
private $throttler; private $throttler;
/** @var Controller */
private $controller;
protected function setUp() { protected function setUp() {
parent::setUp(); parent::setUp();
$this->reflector = new ControllerMethodReflector(); $this->reflector = new ControllerMethodReflector();
$this->session = $this->getMockBuilder('\OC\User\Session') $this->session = $this->createMock(Session::class);
->disableOriginalConstructor() $this->throttler = $this->createMock(Throttler::class);
->getMock(); $this->controller = $this->createMock(Controller::class);
$this->throttler = $this->getMockBuilder('\OC\Security\Bruteforce\Throttler')
->disableOriginalConstructor()
->getMock();
} }
/** /**
@ -49,13 +53,13 @@ class CORSMiddlewareTest extends \Test\TestCase {
'HTTP_ORIGIN' => 'test' 'HTTP_ORIGIN' => 'test'
] ]
], ],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$response = $middleware->afterController($this, __FUNCTION__, new Response()); $response = $middleware->afterController($this->controller, __FUNCTION__, new Response());
$headers = $response->getHeaders(); $headers = $response->getHeaders();
$this->assertEquals('test', $headers['Access-Control-Allow-Origin']); $this->assertEquals('test', $headers['Access-Control-Allow-Origin']);
} }
@ -68,12 +72,12 @@ class CORSMiddlewareTest extends \Test\TestCase {
'HTTP_ORIGIN' => 'test' 'HTTP_ORIGIN' => 'test'
] ]
], ],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$response = $middleware->afterController($this, __FUNCTION__, new Response()); $response = $middleware->afterController($this->controller, __FUNCTION__, new Response());
$headers = $response->getHeaders(); $headers = $response->getHeaders();
$this->assertFalse(array_key_exists('Access-Control-Allow-Origin', $headers)); $this->assertFalse(array_key_exists('Access-Control-Allow-Origin', $headers));
} }
@ -85,13 +89,13 @@ class CORSMiddlewareTest extends \Test\TestCase {
public function testNoOriginHeaderNoCORSHEADER() { public function testNoOriginHeaderNoCORSHEADER() {
$request = new Request( $request = new Request(
[], [],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$response = $middleware->afterController($this, __FUNCTION__, new Response()); $response = $middleware->afterController($this->controller, __FUNCTION__, new Response());
$headers = $response->getHeaders(); $headers = $response->getHeaders();
$this->assertFalse(array_key_exists('Access-Control-Allow-Origin', $headers)); $this->assertFalse(array_key_exists('Access-Control-Allow-Origin', $headers));
} }
@ -108,15 +112,15 @@ class CORSMiddlewareTest extends \Test\TestCase {
'HTTP_ORIGIN' => 'test' 'HTTP_ORIGIN' => 'test'
] ]
], ],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$response = new Response(); $response = new Response();
$response->addHeader('AcCess-control-Allow-Credentials ', 'TRUE'); $response->addHeader('AcCess-control-Allow-Credentials ', 'TRUE');
$middleware->afterController($this, __FUNCTION__, $response); $middleware->afterController($this->controller, __FUNCTION__, $response);
} }
/** /**
@ -126,8 +130,8 @@ class CORSMiddlewareTest extends \Test\TestCase {
public function testNoCORSShouldAllowCookieAuth() { public function testNoCORSShouldAllowCookieAuth() {
$request = new Request( $request = new Request(
[], [],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
@ -139,7 +143,7 @@ class CORSMiddlewareTest extends \Test\TestCase {
->will($this->returnValue(true)); ->will($this->returnValue(true));
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware->beforeController($this, __FUNCTION__, new Response()); $middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -151,8 +155,8 @@ class CORSMiddlewareTest extends \Test\TestCase {
'PHP_AUTH_USER' => 'user', 'PHP_AUTH_USER' => 'user',
'PHP_AUTH_PW' => 'pass' 'PHP_AUTH_PW' => 'pass'
]], ]],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->session->expects($this->once()) $this->session->expects($this->once())
->method('logout'); ->method('logout');
@ -163,7 +167,7 @@ class CORSMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$middleware->beforeController($this, __FUNCTION__, new Response()); $middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -176,8 +180,8 @@ class CORSMiddlewareTest extends \Test\TestCase {
'PHP_AUTH_USER' => 'user', 'PHP_AUTH_USER' => 'user',
'PHP_AUTH_PW' => 'pass' 'PHP_AUTH_PW' => 'pass'
]], ]],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->session->expects($this->once()) $this->session->expects($this->once())
->method('logout'); ->method('logout');
@ -188,7 +192,7 @@ class CORSMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$middleware->beforeController($this, __FUNCTION__, new Response()); $middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -201,8 +205,8 @@ class CORSMiddlewareTest extends \Test\TestCase {
'PHP_AUTH_USER' => 'user', 'PHP_AUTH_USER' => 'user',
'PHP_AUTH_PW' => 'pass' 'PHP_AUTH_PW' => 'pass'
]], ]],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$this->session->expects($this->once()) $this->session->expects($this->once())
->method('logout'); ->method('logout');
@ -213,7 +217,7 @@ class CORSMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$middleware->beforeController($this, __FUNCTION__, new Response()); $middleware->beforeController($this->controller, __FUNCTION__);
} }
public function testAfterExceptionWithSecurityExceptionNoStatus() { public function testAfterExceptionWithSecurityExceptionNoStatus() {
@ -222,11 +226,11 @@ class CORSMiddlewareTest extends \Test\TestCase {
'PHP_AUTH_USER' => 'user', 'PHP_AUTH_USER' => 'user',
'PHP_AUTH_PW' => 'pass' 'PHP_AUTH_PW' => 'pass'
]], ]],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$response = $middleware->afterException($this, __FUNCTION__, new SecurityException('A security exception')); $response = $middleware->afterException($this->controller, __FUNCTION__, new SecurityException('A security exception'));
$expected = new JSONResponse(['message' => 'A security exception'], 500); $expected = new JSONResponse(['message' => 'A security exception'], 500);
$this->assertEquals($expected, $response); $this->assertEquals($expected, $response);
@ -238,11 +242,11 @@ class CORSMiddlewareTest extends \Test\TestCase {
'PHP_AUTH_USER' => 'user', 'PHP_AUTH_USER' => 'user',
'PHP_AUTH_PW' => 'pass' 'PHP_AUTH_PW' => 'pass'
]], ]],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$response = $middleware->afterException($this, __FUNCTION__, new SecurityException('A security exception', 501)); $response = $middleware->afterException($this->controller, __FUNCTION__, new SecurityException('A security exception', 501));
$expected = new JSONResponse(['message' => 'A security exception'], 501); $expected = new JSONResponse(['message' => 'A security exception'], 501);
$this->assertEquals($expected, $response); $this->assertEquals($expected, $response);
@ -258,11 +262,11 @@ class CORSMiddlewareTest extends \Test\TestCase {
'PHP_AUTH_USER' => 'user', 'PHP_AUTH_USER' => 'user',
'PHP_AUTH_PW' => 'pass' 'PHP_AUTH_PW' => 'pass'
]], ]],
$this->getMockBuilder('\OCP\Security\ISecureRandom')->getMock(), $this->createMock(ISecureRandom::class),
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->createMock(IConfig::class)
); );
$middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler); $middleware = new CORSMiddleware($request, $this->reflector, $this->session, $this->throttler);
$middleware->afterException($this, __FUNCTION__, new \Exception('A regular exception')); $middleware->afterException($this->controller, __FUNCTION__, new \Exception('A regular exception'));
} }
} }

View File

@ -131,7 +131,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->with($this->equalTo('files')); ->with($this->equalTo('files'));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
@ -152,7 +152,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
try { try {
$this->reader->reflect(__CLASS__, $method); $this->reader->reflect(__CLASS__, $method);
$sec->beforeController(__CLASS__, $method); $sec->beforeController($this->controller, $method);
} catch (SecurityException $ex){ } catch (SecurityException $ex){
$this->assertEquals($status, $ex->getCode()); $this->assertEquals($status, $ex->getCode());
} }
@ -234,7 +234,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
$sec = $this->getMiddleware(false, false); $sec = $this->getMiddleware(false, false);
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$sec->beforeController(__CLASS__, __FUNCTION__); $sec->beforeController($this->controller, __FUNCTION__);
} }
@ -261,7 +261,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
} }
$this->reader->reflect(__CLASS__, $method); $this->reader->reflect(__CLASS__, $method);
$sec->beforeController(__CLASS__, $method); $sec->beforeController($this->controller, $method);
} }
@ -277,7 +277,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->method('passesStrictCookieCheck') ->method('passesStrictCookieCheck')
->will($this->returnValue(true)); ->will($this->returnValue(true));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
@ -291,7 +291,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->will($this->returnValue(false)); ->will($this->returnValue(false));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -306,7 +306,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->will($this->returnValue(true)); ->will($this->returnValue(true));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -322,7 +322,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->will($this->returnValue(true)); ->will($this->returnValue(true));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -338,7 +338,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->will($this->returnValue(false)); ->will($this->returnValue(false));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
@ -352,7 +352,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->will($this->returnValue(false)); ->will($this->returnValue(false));
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -367,7 +367,7 @@ class SecurityMiddlewareTest extends \Test\TestCase {
->willReturn(true); ->willReturn(true);
$this->reader->reflect(__CLASS__, __FUNCTION__); $this->reader->reflect(__CLASS__, __FUNCTION__);
$this->middleware->beforeController(__CLASS__, __FUNCTION__); $this->middleware->beforeController($this->controller, __FUNCTION__);
} }
public function dataCsrfOcsController() { public function dataCsrfOcsController() {

View File

@ -15,21 +15,21 @@ namespace Test\AppFramework\Middleware;
use OC\AppFramework\Http\Request; use OC\AppFramework\Http\Request;
use OC\AppFramework\Middleware\SessionMiddleware; use OC\AppFramework\Middleware\SessionMiddleware;
use OC\AppFramework\Utility\ControllerMethodReflector; use OC\AppFramework\Utility\ControllerMethodReflector;
use OCP\AppFramework\Controller;
use OCP\AppFramework\Http\Response; use OCP\AppFramework\Http\Response;
class SessionMiddlewareTest extends \Test\TestCase { class SessionMiddlewareTest extends \Test\TestCase {
/** /** @var ControllerMethodReflector */
* @var ControllerMethodReflector
*/
private $reflector; private $reflector;
/** /** @var Request */
* @var Request
*/
private $request; private $request;
/** @var Controller */
private $controller;
protected function setUp() { protected function setUp() {
parent::setUp(); parent::setUp();
@ -39,6 +39,7 @@ class SessionMiddlewareTest extends \Test\TestCase {
$this->getMockBuilder('\OCP\IConfig')->getMock() $this->getMockBuilder('\OCP\IConfig')->getMock()
); );
$this->reflector = new ControllerMethodReflector(); $this->reflector = new ControllerMethodReflector();
$this->controller = $this->createMock(Controller::class);
} }
/** /**
@ -49,7 +50,7 @@ class SessionMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new SessionMiddleware($this->request, $this->reflector, $session); $middleware = new SessionMiddleware($this->request, $this->reflector, $session);
$middleware->beforeController($this, __FUNCTION__); $middleware->beforeController($this->controller, __FUNCTION__);
} }
/** /**
@ -60,7 +61,7 @@ class SessionMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new SessionMiddleware($this->request, $this->reflector, $session); $middleware = new SessionMiddleware($this->request, $this->reflector, $session);
$middleware->afterController($this, __FUNCTION__, new Response()); $middleware->afterController($this->controller, __FUNCTION__, new Response());
} }
public function testSessionClosedOnBeforeController() { public function testSessionClosedOnBeforeController() {
@ -68,7 +69,7 @@ class SessionMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new SessionMiddleware($this->request, $this->reflector, $session); $middleware = new SessionMiddleware($this->request, $this->reflector, $session);
$middleware->beforeController($this, __FUNCTION__); $middleware->beforeController($this->controller, __FUNCTION__);
} }
public function testSessionNotClosedOnAfterController() { public function testSessionNotClosedOnAfterController() {
@ -76,7 +77,7 @@ class SessionMiddlewareTest extends \Test\TestCase {
$this->reflector->reflect($this, __FUNCTION__); $this->reflector->reflect($this, __FUNCTION__);
$middleware = new SessionMiddleware($this->request, $this->reflector, $session); $middleware = new SessionMiddleware($this->request, $this->reflector, $session);
$middleware->afterController($this, __FUNCTION__, new Response()); $middleware->afterController($this->controller, __FUNCTION__, new Response());
} }
/** /**