mirror of
https://github.com/danog/amp.git
synced 2024-12-12 09:29:45 +01:00
134 lines
3.5 KiB
PHP
134 lines
3.5 KiB
PHP
<?php
|
|
|
|
namespace Amp\Test\Iterator;
|
|
|
|
use Amp\Delayed;
|
|
use Amp\Failure;
|
|
use Amp\Iterator;
|
|
use Amp\PHPUnit\AsyncTestCase;
|
|
use Amp\PHPUnit\TestException;
|
|
use Amp\Success;
|
|
|
|
class FromIterableTest extends AsyncTestCase
|
|
{
|
|
const TIMEOUT = 10;
|
|
|
|
public function testSuccessfulPromises(): \Generator
|
|
{
|
|
$expected = \range(1, 3);
|
|
$iterator = Iterator\fromIterable([new Success(1), new Success(2), new Success(3)]);
|
|
|
|
while (yield $iterator->advance()) {
|
|
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
|
|
}
|
|
}
|
|
|
|
public function testFailedPromises(): \Generator
|
|
{
|
|
$exception = new \Exception;
|
|
$iterator = Iterator\fromIterable([new Failure($exception), new Failure($exception)]);
|
|
|
|
try {
|
|
yield $iterator->advance();
|
|
} catch (\Exception $reason) {
|
|
$this->assertSame($exception, $reason);
|
|
}
|
|
}
|
|
|
|
public function testMixedPromises(): \Generator
|
|
{
|
|
$exception = new TestException;
|
|
$expected = \range(1, 2);
|
|
$iterator = Iterator\fromIterable([new Success(1), new Success(2), new Failure($exception), new Success(4)]);
|
|
|
|
try {
|
|
while (yield $iterator->advance()) {
|
|
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
|
|
}
|
|
$this->fail("A failed promise in the iterable should fail the iterator and be thrown from advance()");
|
|
} catch (TestException $reason) {
|
|
$this->assertSame($exception, $reason);
|
|
}
|
|
|
|
$this->assertEmpty($expected);
|
|
}
|
|
|
|
public function testPendingPromises(): \Generator
|
|
{
|
|
$expected = \range(1, 4);
|
|
$iterator = Iterator\fromIterable([new Delayed(30, 1), new Delayed(10, 2), new Delayed(20, 3), new Success(4)]);
|
|
|
|
while (yield $iterator->advance()) {
|
|
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
|
|
}
|
|
}
|
|
|
|
public function testTraversable(): \Generator
|
|
{
|
|
$expected = \range(1, 4);
|
|
$generator = (function () {
|
|
foreach (\range(1, 4) as $value) {
|
|
yield $value;
|
|
}
|
|
})();
|
|
|
|
$iterator = Iterator\fromIterable($generator);
|
|
|
|
while (yield $iterator->advance()) {
|
|
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
|
|
}
|
|
|
|
$this->assertEmpty($expected);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider provideInvalidIteratorArguments
|
|
*/
|
|
public function testInvalid($arg): void
|
|
{
|
|
$this->expectException(\TypeError::class);
|
|
|
|
Iterator\fromIterable($arg);
|
|
}
|
|
|
|
public function provideInvalidIteratorArguments(): array
|
|
{
|
|
return [
|
|
[null],
|
|
[new \stdClass],
|
|
[32],
|
|
[false],
|
|
[true],
|
|
["string"],
|
|
];
|
|
}
|
|
|
|
public function testInterval(): \Generator
|
|
{
|
|
$count = 3;
|
|
$iterator = Iterator\fromIterable(\range(1, $count), self::TIMEOUT);
|
|
|
|
$i = 0;
|
|
while (yield $iterator->advance()) {
|
|
$this->assertSame(++$i, $iterator->getCurrent());
|
|
}
|
|
|
|
$this->assertSame($count, $i);
|
|
}
|
|
|
|
/**
|
|
* @depends testInterval
|
|
*/
|
|
public function testSlowConsumer(): \Generator
|
|
{
|
|
$count = 5;
|
|
$iterator = Iterator\fromIterable(\range(1, $count), self::TIMEOUT);
|
|
|
|
for ($i = 0; yield $iterator->advance(); ++$i) {
|
|
yield new Delayed(self::TIMEOUT * 2);
|
|
}
|
|
|
|
$this->assertSame($count, $i);
|
|
}
|
|
}
|