amp/test/Stream/ConcatTest.php

<?php

namespace Amp\Test\Stream;

use Amp\AsyncGenerator;
use Amp\PHPUnit\AsyncTestCase;
use Amp\PHPUnit\TestException;
use Amp\Stream;

class ConcatTest extends AsyncTestCase
{
    public function getArrays(): array
    {
        return [
            [[\range(1, 3), \range(4, 6)], \range(1, 6)],
            [[\range(1, 5), \range(6, 8)], \range(1, 8)],
            [[\range(1, 4), \range(5, 10)], \range(1, 10)],
        ];
    }

    /**
     * @dataProvider getArrays
     *
     * @param array $iterators
     * @param array $expected
     */
    public function testConcat(array $iterators, array $expected)
    {
        $iterators = \array_map(static function (array $iterator): Stream {
            return Stream\fromIterable($iterator);
        }, $iterators);

        $stream = Stream\concat($iterators);

        while (list($value) = yield $stream->continue()) {
            $this->assertSame(\array_shift($expected), $value);
        }
    }

    /**
     * @depends testConcat
     */
    public function testConcatWithFailedStream()
    {
        $exception = new TestException;
        $expected = \range(1, 6);
        $generator = new AsyncGenerator(static function (callable $yield) use ($exception) {
            yield $yield(6); // Emit once before failing.
            throw $exception;
        });

        $stream = Stream\concat([
            Stream\fromIterable(\range(1, 5)),
            $generator,
            Stream\fromIterable(\range(7, 10)),
        ]);

        try {
            while (list($value) = yield $stream->continue()) {
                $this->assertSame(\array_shift($expected), $value);
            }

            $this->fail("The exception used to fail the stream should be thrown from continue()");
        } catch (TestException $reason) {
            $this->assertSame($exception, $reason);
        }

        $this->assertEmpty($expected);
    }

    public function testNonStream()
    {
        $this->expectException(\TypeError::class);

        /** @noinspection PhpParamsInspection */
        Stream\concat([1]);
    }
}