Merge pull request #9 from Ricorocks-Digital-Agency/response-transfor…

…mation Adds a set method to transform a Response object
Ricorocks-Digital-Agency · Dec 7, 2020 · e049ad3 · e049ad3
2 parents 87a7918 + 23d6d08
commit e049ad3
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 0 deletions.
diff --git a/src/Response/Response.php b/src/Response/Response.php
@@ -24,4 +24,10 @@ public function withXml($xmlRequest, $xmlResponse)
         $this->xmlResponse = $xmlResponse;
         return $this;
     }
+
+    public function set($key, $value): self
+    {
+        data_set($this->response, $key, $value);
+        return $this;
+    }
 }
diff --git a/tests/Hooks/GlobalHooksTest.php b/tests/Hooks/GlobalHooksTest.php
@@ -58,6 +58,18 @@ public function the_beforeRequesting_hooks_can_transform_the_request_object()
         Soap::assertSent(fn($request) => $request->getBody()['hello'] === ['world' => ['foo', 'bar'], 'person' => 'Richard']);
     }
 
+    /** @test */
+    public function the_afterRequesting_hooks_can_transform_the_response_object()
+    {
+        Soap::fake();
+
+        Soap::afterRequesting(fn($request, Response $response) => $response->set('hello.world', ['foo', 'bar']));
+        Soap::afterRequesting(fn($request, Response $response) => $response->set('hello.person', 'Richard'));
+        Soap::to('http://endpoint.com')->Test();
+
+        Soap::assertSent(fn($request, Response $response) => $response->response['hello'] === ['world' => ['foo', 'bar'], 'person' => 'Richard']);
+    }
+
     protected function tearDown(): void
     {
         parent::tearDown();

diff --git a/tests/Hooks/LocalHooksTest.php b/tests/Hooks/LocalHooksTest.php
@@ -92,6 +92,19 @@ public function the_beforeRequesting_hooks_can_transform_the_request_object()
         Soap::assertSent(fn($request) => $request->getBody()['hello'] === ['world' => ['foo', 'bar'], 'person' => 'Richard']);
     }
 
+    /** @test */
+    public function the_afterRequesting_hooks_can_transform_the_response_object()
+    {
+        Soap::fake();
+
+        Soap::to('http://endpoint.com')
+            ->afterRequesting(fn($request, Response $response) => $response->set('hello.world', ['foo', 'bar']))
+            ->afterRequesting(fn($request, Response $response) => $response->set('hello.person', 'Richard'))
+            ->Test();
+
+        Soap::assertSent(fn($request, Response $response) => $response->response['hello'] === ['world' => ['foo', 'bar'], 'person' => 'Richard']);
+    }
+
     protected function setUp(): void
     {
         parent::setUp();