mirror of
https://github.com/danog/amp.git
synced 2024-12-11 17:09:40 +01:00
c12828081f
This has been an edge case potentially hiding some exceptions. The tests have been refactored to error if the loop has watchers leaking from one test to another test.
105 lines
2.9 KiB
PHP
105 lines
2.9 KiB
PHP
<?php
|
|
|
|
namespace Amp\Test;
|
|
|
|
use Amp\Delayed;
|
|
use Amp\Iterator;
|
|
use Amp\Loop;
|
|
use Amp\PHPUnit\TestException;
|
|
use Amp\Producer;
|
|
|
|
class MergeTest extends BaseTest
|
|
{
|
|
public function getArrays()
|
|
{
|
|
return [
|
|
[[\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]],
|
|
];
|
|
}
|
|
|
|
/**
|
|
* @dataProvider getArrays
|
|
*
|
|
* @param array $iterators
|
|
* @param array $expected
|
|
*/
|
|
public function testMerge(array $iterators, array $expected)
|
|
{
|
|
Loop::run(function () use ($iterators, $expected) {
|
|
$iterators = \array_map(function (array $iterator): Iterator {
|
|
return Iterator\fromIterable($iterator);
|
|
}, $iterators);
|
|
|
|
$iterator = Iterator\merge($iterators);
|
|
|
|
while (yield $iterator->advance()) {
|
|
$this->assertSame(\array_shift($expected), $iterator->getCurrent());
|
|
}
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @depends testMerge
|
|
*/
|
|
public function testMergeWithDelayedEmits()
|
|
{
|
|
Loop::run(function () {
|
|
$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);
|
|
}
|
|
});
|
|
|
|
$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
|
|
*/
|
|
public function testMergeWithFailedIterator()
|
|
{
|
|
Loop::run(function () {
|
|
$exception = new TestException;
|
|
$producer = new Producer(function (callable $emit) use ($exception) {
|
|
yield $emit(1); // Emit once before failing.
|
|
throw $exception;
|
|
});
|
|
|
|
$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);
|
|
}
|
|
});
|
|
}
|
|
|
|
/**
|
|
* @expectedException \TypeError
|
|
*/
|
|
public function testNonIterator()
|
|
{
|
|
Iterator\merge([1]);
|
|
}
|
|
}
|