1 files changed, 72 insertions, 0 deletions
diff --git a/tests/Router/Middleware/DetermineRouteTest.php b/tests/Router/Middleware/DetermineRouteTest.php
new file mode 100644
index 0000000..9e8d764
--- /dev/null
+++ b/tests/Router/Middleware/DetermineRouteTest.php
@@ -0,0 +1,72 @@
+<?php
+
+use Jasny\Router\Route;
+use Jasny\Router\Routes;
+use Jasny\Router\Middleware\DetermineRoute;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Message\ResponseInterface;
+
+use Jasny\Router\TestHelpers;
+
+/**
+ * @covers Jasny\Router\Middleware\DetermineRoute
+ */
+class DetermineRouteTest extends PHPUnit_Framework_TestCase
+{
+    use TestHelpers;
+    
+    public function testConstruct()
+    {
+        $routes = $this->createMock(Routes::class);
+        $middelware = new DetermineRoute($routes);
+        
+        $this->assertSame($routes, $middelware->getRoutes());
+    }
+    
+    public function routeProvider()
+    {
+        return [
+            [$this->createMock(Route::class)],
+            [null]
+        ];
+    }
+
+    /**
+     * @dataProvider routeProvider
+     * 
+     * @param Route|null $route
+     */
+    public function testInvoke($route)
+    {
+        $requestWithRoute = $this->createMock(ServerRequestInterface::class);
+        
+        $request = $this->createMock(ServerRequestInterface::class);
+        $request->expects($this->once())->method('withAttribute')->with('route', $route)
+            ->willReturn($requestWithRoute);
+        
+        $response = $this->createMock(ResponseInterface::class);
+        
+        $next = $this->createCallbackMock($this->once(), [$requestWithRoute, $response]);
+        
+        $routes = $this->createMock(Routes::class);
+        $routes->expects($this->once())->method('getRoute')->with($request)->willReturn($route);
+        
+        $middelware = new DetermineRoute($routes);
+        
+        $middelware($request, $response, $next);
+    }
+    
+    /**
+     * @expectedException InvalidArgumentException
+     */
+    public function testInvokeWithInvalidCallback()
+    {
+        $request = $this->createMock(ServerRequestInterface::class);
+        $response = $this->createMock(ResponseInterface::class);
+        
+        $routes = $this->createMock(Routes::class);
+        $middelware = new DetermineRoute($routes);
+        
+        $middelware($request, $response, 'not a function');
+    }
+}