#2: LoadBalancer NoAvailableHostException

src/Chooser/ChooserInterface.php

@@ -2,6 +2,8 @@
 
 namespace NBN\LoadBalancer\Chooser;
 
+use Symfony\Component\HttpFoundation\Request;
+
 /**
  * @author Nicolas Bastien <[email protected]>
  */
@@ -12,5 +14,5 @@ interface ChooserInterface
      * @param  array    $hosts
      * @return Response
      */
-    public function handleRequest(Request $request, array $hosts);
+    public function getAvailableHost(Request $request, array $hosts);
 }

src/Exception/NoAvailableHostException.php

@@ -0,0 +1,10 @@
+<?php
+
+namespace NBN\LoadBalancer\Exception;
+
+/**
+ * @author Nicolas Bastien <[email protected]>
+ */
+class NoAvailableHostException extends \RuntimeException
+{
+}

src/LoadBalancer.php

@@ -49,5 +49,9 @@ public function handleRequest(Request $request)
             throw new NoRegisteredHostException();
         }
 
+        $host = $this->chooser->getAvailableHost($request, $this->hosts);
+        if ($host == null) {
+            throw new NoAvailableHostException();
+        }
     }
 }

tests/Unit/LoadBalancerTest.php

@@ -2,6 +2,7 @@
 
 namespace NBN\LoadBalancer;
 
+use NBN\LoadBalancer\Exception\NoAvailableHostException;
 use NBN\LoadBalancer\Exception\NoRegisteredHostException;
 
 /**
@@ -24,4 +25,19 @@ public function testHandleRequestNoHostException()
         $this->expectException(NoRegisteredHostException::class);
         $loadBalancer->handleRequest($resquest->reveal());
     }
+
+    /**
+     * @test LoadBalancer::handleRequest()
+     */
+    public function testHandleRequestNoAvailableException()
+    {
+        $chooser  = $this->prophesize('NBN\LoadBalancer\Chooser\ChooserInterface');
+        $host     = $this->prophesize('NBN\LoadBalancer\Host\HostInterface');
+        $resquest = $this->prophesize('Symfony\Component\HttpFoundation\Request');
+
+        $loadBalancer = new LoadBalancer([$host], $chooser->reveal());
+
+        $this->expectException(NoAvailableHostException::class);
+        $loadBalancer->handleRequest($resquest->reveal());
+    }
 }