From edce55989327f4275c8c94db17183ad3780232e6 Mon Sep 17 00:00:00 2001
From: Chris Kankiewicz <Chris@ChrisKankiewicz.com>
Date: Sat, 4 Jan 2020 02:29:39 -0700
Subject: [PATCH] Added 404 response test

---
 .../Controllers/DirectoryControllerTest.php   | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/tests/Unit/Controllers/DirectoryControllerTest.php b/tests/Unit/Controllers/DirectoryControllerTest.php
index bdcde64..067dee3 100644
--- a/tests/Unit/Controllers/DirectoryControllerTest.php
+++ b/tests/Unit/Controllers/DirectoryControllerTest.php
@@ -31,4 +31,24 @@ class DirectoryControllerTest extends TestCase
         $this->assertInstanceOf(ResponseInterface::class, $response);
         $this->assertEquals(200, $response->getStatusCode());
     }
+
+    public function test_it_returns_a_404_error_when_not_found()
+    {
+        $this->container->call(ViewComposer::class);
+
+        $controller = new DirectoryController(
+            $this->container,
+            $this->config,
+            $this->container->get(Twig::class)
+        );
+
+        $response = $controller(
+            new Finder(),
+            new Response(),
+            '404'
+        );
+
+        $this->assertInstanceOf(ResponseInterface::class, $response);
+        $this->assertEquals(404, $response->getStatusCode());
+    }
 }