bee4 / transport

 * WithBodyStreamTrait unit test definition

 * @package Bee4\Test\Transport\Message

 */

class WithBodyStreamTraitTest extends WithBodyTraitTest

{

    /**

     * Test all headers collection manipulation function

     */

    public function testBody($mock = null)

    {

        $mock = $this->getObjectForTrait(

            'Bee4\Transport\Message\WithBodyStreamTrait'

        );

        parent::testBody($mock);

        $stream = tmpfile();

        fwrite($stream, 'toto');

        $mock->setBody($stream);

        $this->assertEquals($stream, $mock->getBody());

        $this->assertTrue($mock->hasBodyStream());

        $this->assertEquals(4, $mock->getBodyLength());

    }

}

 * WithBodyTrait unit test definition

 * @package Bee4\Test\Transport\Message

 */

class WithBodyTraitTest extends \PHPUnit_Framework_TestCase

{

    /**

     * Test all headers collection manipulation function

     */

    public function testBody($mock = null)

    {

        if (null === $mock) {

            $mock = $this->getObjectForTrait('Bee4\Transport\Message\WithBodyTrait');

        }

        $body = 'Just a sample body';

        $this->assertFalse($mock->getBody());

        $mock->setBody($body);

        $this->assertEquals($body, $mock->getBody());

        $this->assertEquals(strlen($body), $mock->getBodyLength());

    }

}