From 402550494275e9224049a69ba5c0aab350e07fac Mon Sep 17 00:00:00 2001
From: Brent Scheffler <brent@brentscheffler.com>
Date: Tue, 18 Jun 2019 18:57:55 -0700
Subject: [PATCH] Adding tests for StreamContextHandler.

---
 tests/StreamContextHandlerTest.php | 63 +++++++++++++++++++++++++++++-
 1 file changed, 61 insertions(+), 2 deletions(-)

diff --git a/tests/StreamContextHandlerTest.php b/tests/StreamContextHandlerTest.php
index 9d95b47..ebe3364 100644
--- a/tests/StreamContextHandlerTest.php
+++ b/tests/StreamContextHandlerTest.php
@@ -2,14 +2,16 @@
 
 namespace Shuttle\Tests;
 
-use PHPUnit\Framework\TestCase;
-use Shuttle\Handler\StreamContextHandler;
 use Capsule\Request;
 use Capsule\Response;
 use Capsule\Stream\BufferStream;
+use PHPUnit\Framework\TestCase;
+use Shuttle\Handler\StreamContextHandler;
+use Shuttle\RequestException;
 
 /**
  * @covers Shuttle\Handler\StreamContextHandler
+ * @covers Shuttle\RequestException
  */
 class StreamContextHandlerTest extends TestCase
 {
@@ -42,6 +44,63 @@ public function test_set_debug()
         $this->assertTrue($debug->getValue($streamHandler));
     }
 
+    public function test_build_request_headers()
+    {
+        $streamHandler = new StreamContextHandler;
+        $reflection = new \ReflectionClass($streamHandler);
+
+        $method = $reflection->getMethod('buildRequestHeaders');
+        $method->setAccessible(true);
+
+        $headers = $method->invokeArgs($streamHandler, [
+            ["X-Header-1" => ["Foo"], "X-Header-2" => ["Bar"]]
+        ]);
+
+        $this->assertEquals([
+            "X-Header-1: Foo",
+            "X-Header-2: Bar"
+        ], $headers);
+    }
+
+    public function test_build_http_context()
+    {
+        $streamHandler = new StreamContextHandler;
+        $reflection = new \ReflectionClass($streamHandler);
+
+        $method = $reflection->getMethod('buildHttpContext');
+        $method->setAccessible(true);
+
+        $request = new Request("post", "http://example.com", "Ok", ["X-Header-1" => "Foo"], "2");
+
+        $httpContext = $method->invoke($streamHandler, $request);
+
+        $this->assertEquals([
+            'follow_location' => 1,
+            'request_fulluri' => false,
+            'max_redirects' => 10,
+            'ignore_errors' => true,
+            'timeout' => 120,
+            'method' => 'POST',
+            'protocol_version' => '2',
+            'header' => ["X-Header-1: Foo"],
+            'content' => 'Ok'
+        ], $httpContext);
+    }
+
+    public function test_build_stream_with_invalid_uri_throws()
+    {
+        $streamHandler = new StreamContextHandler;
+        $reflection = new \ReflectionClass($streamHandler);
+
+        $method = $reflection->getMethod('buildStream');
+        $method->setAccessible(true);
+
+        $request = new Request("get", "http://localhost:1");
+
+        $this->expectException(RequestException::class);
+        $stream = $method->invokeArgs($streamHandler, [$request, ['http' => ['method' => 'GET', 'protocol_version' => '1.1']]]);
+    }
+
     public function test_execute_request()
     {
         $streamHandler = new StreamContextHandler;