packages/sql-faker/tests/Unit/MySql/Generation/Rewrite/LoadSourceCountRuleTest.php

1<?php
2
3declare(strict_types=1);
4
5namespace Tests\Unit\SqlFaker\MySql\Generation\Rewrite;
6
7use PHPUnit\Framework\Attributes\CoversClass;
8use PHPUnit\Framework\Attributes\UsesClass;
9use PHPUnit\Framework\TestCase;
10use SqlFaker\Generation\Derivation\DerivationTrace;
11use SqlFaker\Grammar\Model\NonTerminal;
12use SqlFaker\Grammar\Model\Production;
13use SqlFaker\Grammar\Model\Terminal;
14use SqlFaker\MySql\Generation\Rewrite\LoadSourceCountRule;
15
16#[CoversClass(LoadSourceCountRule::class)]
17#[UsesClass(DerivationTrace::class)]
18#[UsesClass(NonTerminal::class)]
19#[UsesClass(Production::class)]
20#[UsesClass(Terminal::class)]
21#[UsesClass(\SqlFaker\Generation\Token\ProductionOccurrence::class)]
22#[UsesClass(\SqlFaker\Generation\Token\TerminalOccurrence::class)]
23#[UsesClass(\SqlFaker\Generation\Token\TerminalSequence::class)]
24final class LoadSourceCountRuleTest extends TestCase
25{
26    public function testRewriteConstrainsOnlyTheCountClause(): void
27    {
28        $trace = new DerivationTrace('root');
29        $trace->expand(0, new Production([new Terminal('NUM'), new NonTerminal('opt_source_count')]), 0);
30        $trace->expand(1, new Production([new NonTerminal('IDENT_sys'), new Terminal('NUM')]), 0);
31        $trace->expand(1, new Production([new Terminal('IDENT')]), 0);
32        $input = $trace->terminals();
33        $result = (new LoadSourceCountRule())->rewrite($input);
34        self::assertSame(['NUM', 'LOAD_COUNT_NAME', 'LOAD_SOURCE_COUNT'], $result->names());
35        self::assertSame($input->terminals[1]->id, $result->terminals[1]->id);
36        self::assertSame($input->original, $result->original);
37        self::assertCount(count($result->terminals), array_unique(array_map(static fn ($terminal): int => $terminal->id, $result->terminals)));
38    }
39}
40