1
0
mirror of https://github.com/danog/amp.git synced 2024-12-11 17:09:40 +01:00
amp/test/MergeTest.php
Niklas Keller c12828081f Fix exceptions bubbling from Coroutine::__destruct
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.
2019-05-31 11:38:02 -05:00

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]);
}
}