1
0
mirror of https://github.com/danog/amp.git synced 2024-12-11 17:09:40 +01:00
amp/test/Iterator/MergeTest.php

98 lines
2.7 KiB
PHP
Raw Normal View History

<?php
2020-05-13 17:15:21 +02:00
namespace Amp\Test\Iterator;
2017-05-02 07:22:53 +02:00
use Amp\Delayed;
2017-04-27 17:51:06 +02:00
use Amp\Iterator;
2020-09-28 05:19:52 +02:00
use Amp\PHPUnit\AsyncTestCase;
use Amp\PHPUnit\TestException;
2017-03-14 00:52:57 +01:00
use Amp\Producer;
2020-09-28 05:19:52 +02:00
class MergeTest extends AsyncTestCase
2018-06-18 20:00:01 +02:00
{
2020-09-28 05:19:52 +02:00
public function getArrays(): array
2018-06-18 20:00:01 +02:00
{
return [
2017-03-14 00:52:57 +01:00
[[\range(1, 3), \range(4, 6)], [1, 4, 2, 5, 3, 6]],
[[\range(1, 5), \range(6, 8)], [1, 6, 2, 7, 3, 8, 4, 5]],
[[\range(1, 4), \range(5, 10)], [1, 5, 2, 6, 3, 7, 4, 8, 9, 10]],
];
}
/**
2017-03-14 00:52:57 +01:00
* @dataProvider getArrays
*
2017-04-27 17:51:06 +02:00
* @param array $iterators
* @param array $expected
*/
2020-09-28 05:19:52 +02:00
public function testMerge(array $iterators, array $expected): \Generator
2018-06-18 20:00:01 +02:00
{
2020-09-28 05:19:52 +02:00
$iterators = \array_map(function (array $iterator): Iterator {
return Iterator\fromIterable($iterator);
}, $iterators);
2020-09-28 05:19:52 +02:00
$iterator = Iterator\merge($iterators);
2017-03-14 00:52:57 +01:00
2020-09-28 05:19:52 +02:00
while (yield $iterator->advance()) {
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
}
2017-04-27 17:32:53 +02:00
}
/**
* @depends testMerge
*/
2020-09-28 05:19:52 +02:00
public function testMergeWithDelayedEmits(): \Generator
2018-06-18 20:00:01 +02:00
{
2020-09-28 05:19:52 +02:00
$iterators = [];
$values1 = [new Delayed(10, 1), new Delayed(50, 2), new Delayed(70, 3)];
$values2 = [new Delayed(20, 4), new Delayed(40, 5), new Delayed(60, 6)];
$expected = [1, 4, 5, 2, 6, 3];
$iterators[] = new Producer(function (callable $emit) use ($values1) {
foreach ($values1 as $value) {
yield $emit($value);
2017-04-27 17:32:53 +02:00
}
});
2020-09-28 05:19:52 +02:00
$iterators[] = new Producer(function (callable $emit) use ($values2) {
foreach ($values2 as $value) {
yield $emit($value);
}
});
$iterator = Iterator\merge($iterators);
while (yield $iterator->advance()) {
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
}
}
/**
* @depends testMerge
*/
2020-09-28 05:19:52 +02:00
public function testMergeWithFailedIterator(): \Generator
2018-06-18 20:00:01 +02:00
{
2020-09-28 05:19:52 +02:00
$exception = new TestException;
$producer = new Producer(function (callable $emit) use ($exception) {
yield $emit(1); // Emit once before failing.
throw $exception;
});
2020-09-28 05:19:52 +02:00
$iterator = Iterator\merge([$producer, Iterator\fromIterable(\range(1, 5))]);
try {
while (yield $iterator->advance()) ;
$this->fail("The exception used to fail the iterator should be thrown from advance()");
} catch (TestException $reason) {
$this->assertSame($exception, $reason);
}
}
2018-06-18 20:00:01 +02:00
public function testNonIterator()
{
2020-09-28 05:19:52 +02:00
$this->expectException(\TypeError::class);
2017-04-27 17:51:06 +02:00
Iterator\merge([1]);
}
}