1212use Exception ;
1313use PHPUnit_Framework_TestCase as TestCase ;
1414use Prophecy \Argument ;
15- use Psr \Http \Message \RequestInterface ;
15+ use Psr \Http \Message \ServerRequestInterface ;
1616use Psr \Http \Message \ResponseInterface ;
1717use Psr \Http \Message \StreamInterface ;
1818use Zend \Expressive \Template \TemplateRendererInterface ;
@@ -30,7 +30,7 @@ public function getTemplateImplementation()
3030
3131 public function getRequest ($ stream )
3232 {
33- $ request = $ this ->prophesize (RequestInterface ::class);
33+ $ request = $ this ->prophesize (ServerRequestInterface ::class);
3434 $ request ->getBody ()->will (function () use ($ stream ) {
3535 return $ stream ->reveal ();
3636 });
@@ -130,6 +130,9 @@ public function testInvocationWithoutErrorAndResponseSameAsOriginalReturns404Res
130130 $ handler ->setOriginalResponse ($ response ->reveal ());
131131
132132 $ request = $ this ->getRequest ($ this ->getStream ());
133+ $ request
134+ ->getAttribute ('originalResponse ' , Argument::that ([$ response , 'reveal ' ]))
135+ ->will ([$ response , 'reveal ' ]);
133136
134137 $ result = $ handler ($ request ->reveal (), $ response ->reveal ());
135138 $ this ->assertSame ($ expected ->reveal (), $ result );
@@ -151,6 +154,9 @@ public function testInvocationWithoutErrorAndResponseSameAsOriginalWithNewBodyCo
151154 $ handler ->setOriginalResponse ($ response ->reveal ());
152155
153156 $ request = $ this ->getRequest ($ this ->getStream ());
157+ $ request
158+ ->getAttribute ('originalResponse ' , Argument::that ([$ response , 'reveal ' ]))
159+ ->will ([$ response , 'reveal ' ]);
154160
155161 $ result = $ handler ($ request ->reveal (), $ response ->reveal ());
156162 $ this ->assertSame ($ response ->reveal (), $ result );
@@ -167,6 +173,9 @@ public function testInvocationWithoutErrorAndResponseDifferentThanOriginalReturn
167173
168174 $ response = $ this ->getResponse ($ this ->getStream ());
169175 $ request = $ this ->getRequest ($ this ->getStream ());
176+ $ request
177+ ->getAttribute ('originalResponse ' , Argument::that ([$ response , 'reveal ' ]))
178+ ->will ([$ response , 'reveal ' ]);
170179
171180 $ result = $ handler ($ request ->reveal (), $ response ->reveal ());
172181 $ this ->assertSame ($ response ->reveal (), $ result );
@@ -243,6 +252,9 @@ public function testInvocationWithoutErrorAndResponseSameAsOriginalCanReturnTemp
243252
244253 $ request = $ this ->getRequest ($ this ->getStream ());
245254 $ request ->getUri ()->shouldBeCalled ();
255+ $ request
256+ ->getAttribute ('originalResponse ' , Argument::that ([$ response , 'reveal ' ]))
257+ ->will ([$ response , 'reveal ' ]);
246258
247259 $ result = $ handler ($ request ->reveal (), $ response ->reveal ());
248260 $ this ->assertSame ($ expected ->reveal (), $ result );
0 commit comments