<?php declare(strict_types=1);

namespace Rector\Tests\PhpParser\Printer;

use Iterator;
use PhpParser\Node\Expr\Yield_;
use PhpParser\Node\Scalar\String_;
use Rector\PhpParser\Printer\BetterStandardPrinter;
use Rector\Tests\AbstractContainerAwareTestCase;

final class BetterStandardPrinterTest extends AbstractContainerAwareTestCase
{
    /**
     * @var BetterStandardPrinter
     */
    private $betterStandardPrinter;

    protected function setUp(): void
    {
        $this->betterStandardPrinter = $this->container->get(BetterStandardPrinter::class);
    }

    /**
     * @dataProvider provideDataForDoubleSlashEscaping()
     */
    public function testDoubleSlashEscaping(string $content, string $expectedOutput): void
    {
        $printed = $this->betterStandardPrinter->prettyPrint([new String_($content)]);
        $this->assertSame($expectedOutput, $printed);
    }

    public function provideDataForDoubleSlashEscaping(): Iterator
    {
        yield ['Vendor\Name', "'Vendor\Name'"];
        yield ['Vendor\\', "'Vendor\\\\'"];
        yield ['Vendor\'Name', "'Vendor\'Name'"];
    }

    public function testYield(): void
    {
        $printed = $this->betterStandardPrinter->prettyPrint([new Yield_(new String_('value'))]);
        $this->assertSame("yield 'value'", $printed);

        $printed = $this->betterStandardPrinter->prettyPrint([new Yield_()]);
        $this->assertSame('yield', $printed);
    }
}