1
0
mirror of https://github.com/danog/amp.git synced 2024-12-03 09:57:51 +01:00
amp/test/YielderTraitTest.php
2020-09-24 12:53:28 -05:00

225 lines
5.6 KiB
PHP

<?php
namespace Amp\Test;
use Amp\DisposedException;
use Amp\Loop;
use Amp\Promise;
use Amp\Stream;
use Amp\Success;
use PHPUnit\Framework\TestCase;
class Yielder implements Stream
{
use \Amp\Internal\Yielder {
createStream as public;
}
}
class YielderTraitTest extends TestCase
{
/** @var Yielder */
private $source;
public function setUp()
{
$this->source = new Yielder;
}
public function testYield()
{
Loop::run(function () {
$value = 'Yielded Value';
$promise = $this->source->yield($value);
$stream = $this->source->createStream();
$this->assertSame([$value, 0], yield $stream->continue());
$this->assertInstanceOf(Promise::class, $promise);
$this->assertNull(yield $promise);
});
}
/**
* @depends testYield
*/
public function testYieldAfterComplete()
{
$this->expectException(\Error::class);
$this->expectExceptionMessage('Streams cannot yield values after calling complete');
$this->source->complete();
$this->source->yield(1);
}
/**
* @depends testYield
*/
public function testYieldingNull()
{
Loop::run(function () {
$this->source->yield(null);
$this->assertSame([null, 0], yield $this->source->createStream()->continue());
});
}
/**
* @depends testYield
*/
public function testYieldingPromise()
{
$this->expectException(\Error::class);
$this->expectExceptionMessage('Streams cannot yield promises');
$this->source->yield(new Success);
}
public function testDoubleComplete()
{
$this->expectException(\Error::class);
$this->expectExceptionMessage('Stream has already been completed');
$this->source->complete();
$this->source->complete();
}
public function testDoubleFail()
{
$this->expectException(\Error::class);
$this->expectExceptionMessage('Stream has already been completed');
$this->source->fail(new \Exception);
$this->source->fail(new \Exception);
}
public function testDoubleStart()
{
$stream = $this->source->createStream();
$this->expectException(\Error::class);
$this->expectExceptionMessage('A stream may be started only once');
$stream = $this->source->createStream();
}
public function testYieldAfterContinue()
{
Loop::run(function () {
$value = 'Yielded Value';
$stream = $this->source->createStream();
$promise = $stream->continue();
$this->assertInstanceOf(Promise::class, $promise);
$this->assertNull(yield $this->source->yield($value));
$this->assertSame([$value, 0], yield $promise);
});
}
public function testContinueAfterComplete()
{
Loop::run(function () {
$stream = $this->source->createStream();
$this->source->complete();
$promise = $stream->continue();
$this->assertInstanceOf(Promise::class, $promise);
$this->assertNull(yield $promise);
});
}
public function testContinueAfterFail()
{
Loop::run(function () {
$stream = $this->source->createStream();
$this->source->fail(new \Exception('Stream failed'));
$promise = $stream->continue();
$this->assertInstanceOf(Promise::class, $promise);
$this->expectException(\Exception::class);
$this->expectExceptionMessage('Stream failed');
yield $promise;
});
}
public function testCompleteAfterContinue()
{
Loop::run(function () {
$stream = $this->source->createStream();
$promise = $stream->continue();
$this->assertInstanceOf(Promise::class, $promise);
$this->source->complete();
$this->assertNull(yield $promise);
});
}
public function testDestroyingStreamRelievesBackPressure()
{
$stream = $this->source->createStream();
$invoked = 0;
$onResolved = function () use (&$invoked) {
$invoked++;
};
foreach (\range(1, 5) as $value) {
$promise = $this->source->yield($value);
$promise->onResolve($onResolved);
}
$this->assertSame(0, $invoked);
unset($stream); // Should relieve all back-pressure.
$this->assertSame(5, $invoked);
$this->source->complete(); // Should not throw.
$this->expectException(\Error::class);
$this->expectExceptionMessage('Stream has already been completed');
$this->source->complete(); // Should throw.
}
public function testYieldAfterDisposal()
{
$this->expectException(DisposedException::class);
$this->expectExceptionMessage('The stream has been disposed');
Loop::run(function () {
$stream = $this->source->createStream();
$promise = $this->source->yield(1);
$stream->dispose();
$this->assertNull(yield $promise);
yield $this->source->yield(1);
});
}
public function testYieldAfterDestruct()
{
$this->expectException(DisposedException::class);
$this->expectExceptionMessage('The stream has been disposed');
Loop::run(function () {
$stream = $this->source->createStream();
$promise = $this->source->yield(1);
unset($stream);
$this->assertNull(yield $promise);
yield $this->source->yield(1);
});
}
}