<?php

namespace Amp\ByteStream\Test;

use Amp\Parser\InvalidDelimiterError;
use Amp\Parser\Parser;
use PHPUnit\Framework\TestCase;

class ParserTest extends TestCase {
    public function testIntDelimiter() {
        $parser = new Parser((function () use (&$value) {
            $value = yield 6;
        })());

        $parser->push("foobarfoo\r\n");

        $this->assertSame("foobar", $value);
    }

    public function testStringDelimiter() {
        $parser = new Parser((function () use (&$value1, &$value2) {
            $value1 = yield "bar";
            $value2 = yield "\r\n";
        })());

        $parser->push("foobarbaz\r\n");

        $this->assertSame("foo", $value1);
        $this->assertSame("baz", $value2);
    }

    public function testUndelimited() {
        $parser = new Parser((function () use (&$value) {
            $value = yield;
        })());

        $parser->push("foobarbaz\r\n");

        $this->assertSame("foobarbaz\r\n", $value);
    }

    public function testEndedGeneratorThrows() {
        $parser = new Parser((function () {
            if (false) {
                yield;
            }
        })());

        $this->expectException(\Error::class);
        $this->expectExceptionMessage("The parser is no longer writable");

        $parser->push("test");
    }

    public function testThrowingGeneratorEndsWhenDirectlyThrowing() {
        $this->expectException(\RuntimeException::class);

        new Parser((function () {
            if (false) {
                yield;
            }

            throw new \RuntimeException;
        })());
    }

    public function testThrowingGeneratorEndsWhenThrowingLater() {
        $parser = new Parser((function () {
            yield 3;

            throw new \RuntimeException;
        })());

        $this->expectException(\RuntimeException::class);

        $parser->push("abc");
    }

    public function testLengthDelimiterPartialPush() {
        $ok = false;

        $parser = new Parser((function () use (&$ok) {
            yield 6;
            $ok = true;
        })());

        $parser->push("abc\r\n");
        $this->assertFalse($ok);

        $parser->push("x");
        $this->assertTrue($ok);
    }

    public function testThrowsOnInvalidYield() {
        $this->expectException(InvalidDelimiterError::class);

        new Parser((function () {
            yield true;
        })());
    }

    public function testThrowsOnLaterInvalidYield() {
        $parser = new Parser((function () {
            yield 3;
            yield true;
        })());

        $this->expectException(InvalidDelimiterError::class);

        $parser->push("abcd");
    }

    public function testCancelReturnsInternalBuffer() {
        $parser = new Parser((function () {
            yield 3;
        })());

        $parser->push("abcd");

        $this->assertSame("d", $parser->cancel());
    }
}