amp/test/IntervalTest.php

<?php

namespace Amp\Test;

use Amp;
use Amp\Pause;
use AsyncInterop\Loop;

class IntervalTest extends \PHPUnit_Framework_TestCase {
    const TIMEOUT = 10;

    public function testInterval() {
        $count = 3;
        $stream = Amp\interval(self::TIMEOUT, $count);

        $i = 0;
        $stream = Amp\each($stream, function ($value) use (&$i) {
            $this->assertSame(++$i, $value);
        });

        Amp\wait($stream);

        $this->assertSame($count, $i);
    }

    /**
     * @depends testInterval
     */
    public function testSlowConsumer() {
        $invoked = 0;
        $count = 5;
        Loop::execute(function () use (&$invoked, $count) {
            $stream = Amp\interval(self::TIMEOUT, $count);

            $stream->listen(function () use (&$invoked) {
                ++$invoked;
                return new Pause(self::TIMEOUT * 2);
            });
        });

        $this->assertSame($count, $invoked);
    }

    /**
     * @expectedException \Error
     * @expectedExceptionMessage The number of times to emit must be a positive value
     */
    public function testInvalidCount() {
        $stream = Amp\interval(self::TIMEOUT, -1);
    }
}