diff --git a/test/src/Doubles/NextMock.php b/test/src/Doubles/NextMock.php index 02015ff..92a4ab0 100644 --- a/test/src/Doubles/NextMock.php +++ b/test/src/Doubles/NextMock.php @@ -10,6 +10,7 @@ class NextMock public $called = false; public $request = null; public $response = null; + public $upstreamResponse = null; public function __invoke( ServerRequestInterface $request, @@ -18,6 +19,10 @@ class NextMock $this->called = true; $this->request = $request; $this->response = $response; - return $response; + if ($this->upstreamResponse) { + return $this->upstreamResponse; + } else { + return $response; + } } }