52 lines
1.6 KiB
PHP
52 lines
1.6 KiB
PHP
<?php
|
|
|
|
namespace WellRESTed\Test\Unit\Routing;
|
|
|
|
use Prophecy\Argument;
|
|
use WellRESTed\Routing\ResponsePrep\HeadPrep;
|
|
|
|
/**
|
|
* @covers WellRESTed\Routing\ResponsePrep\HeadPrep
|
|
* @uses WellRESTed\Message\NullStream
|
|
*/
|
|
class HeadPrepTest extends \PHPUnit_Framework_TestCase
|
|
{
|
|
private $request;
|
|
private $response;
|
|
private $body;
|
|
|
|
public function setUp()
|
|
{
|
|
parent::setUp();
|
|
$this->body = $this->prophesize('Psr\Http\Message\StreamInterface');
|
|
$this->body->getSize()->willReturn(1024);
|
|
$this->request = $this->prophesize('Psr\Http\Message\ServerRequestInterface');
|
|
$this->response = $this->prophesize('Psr\Http\Message\ResponseInterface');
|
|
$this->response->getBody()->willReturn($this->body->reveal());
|
|
$this->response->withBody(Argument::any())->will(function ($args) {
|
|
$this->getBody()->willReturn($args[0]);
|
|
return $this;
|
|
});
|
|
}
|
|
|
|
public function testReplacesBodyForHeadRequest()
|
|
{
|
|
$this->request->getMethod()->willReturn("HEAD");
|
|
$request = $this->request->reveal();
|
|
$response = $this->response->reveal();
|
|
$prep = new HeadPrep();
|
|
$prep->dispatch($request, $response);
|
|
$this->assertSame(0, $response->getBody()->getSize());
|
|
}
|
|
|
|
public function testDoesNotReplaceBodyForNonHeadRequests()
|
|
{
|
|
$this->request->getMethod()->willReturn("GET");
|
|
$request = $this->request->reveal();
|
|
$response = $this->response->reveal();
|
|
$prep = new HeadPrep();
|
|
$prep->dispatch($request, $response);
|
|
$this->response->withBody(Argument::any())->shouldNotHaveBeenCalled();
|
|
}
|
|
}
|