Assert that response is instance of TimberResponse

Rareloop · Mar 20, 2024 · d6cba01 · d6cba01
1 parent 8e0f530
commit d6cba01
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/Unit/Http/Middleware/PasswordProtectTest.php b/tests/Unit/Http/Middleware/PasswordProtectTest.php
@@ -11,6 +11,7 @@
 use Psr\Http\Message\ServerRequestInterface;
 use Psr\Http\Server\RequestHandlerInterface;
 use Rareloop\Lumberjack\Http\Middleware\PasswordProtected;
+use Rareloop\Lumberjack\Http\Responses\TimberResponse;
 use Rareloop\Lumberjack\Test\Unit\BrainMonkeyPHPUnitIntegration;
 
 /**
@@ -108,8 +109,10 @@ public function it_renders_the_password_template_when_needed()
         $handler->shouldReceive('handle')->never();
 
         $middleware = new PasswordProtected;
+        $response = $middleware->process($request, $handler);
 
-        $this->assertSame('testing123', $middleware->process($request, $handler)->getBody()->getContents());
+        $this->assertInstanceOf(TimberResponse::class, $response);
+        $this->assertSame('testing123', $response->getBody()->getContents());
         $this->assertTrue(Filters\applied('lumberjack/password_protect_template') > 0);
     }
 }