1 files changed, 13 insertions, 1 deletions
diff --git a/tests/Unit/RouteDefinitionTest.php b/tests/Unit/RouteDefinitionTest.php
index 0e02bac..6e3900e 100644
--- a/tests/Unit/RouteDefinitionTest.php
+++ b/tests/Unit/RouteDefinitionTest.php
@@ -7,6 +7,7 @@ use Lightscale\Router\Group;
 use Lightscale\Router\Route;
 use Lightscale\Router\RouteDefinition;
 use Lightscale\Router\Router;
+use Lightscale\Router\Test\Utils\TestMiddleware;
 
 it('initalizes')
     ->expect(fn () => new RouteDefinition(
@@ -42,4 +43,15 @@ it('gets the route', function () {
     expect($def->getRoute())->toBe($route);
 });
 
-it('adds middleware to route')->todo();
+it('adds middleware to route', function () {
+    $def = new RouteDefinition(
+        new Router(),
+        $route = new Route(HttpMethod::Get, fn () => null),
+    );
+
+    $mw = new TestMiddleware();
+    expect($def->middleware($mw))->toBe($def);
+    expect($route->getMiddleware())
+        ->toHaveCount(1)
+        ->{0}->toBe($mw);
+});