ArrayParserTest.php

<?php

declare(strict_types=1);

namespace Rector\Tests\BetterPhpDocParser\PhpDocParser\StaticDoctrineAnnotationParser;

use Iterator;
use PhpParser\Node\Scalar\String_;
use PHPUnit\Framework\Attributes\DataProvider;
use Rector\BetterPhpDocParser\PhpDoc\ArrayItemNode;
use Rector\BetterPhpDocParser\PhpDoc\StringNode;
use Rector\BetterPhpDocParser\PhpDocInfo\TokenIteratorFactory;
use Rector\BetterPhpDocParser\PhpDocParser\StaticDoctrineAnnotationParser\ArrayParser;
use Rector\Testing\PHPUnit\AbstractLazyTestCase;

final class ArrayParserTest extends AbstractLazyTestCase
{
    private ArrayParser $arrayParser;

    private TokenIteratorFactory $tokenIteratorFactory;

    protected function setUp(): void
    {
        parent::setUp();

        $this->arrayParser = $this->make(ArrayParser::class);
        $this->tokenIteratorFactory = $this->make(TokenIteratorFactory::class);
    }

    /**
     * @param ArrayItemNode[] $expectedArrayItemNodes
     */
    #[DataProvider('provideData')]
    public function test(string $docContent, array $expectedArrayItemNodes): void
    {
        $betterTokenIterator = $this->tokenIteratorFactory->create($docContent);

        $string = new String_('some_node');

        $arrayItemNodes = $this->arrayParser->parseCurlyArray($betterTokenIterator, $string);
        $this->assertEquals($expectedArrayItemNodes, $arrayItemNodes);
    }

    /**
     * @return Iterator<array<int, (array<int, ArrayItemNode>|string)>>
     */
    public static function provideData(): Iterator
    {
        yield ['{key: "value"}', [new ArrayItemNode(new StringNode('value'), 'key')]];

        yield ['{"key": "value"}', [new ArrayItemNode(new StringNode('value'), new StringNode('key'))]];

        yield ['{"value", "value2"}', [
            new ArrayItemNode(new StringNode('value')),
            new ArrayItemNode(new StringNode('value2')),
        ]];
    }
}