guzzle/tests/Guzzle/Tests/Http/Subscriber/PrepareRequestBodyTest.php

<?php

namespace Guzzle\Tests\Http\Message;

use Guzzle\Http\Adapter\Transaction;
use Guzzle\Http\Client;
use Guzzle\Http\Event\RequestBeforeSendEvent;
use Guzzle\Http\Subscriber\PrepareRequestBody;
use Guzzle\Http\Message\Request;
use Guzzle\Stream\NoSeekStream;
use Guzzle\Stream\Stream;

/**
 * @covers Guzzle\Http\Subscriber\PrepareRequestBody
 */
class PrepareRequestBodyTest extends \PHPUnit_Framework_TestCase
{
    public function testIgnoresRequestsWithNoBody()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
        $this->assertFalse($t->getRequest()->hasHeader('Expect'));
    }

    public function testAppliesPostBody()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $p = $this->getMockBuilder('Guzzle\Http\Message\Post\PostBody')
            ->setMethods(['applyRequestHeaders'])
            ->getMockForAbstractClass();
        $p->expects($this->once())
            ->method('applyRequestHeaders');
        $t->getRequest()->setBody($p);
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
    }

    public function testAddsExpectHeaderWithTrue()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $t->getRequest()->getConfig()->set('expect', true);
        $t->getRequest()->setBody(Stream::factory('foo', true));
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
        $this->assertEquals('100-Continue', $t->getRequest()->getHeader('Expect'));
    }

    public function testAddsExpectHeaderBySize()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $t->getRequest()->getConfig()->set('expect', 2);
        $t->getRequest()->setBody(Stream::factory('foo', true));
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
        $this->assertTrue($t->getRequest()->hasHeader('Expect'));
    }

    public function testDoesNotAddExpectHeaderBySize()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $t->getRequest()->getConfig()->set('expect', 10);
        $t->getRequest()->setBody(Stream::factory('foo', true));
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
        $this->assertFalse($t->getRequest()->hasHeader('Expect'));
    }

    public function testAddsExpectHeaderForNonSeekable()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $t->getRequest()->setBody(new NoSeekStream(Stream::factory('foo', true)));
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
        $this->assertTrue($t->getRequest()->hasHeader('Expect'));
    }

    public function testRemovesContentLengthWhenSendingWithChunked()
    {
        $s = new PrepareRequestBody();
        $t = $this->getTrans();
        $t->getRequest()->setBody(Stream::factory('foo', true));
        $t->getRequest()->setHeader('Transfer-Encoding', 'chunked');
        $s->onRequestBeforeSend(new RequestBeforeSendEvent($t));
        $this->assertFalse($t->getRequest()->hasHeader('Content-Length'));
    }

    private function getTrans()
    {
        return new Transaction(new Client(), new Request('PUT', '/'));
    }
}