|
4 | 4 |
|
5 | 5 | namespace PhpMyAdmin\SqlParser\Tests\Builder; |
6 | 6 |
|
| 7 | +use Generator; |
| 8 | +use PhpMyAdmin\SqlParser\Context; |
7 | 9 | use PhpMyAdmin\SqlParser\Parser; |
8 | 10 | use PhpMyAdmin\SqlParser\Tests\TestCase; |
9 | 11 |
|
10 | 12 | class InsertStatementTest extends TestCase |
11 | 13 | { |
12 | | - public function testBuilder(): void |
| 14 | + /** @var int */ |
| 15 | + private $sqlMode; |
| 16 | + |
| 17 | + public function setUp(): void |
13 | 18 | { |
14 | | - /* Assertion 1 */ |
15 | | - $parser = new Parser('INSERT INTO tbl(`col1`, `col2`, `col3`) VALUES (1, "str", 3.14)'); |
16 | | - $stmt = $parser->statements[0]; |
17 | | - $this->assertEquals( |
18 | | - 'INSERT INTO tbl(`col1`, `col2`, `col3`) VALUES (1, "str", 3.14)', |
19 | | - $stmt->build() |
20 | | - ); |
21 | | - |
22 | | - /* Assertion 2 */ |
23 | | - /* Reserved keywords (with backquotes as field name) */ |
24 | | - $parser = new Parser('INSERT INTO tbl(`order`) VALUES (1)'); |
25 | | - $stmt = $parser->statements[0]; |
26 | | - $this->assertEquals( |
27 | | - 'INSERT INTO tbl(`order`) VALUES (1)', |
28 | | - $stmt->build() |
29 | | - ); |
30 | | - |
31 | | - /* Assertion 3 */ |
32 | | - /* INSERT ... SET ... */ |
33 | | - $parser = new Parser('INSERT INTO tbl SET FOO = 1'); |
34 | | - $stmt = $parser->statements[0]; |
35 | | - $this->assertEquals( |
36 | | - 'INSERT INTO tbl SET FOO = 1', |
37 | | - $stmt->build() |
38 | | - ); |
39 | | - |
40 | | - /* Assertion 4 */ |
41 | | - /* INSERT ... SELECT ... */ |
42 | | - $parser = new Parser('INSERT INTO tbl SELECT * FROM bar'); |
43 | | - $stmt = $parser->statements[0]; |
44 | | - $this->assertEquals( |
45 | | - 'INSERT INTO tbl SELECT * FROM bar', |
46 | | - $stmt->build() |
47 | | - ); |
48 | | - |
49 | | - /* Assertion 5 */ |
50 | | - /* INSERT ... ON DUPLICATE KEY UPDATE ... */ |
51 | | - $parser = new Parser('INSERT INTO tbl SELECT * FROM bar ON DUPLICATE KEY UPDATE baz = 1'); |
| 19 | + parent::setUp(); |
| 20 | + |
| 21 | + $this->sqlMode = Context::getMode(); |
| 22 | + } |
| 23 | + |
| 24 | + protected function tearDown(): void |
| 25 | + { |
| 26 | + Context::setMode($this->sqlMode); |
| 27 | + |
| 28 | + parent::tearDown(); |
| 29 | + } |
| 30 | + |
| 31 | + /** @dataProvider providerForTestBuilder */ |
| 32 | + public function testBuilder(string $sql): void |
| 33 | + { |
| 34 | + $parser = new Parser($sql); |
52 | 35 | $stmt = $parser->statements[0]; |
53 | | - $this->assertEquals( |
54 | | - 'INSERT INTO tbl SELECT * FROM bar ON DUPLICATE KEY UPDATE baz = 1', |
55 | | - $stmt->build() |
56 | | - ); |
57 | | - |
58 | | - /* Assertion 6 */ |
59 | | - /* INSERT [OPTIONS] INTO ... */ |
60 | | - $parser = new Parser('INSERT DELAYED IGNORE INTO tbl SELECT * FROM bar'); |
| 36 | + $this->assertEquals($sql, $stmt->build()); |
| 37 | + } |
| 38 | + |
| 39 | + /** @return Generator<string, list<string>> */ |
| 40 | + public function providerForTestBuilder(): Generator |
| 41 | + { |
| 42 | + yield 'INSERT ... VALUES ...' => ['INSERT INTO tbl(`col1`, `col2`, `col3`) VALUES (1, "str", 3.14)']; |
| 43 | + |
| 44 | + yield 'Reserved keywords (with backquotes as field name)' => ['INSERT INTO tbl(`order`) VALUES (1)']; |
| 45 | + |
| 46 | + yield 'INSERT ... SET ...' => ['INSERT INTO tbl SET FOO = 1']; |
| 47 | + |
| 48 | + yield 'INSERT ... SELECT ... ' => ['INSERT INTO tbl SELECT * FROM bar']; |
| 49 | + |
| 50 | + yield 'INSERT ... ON DUPLICATE KEY UPDATE ...' => ['INSERT INTO tbl SELECT * FROM bar ON DUPLICATE KEY UPDATE baz = 1']; |
| 51 | + |
| 52 | + yield 'INSERT [OPTIONS] INTO ...' => ['INSERT DELAYED IGNORE INTO tbl SELECT * FROM bar']; |
| 53 | + } |
| 54 | + |
| 55 | + public function testBuilderAnsi(): void |
| 56 | + { |
| 57 | + Context::setMode(Context::SQL_MODE_ANSI); |
| 58 | + $sql = "INSERT INTO foo (bar, baz) VALUES ('bar', 'baz')"; |
| 59 | + $parser = new Parser($sql); |
61 | 60 | $stmt = $parser->statements[0]; |
62 | | - $this->assertEquals( |
63 | | - 'INSERT DELAYED IGNORE INTO tbl SELECT * FROM bar', |
64 | | - $stmt->build() |
65 | | - ); |
| 61 | + $this->assertEquals('INSERT INTO foo("bar", "baz") VALUES (\'bar\', \'baz\')', $stmt->build()); |
66 | 62 | } |
67 | 63 | } |
0 commit comments