AddHostPlugin must also handle the port

CHANGELOG.md

@@ -1,5 +1,10 @@
 # Change Log
 
+## Unreleased
+
+### Changed
+
+- AddHostPlugin also sets the port if specified
 
 ## 1.2.0 - 2016-07-14
 

spec/Plugin/AddHostPluginSpec.php

@@ -39,12 +39,14 @@ function it_adds_domain(
     ) {
         $host->getScheme()->shouldBeCalled()->willReturn('http://');
         $host->getHost()->shouldBeCalled()->willReturn('example.com');
+        $host->getPort()->shouldBeCalled()->willReturn(8000);
 
         $request->getUri()->shouldBeCalled()->willReturn($uri);
         $request->withUri($uri)->shouldBeCalled()->willReturn($request);
 
         $uri->withScheme('http://')->shouldBeCalled()->willReturn($uri);
         $uri->withHost('example.com')->shouldBeCalled()->willReturn($uri);
+        $uri->withPort(8000)->shouldBeCalled()->willReturn($uri);
         $uri->getHost()->shouldBeCalled()->willReturn('');
 
         $this->beConstructedWith($host);
@@ -58,13 +60,14 @@ function it_replaces_domain(
     ) {
         $host->getScheme()->shouldBeCalled()->willReturn('http://');
         $host->getHost()->shouldBeCalled()->willReturn('example.com');
+        $host->getPort()->shouldBeCalled()->willReturn(8000);
 
         $request->getUri()->shouldBeCalled()->willReturn($uri);
         $request->withUri($uri)->shouldBeCalled()->willReturn($request);
 
         $uri->withScheme('http://')->shouldBeCalled()->willReturn($uri);
         $uri->withHost('example.com')->shouldBeCalled()->willReturn($uri);
-
+        $uri->withPort(8000)->shouldBeCalled()->willReturn($uri);
 
         $this->beConstructedWith($host, ['replace' => true]);
         $this->handleRequest($request, function () {}, function () {});

src/Plugin/AddHostPlugin.php

@@ -8,7 +8,7 @@
 use Symfony\Component\OptionsResolver\OptionsResolver;
 
 /**
- * Add schema and host to a request. Can be set to overwrite the schema and host if desired.
+ * Add schema, host and port to a request. Can be set to overwrite the schema and host if desired.
  *
  * @author Tobias Nyholm <[email protected]>
  */
@@ -52,8 +52,11 @@ public function __construct(UriInterface $host, array $config = [])
     public function handleRequest(RequestInterface $request, callable $next, callable $first)
     {
         if ($this->replace || $request->getUri()->getHost() === '') {
-            $uri = $request->getUri()->withHost($this->host->getHost());
-            $uri = $uri->withScheme($this->host->getScheme());
+            $uri = $request->getUri()
+                ->withHost($this->host->getHost())
+                ->withScheme($this->host->getScheme())
+                ->withPort($this->host->getPort())
+            ;
 
             $request = $request->withUri($uri);
         }