1
0
mirror of https://github.com/danog/amp.git synced 2024-12-12 09:29:45 +01:00
amp/test/Pipeline/FromIterableTest.php

137 lines
3.4 KiB
PHP
Raw Normal View History

<?php
2020-08-23 16:18:28 +02:00
namespace Amp\Test\Pipeline;
use Amp\Delayed;
use Amp\Failure;
2020-05-17 21:41:42 +02:00
use Amp\PHPUnit\AsyncTestCase;
use Amp\PHPUnit\TestException;
2020-08-23 16:18:28 +02:00
use Amp\Pipeline;
use Amp\Success;
2020-05-17 21:41:42 +02:00
class FromIterableTest extends AsyncTestCase
2018-06-18 20:00:01 +02:00
{
const TIMEOUT = 10;
2018-06-18 20:00:01 +02:00
public function testSuccessfulPromises()
{
2020-05-17 21:41:42 +02:00
$expected = \range(1, 3);
2020-08-23 16:18:28 +02:00
$pipeline = Pipeline\fromIterable([new Success(1), new Success(2), new Success(3)]);
2017-01-06 23:16:06 +01:00
2020-08-23 16:18:28 +02:00
while (null !== $value = yield $pipeline->continue()) {
2020-05-21 17:11:22 +02:00
$this->assertSame(\array_shift($expected), $value);
2020-05-17 21:41:42 +02:00
}
}
2017-01-06 23:16:06 +01:00
2018-06-18 20:00:01 +02:00
public function testFailedPromises()
{
2020-05-17 21:41:42 +02:00
$exception = new \Exception;
2020-08-23 16:18:28 +02:00
$iterator = Pipeline\fromIterable([new Failure($exception), new Failure($exception)]);
2020-05-17 21:41:42 +02:00
$this->expectExceptionObject($exception);
yield $iterator->continue();
}
2017-01-06 23:16:06 +01:00
2018-06-18 20:00:01 +02:00
public function testMixedPromises()
{
2020-05-17 21:41:42 +02:00
$exception = new TestException;
$expected = \range(1, 2);
2020-08-23 16:18:28 +02:00
$pipeline = Pipeline\fromIterable([new Success(1), new Success(2), new Failure($exception), new Success(4)]);
2020-05-17 21:41:42 +02:00
try {
2020-08-23 16:18:28 +02:00
while (null !== $value = yield $pipeline->continue()) {
2020-05-21 17:11:22 +02:00
$this->assertSame(\array_shift($expected), $value);
2017-04-27 17:32:53 +02:00
}
2020-08-23 16:18:28 +02:00
$this->fail("A failed promise in the iterable should fail the pipeline and be thrown from continue()");
2020-05-17 21:41:42 +02:00
} catch (TestException $reason) {
$this->assertSame($exception, $reason);
}
2017-01-06 23:16:06 +01:00
2020-05-17 21:41:42 +02:00
$this->assertEmpty($expected);
}
2017-01-06 23:16:06 +01:00
2018-06-18 20:00:01 +02:00
public function testPendingPromises()
{
2020-05-17 21:41:42 +02:00
$expected = \range(1, 4);
2020-08-23 16:18:28 +02:00
$pipeline = Pipeline\fromIterable([
2020-05-17 21:41:42 +02:00
new Delayed(30, 1),
new Delayed(10, 2),
new Delayed(20, 3),
new Success(4),
]);
2020-08-23 16:18:28 +02:00
while (null !== $value = yield $pipeline->continue()) {
2020-05-21 17:11:22 +02:00
$this->assertSame(\array_shift($expected), $value);
2020-05-17 21:41:42 +02:00
}
}
2017-01-06 23:16:06 +01:00
2018-06-18 20:00:01 +02:00
public function testTraversable()
{
2020-05-17 21:41:42 +02:00
$expected = \range(1, 4);
$generator = (static function () {
foreach (\range(1, 4) as $value) {
yield $value;
}
})();
2017-01-06 23:16:06 +01:00
2020-08-23 16:18:28 +02:00
$pipeline = Pipeline\fromIterable($generator);
2017-01-06 23:16:06 +01:00
2020-08-23 16:18:28 +02:00
while (null !== $value = yield $pipeline->continue()) {
2020-05-21 17:11:22 +02:00
$this->assertSame(\array_shift($expected), $value);
2020-05-17 21:41:42 +02:00
}
2017-01-06 23:16:06 +01:00
2020-05-17 21:41:42 +02:00
$this->assertEmpty($expected);
2017-01-04 02:10:27 +01:00
}
2017-01-06 23:16:06 +01:00
/**
* @dataProvider provideInvalidIteratorArguments
2017-01-06 23:16:06 +01:00
*/
2018-06-18 20:00:01 +02:00
public function testInvalid($arg)
{
2020-05-13 17:15:21 +02:00
$this->expectException(\TypeError::class);
2020-08-23 16:18:28 +02:00
Pipeline\fromIterable($arg);
2017-01-06 23:16:06 +01:00
}
2020-05-17 21:41:42 +02:00
public function provideInvalidIteratorArguments(): array
2018-06-18 20:00:01 +02:00
{
2017-01-06 23:16:06 +01:00
return [
[null],
[new \stdClass],
[32],
[false],
[true],
["string"],
];
}
2018-06-18 20:00:01 +02:00
public function testInterval()
{
2020-05-17 21:41:42 +02:00
$count = 3;
2020-08-23 16:18:28 +02:00
$pipeline = Pipeline\fromIterable(\range(1, $count), self::TIMEOUT);
2020-05-17 21:41:42 +02:00
$i = 0;
2020-08-23 16:18:28 +02:00
while (null !== $value = yield $pipeline->continue()) {
2020-05-21 17:11:22 +02:00
$this->assertSame(++$i, $value);
2020-05-17 21:41:42 +02:00
}
2020-05-17 21:41:42 +02:00
$this->assertSame($count, $i);
}
/**
* @depends testInterval
*/
2018-06-18 20:00:01 +02:00
public function testSlowConsumer()
{
$count = 5;
2020-08-23 16:18:28 +02:00
$pipeline = Pipeline\fromIterable(\range(1, $count), self::TIMEOUT);
2020-08-23 16:18:28 +02:00
for ($i = 0; $value = yield $pipeline->continue(); ++$i) {
2020-05-17 21:41:42 +02:00
yield new Delayed(self::TIMEOUT * 2);
}
2020-05-17 21:41:42 +02:00
$this->assertSame($count, $i);
}
}