213 lines
6.8 KiB
PHP
213 lines
6.8 KiB
PHP
<?php
|
|
|
|
namespace GuzzleHttp\Tests\Psr7;
|
|
|
|
use GuzzleHttp\Psr7;
|
|
use GuzzleHttp\Psr7\CachingStream;
|
|
use GuzzleHttp\Psr7\Stream;
|
|
|
|
/**
|
|
* @covers GuzzleHttp\Psr7\CachingStream
|
|
*/
|
|
class CachingStreamTest extends BaseTest
|
|
{
|
|
/** @var CachingStream */
|
|
private $body;
|
|
/** @var Stream */
|
|
private $decorated;
|
|
|
|
/**
|
|
* @before
|
|
*/
|
|
public function setUpTest()
|
|
{
|
|
$this->decorated = Psr7\Utils::streamFor('testing');
|
|
$this->body = new CachingStream($this->decorated);
|
|
}
|
|
|
|
/**
|
|
* @after
|
|
*/
|
|
public function tearDownTest()
|
|
{
|
|
$this->decorated->close();
|
|
$this->body->close();
|
|
}
|
|
|
|
public function testUsesRemoteSizeIfAvailable()
|
|
{
|
|
$body = Psr7\Utils::streamFor('test');
|
|
$caching = new CachingStream($body);
|
|
self::assertSame(4, $caching->getSize());
|
|
}
|
|
|
|
public function testUsesRemoteSizeIfNotAvailable()
|
|
{
|
|
$body = new Psr7\PumpStream(function () {
|
|
return 'a';
|
|
});
|
|
$caching = new CachingStream($body);
|
|
self::assertNull($caching->getSize());
|
|
}
|
|
|
|
public function testReadsUntilCachedToByte()
|
|
{
|
|
$this->body->seek(5);
|
|
self::assertSame('n', $this->body->read(1));
|
|
$this->body->seek(0);
|
|
self::assertSame('t', $this->body->read(1));
|
|
}
|
|
|
|
public function testCanSeekNearEndWithSeekEnd()
|
|
{
|
|
$baseStream = Psr7\Utils::streamFor(implode('', range('a', 'z')));
|
|
$cached = new CachingStream($baseStream);
|
|
$cached->seek(-1, SEEK_END);
|
|
self::assertSame(25, $baseStream->tell());
|
|
self::assertSame('z', $cached->read(1));
|
|
self::assertSame(26, $cached->getSize());
|
|
}
|
|
|
|
public function testCanSeekToEndWithSeekEnd()
|
|
{
|
|
$baseStream = Psr7\Utils::streamFor(implode('', range('a', 'z')));
|
|
$cached = new CachingStream($baseStream);
|
|
$cached->seek(0, SEEK_END);
|
|
self::assertSame(26, $baseStream->tell());
|
|
self::assertSame('', $cached->read(1));
|
|
self::assertSame(26, $cached->getSize());
|
|
}
|
|
|
|
public function testCanUseSeekEndWithUnknownSize()
|
|
{
|
|
$baseStream = Psr7\Utils::streamFor('testing');
|
|
$decorated = Psr7\FnStream::decorate($baseStream, [
|
|
'getSize' => function () {
|
|
return null;
|
|
}
|
|
]);
|
|
$cached = new CachingStream($decorated);
|
|
$cached->seek(-1, SEEK_END);
|
|
self::assertSame('g', $cached->read(1));
|
|
}
|
|
|
|
public function testRewindUsesSeek()
|
|
{
|
|
$a = Psr7\Utils::streamFor('foo');
|
|
$d = $this->getMockBuilder('GuzzleHttp\Psr7\CachingStream')
|
|
->setMethods(['seek'])
|
|
->setConstructorArgs([$a])
|
|
->getMock();
|
|
$d->expects(self::once())
|
|
->method('seek')
|
|
->with(0)
|
|
->will(self::returnValue(true));
|
|
$d->seek(0);
|
|
}
|
|
|
|
public function testCanSeekToReadBytes()
|
|
{
|
|
self::assertSame('te', $this->body->read(2));
|
|
$this->body->seek(0);
|
|
self::assertSame('test', $this->body->read(4));
|
|
self::assertSame(4, $this->body->tell());
|
|
$this->body->seek(2);
|
|
self::assertSame(2, $this->body->tell());
|
|
$this->body->seek(2, SEEK_CUR);
|
|
self::assertSame(4, $this->body->tell());
|
|
self::assertSame('ing', $this->body->read(3));
|
|
}
|
|
|
|
public function testCanSeekToReadBytesWithPartialBodyReturned()
|
|
{
|
|
$stream = fopen('php://temp', 'r+');
|
|
fwrite($stream, 'testing');
|
|
fseek($stream, 0);
|
|
|
|
$this->decorated = $this->getMockBuilder('\GuzzleHttp\Psr7\Stream')
|
|
->setConstructorArgs([$stream])
|
|
->setMethods(['read'])
|
|
->getMock();
|
|
|
|
$this->decorated->expects(self::exactly(2))
|
|
->method('read')
|
|
->willReturnCallback(function ($length) use ($stream) {
|
|
return fread($stream, 2);
|
|
});
|
|
|
|
$this->body = new CachingStream($this->decorated);
|
|
|
|
self::assertSame(0, $this->body->tell());
|
|
$this->body->seek(4, SEEK_SET);
|
|
self::assertSame(4, $this->body->tell());
|
|
|
|
$this->body->seek(0);
|
|
self::assertSame('test', $this->body->read(4));
|
|
}
|
|
|
|
public function testWritesToBufferStream()
|
|
{
|
|
$this->body->read(2);
|
|
$this->body->write('hi');
|
|
$this->body->seek(0);
|
|
self::assertSame('tehiing', (string) $this->body);
|
|
}
|
|
|
|
public function testSkipsOverwrittenBytes()
|
|
{
|
|
$decorated = Psr7\Utils::streamFor(
|
|
implode("\n", array_map(function ($n) {
|
|
return str_pad($n, 4, '0', STR_PAD_LEFT);
|
|
}, range(0, 25)))
|
|
);
|
|
|
|
$body = new CachingStream($decorated);
|
|
|
|
self::assertSame("0000\n", Psr7\Utils::readLine($body));
|
|
self::assertSame("0001\n", Psr7\Utils::readLine($body));
|
|
// Write over part of the body yet to be read, so skip some bytes
|
|
self::assertSame(5, $body->write("TEST\n"));
|
|
self::assertSame(5, Helpers::readObjectAttribute($body, 'skipReadBytes'));
|
|
// Read, which skips bytes, then reads
|
|
self::assertSame("0003\n", Psr7\Utils::readLine($body));
|
|
self::assertSame(0, Helpers::readObjectAttribute($body, 'skipReadBytes'));
|
|
self::assertSame("0004\n", Psr7\Utils::readLine($body));
|
|
self::assertSame("0005\n", Psr7\Utils::readLine($body));
|
|
|
|
// Overwrite part of the cached body (so don't skip any bytes)
|
|
$body->seek(5);
|
|
self::assertSame(5, $body->write("ABCD\n"));
|
|
self::assertSame(0, Helpers::readObjectAttribute($body, 'skipReadBytes'));
|
|
self::assertSame("TEST\n", Psr7\Utils::readLine($body));
|
|
self::assertSame("0003\n", Psr7\Utils::readLine($body));
|
|
self::assertSame("0004\n", Psr7\Utils::readLine($body));
|
|
self::assertSame("0005\n", Psr7\Utils::readLine($body));
|
|
self::assertSame("0006\n", Psr7\Utils::readLine($body));
|
|
self::assertSame(5, $body->write("1234\n"));
|
|
self::assertSame(5, Helpers::readObjectAttribute($body, 'skipReadBytes'));
|
|
|
|
// Seek to 0 and ensure the overwritten bit is replaced
|
|
$body->seek(0);
|
|
self::assertSame("0000\nABCD\nTEST\n0003\n0004\n0005\n0006\n1234\n0008\n0009\n", $body->read(50));
|
|
|
|
// Ensure that casting it to a string does not include the bit that was overwritten
|
|
$this->assertStringContainsStringGuzzle("0000\nABCD\nTEST\n0003\n0004\n0005\n0006\n1234\n0008\n0009\n", (string) $body);
|
|
}
|
|
|
|
public function testClosesBothStreams()
|
|
{
|
|
$s = fopen('php://temp', 'r');
|
|
$a = Psr7\Utils::streamFor($s);
|
|
$d = new CachingStream($a);
|
|
$d->close();
|
|
self::assertFalse(is_resource($s));
|
|
}
|
|
|
|
public function testEnsuresValidWhence()
|
|
{
|
|
$this->expectExceptionGuzzle('InvalidArgumentException');
|
|
|
|
$this->body->seek(10, -123456);
|
|
}
|
|
}
|