forked from phpmyadmin/sql-parser
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathWithStatementTest.php
More file actions
115 lines (97 loc) · 3.62 KB
/
WithStatementTest.php
File metadata and controls
115 lines (97 loc) · 3.62 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
<?php
declare(strict_types=1);
namespace PhpMyAdmin\SqlParser\Tests\Parser;
use PhpMyAdmin\SqlParser\Components\WithKeyword;
use PhpMyAdmin\SqlParser\Lexer;
use PhpMyAdmin\SqlParser\Parser;
use PhpMyAdmin\SqlParser\Tests\TestCase;
use stdClass;
class WithStatementTest extends TestCase
{
public function testWith(): void
{
$sql = <<<SQL
WITH categories(identifier, name, parent_id) AS (
SELECT c.identifier, c.name, c.parent_id FROM category c WHERE c.identifier = 'a'
UNION ALL
SELECT c.identifier, c.name, c.parent_id FROM categories, category c WHERE c.identifier = categories.parent_id
), foo AS ( SELECT * FROM test )
SELECT * FROM categories
SQL;
$lexer = new Lexer($sql);
$lexerErrors = $this->getErrorsAsArray($lexer);
$this->assertCount(0, $lexerErrors);
$parser = new Parser($lexer->list);
$parserErrors = $this->getErrorsAsArray($parser);
$this->assertCount(0, $parserErrors);
$this->assertCount(2, $parser->statements);
// phpcs:disable Generic.Files.LineLength.TooLong
$expected = <<<SQL
WITH categories(identifier, name, parent_id) AS (SELECT c.identifier, c.name, c.parent_id FROM category AS `c` WHERE c.identifier = 'a' UNION ALL SELECT c.identifier, c.name, c.parent_id FROM categories, category AS `c` WHERE c.identifier = categories.parent_id), foo AS (SELECT * FROM test)
SQL;
// phpcs:enable
$this->assertEquals($expected, $parser->statements[0]->build());
$this->assertEquals('SELECT * FROM categories', $parser->statements[1]->build());
}
public function testWithHasErrors(): void
{
$sql = <<<SQL
WITH categories(identifier, name, parent_id) AS (
SOMETHING * FROM foo
)
SELECT * FROM categories
SQL;
$lexer = new Lexer($sql);
$lexerErrors = $this->getErrorsAsArray($lexer);
$this->assertCount(0, $lexerErrors);
$parser = new Parser($lexer->list);
$parserErrors = $this->getErrorsAsArray($parser);
$this->assertCount(2, $parserErrors);
}
public function testWithEmbedParenthesis(): void
{
$sql = <<<SQL
WITH categories AS (
SELECT * FROM (SELECT * FROM foo)
)
SELECT * FROM categories
SQL;
$lexer = new Lexer($sql);
$lexerErrors = $this->getErrorsAsArray($lexer);
$this->assertCount(0, $lexerErrors);
$parser = new Parser($lexer->list);
$parserErrors = $this->getErrorsAsArray($parser);
$this->assertCount(0, $parserErrors);
// phpcs:disable Generic.Files.LineLength.TooLong
$expected = <<<SQL
WITH categories AS (SELECT * FROM (SELECT * FROM foo))
SQL;
// phpcs:enable
$this->assertEquals($expected, $parser->statements[0]->build());
}
public function testWithHasUnclosedParenthesis(): void
{
$sql = <<<SQL
WITH categories(identifier, name, parent_id) AS (
SELECT * FROM (SELECT * FROM foo
)
SELECT * FROM categories
SQL;
$lexer = new Lexer($sql);
$lexerErrors = $this->getErrorsAsArray($lexer);
$this->assertCount(0, $lexerErrors);
$parser = new Parser($lexer->list);
$parserErrors = $this->getErrorsAsArray($parser);
$this->assertEquals($parserErrors[0][0], 'A closing bracket was expected.');
}
public function testBuildWrongWithKeyword(): void
{
$this->expectExceptionMessage('Can not build a component that is not a WithKeyword');
WithKeyword::build(new stdClass());
}
public function testBuildBadWithKeyword(): void
{
$this->expectExceptionMessage('No statement inside WITH');
WithKeyword::build(new WithKeyword('test'));
}
}