Wither Method added for defaultReturnType

selamiphp · Dec 20, 2018 · f127729 · f127729
1 parent 3c4145d
commit f127729
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/src/Dispatcher.php b/src/Dispatcher.php
@@ -82,7 +82,6 @@ private function createCachedRoute($routeCollector) : void
         }
     }
 
-
     private function cachedDispatcher() : FastRoute\Dispatcher\GroupCountBased
     {
         $dispatchData = include $this->cachedFile;

diff --git a/src/Router.php b/src/Router.php
@@ -118,12 +118,20 @@ public function __construct(
         $this->defaultReturnType = ($defaultReturnType >=1 && $defaultReturnType <=7) ? $defaultReturnType : self::HTML;
     }
 
+    public function withDefaultReturnType(int $defaultReturnType) : self
+    {
+        $new = clone $this;
+        $new->$defaultReturnType = $defaultReturnType;
+        return $new;
+    }
+
     public function withSubFolder(string $folder) : self
     {
         $new = clone $this;
         $new->requestedPath = $this->extractFolder($this->requestedPath, $folder);
         return $new;
     }
+
     public function withCacheFile(string $fileName) : self
     {
         $new = clone $this;
@@ -145,7 +153,6 @@ private function extractFolder(string $requestPath, string $folder) : string
         return $requestPath;
     }
 
-
     public function add(
         $requestMethods,
         string $route,

diff --git a/test/RouterTest.php b/test/RouterTest.php
@@ -112,7 +112,8 @@ public function shouldThrowExceptionForInvalidCachedFile() : void
             $this->request->getUri()->getPath()
         );
         $router = $router->withSubFolder($this->config['folder'])
-            ->withCacheFile('/tmp/failed.cache');
+            ->withCacheFile('/tmp/failed.cache')
+            ->withDefaultReturnType(Router::HTML);
         $router->add(Router::GET, '/', 'app/main', Router::HTML, 'home');
         $router->getRoute();
         $this->assertFileExists(
-Original file line number
+Diff line change
@@ Expand Up / @@ -82,7 +82,6 @@ private function createCachedRoute($routeCollector) : void @@
             }
         }
         private function cachedDispatcher() : FastRoute\Dispatcher\GroupCountBased
         {
             $dispatchData = include $this->cachedFile;
@@ Expand Down @@