SQLParserUtilsTest.php 23 KB
Newer Older
1 2 3 4 5
<?php

namespace Doctrine\Tests\DBAL;

use Doctrine\DBAL\Connection;
6
use Doctrine\DBAL\ParameterType;
7
use Doctrine\DBAL\SQLParserUtils;
Sergei Morozov's avatar
Sergei Morozov committed
8
use Doctrine\DBAL\SQLParserUtilsException;
Sergei Morozov's avatar
Sergei Morozov committed
9
use Doctrine\Tests\DbalTestCase;
10 11 12

/**
 * @group DBAL-78
13
 * @group DDC-1372
14
 */
Sergei Morozov's avatar
Sergei Morozov committed
15
class SQLParserUtilsTest extends DbalTestCase
16
{
17
    public function dataGetPlaceholderPositions()
18
    {
Sergei Morozov's avatar
Sergei Morozov committed
19
        return [
20
            // none
Sergei Morozov's avatar
Sergei Morozov committed
21 22
            ['SELECT * FROM Foo', true, []],
            ['SELECT * FROM Foo', false, []],
23

24
            // Positionals
Sergei Morozov's avatar
Sergei Morozov committed
25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46
            ['SELECT ?', true, [7]],
            ['SELECT * FROM Foo WHERE bar IN (?, ?, ?)', true, [32, 35, 38]],
            ['SELECT ? FROM ?', true, [7, 14]],
            ['SELECT "?" FROM foo', true, []],
            ["SELECT '?' FROM foo", true, []],
            ['SELECT `?` FROM foo', true, []], // Ticket DBAL-552
            ['SELECT [?] FROM foo', true, []],
            ["SELECT 'Doctrine\DBAL?' FROM foo", true, []], // Ticket DBAL-558
            ['SELECT "Doctrine\DBAL?" FROM foo', true, []], // Ticket DBAL-558
            ['SELECT `Doctrine\DBAL?` FROM foo', true, []], // Ticket DBAL-558
            ['SELECT [Doctrine\DBAL?] FROM foo', true, []], // Ticket DBAL-558
            ['SELECT "?" FROM foo WHERE bar = ?', true, [32]],
            ["SELECT '?' FROM foo WHERE bar = ?", true, [32]],
            ['SELECT `?` FROM foo WHERE bar = ?', true, [32]], // Ticket DBAL-552
            ['SELECT [?] FROM foo WHERE bar = ?', true, [32]],
            ['SELECT * FROM foo WHERE jsonb_exists_any(foo.bar, ARRAY[?])', true, [56]], // Ticket GH-2295
            ["SELECT 'Doctrine\DBAL?' FROM foo WHERE bar = ?", true, [45]], // Ticket DBAL-558
            ['SELECT "Doctrine\DBAL?" FROM foo WHERE bar = ?', true, [45]], // Ticket DBAL-558
            ['SELECT `Doctrine\DBAL?` FROM foo WHERE bar = ?', true, [45]], // Ticket DBAL-558
            ['SELECT [Doctrine\DBAL?] FROM foo WHERE bar = ?', true, [45]], // Ticket DBAL-558
            ["SELECT * FROM FOO WHERE bar = 'it\\'s a trap? \\\\' OR bar = ?\nAND baz = \"\\\"quote\\\" me on it? \\\\\" OR baz = ?", true, [58, 104]],
            ['SELECT * FROM foo WHERE foo = ? AND bar = ?', true, [1 => 42, 0 => 30]], // explicit keys
47

48
            // named
Sergei Morozov's avatar
Sergei Morozov committed
49 50 51 52 53 54 55 56 57 58 59
            ['SELECT :foo FROM :bar', false, [7 => 'foo', 17 => 'bar']],
            ['SELECT * FROM Foo WHERE bar IN (:name1, :name2)', false, [32 => 'name1', 40 => 'name2']],
            ['SELECT ":foo" FROM Foo WHERE bar IN (:name1, :name2)', false, [37 => 'name1', 45 => 'name2']],
            ["SELECT ':foo' FROM Foo WHERE bar IN (:name1, :name2)", false, [37 => 'name1', 45 => 'name2']],
            ['SELECT :foo_id', false, [7 => 'foo_id']], // Ticket DBAL-231
            ['SELECT @rank := 1', false, []], // Ticket DBAL-398
            ['SELECT @rank := 1 AS rank, :foo AS foo FROM :bar', false, [27 => 'foo', 44 => 'bar']], // Ticket DBAL-398
            ['SELECT * FROM Foo WHERE bar > :start_date AND baz > :start_date', false, [30 => 'start_date', 52 => 'start_date']], // Ticket GH-113
            ['SELECT foo::date as date FROM Foo WHERE bar > :start_date AND baz > :start_date', false, [46 => 'start_date', 68 => 'start_date']], // Ticket GH-259
            ['SELECT `d.ns:col_name` FROM my_table d WHERE `d.date` >= :param1', false, [57 => 'param1']], // Ticket DBAL-552
            ['SELECT [d.ns:col_name] FROM my_table d WHERE [d.date] >= :param1', false, [57 => 'param1']], // Ticket DBAL-552
60 61
            ['SELECT * FROM foo WHERE jsonb_exists_any(foo.bar, ARRAY[:foo])', false, [56 => 'foo']], // Ticket GH-2295
            ['SELECT * FROM foo WHERE jsonb_exists_any(foo.bar, array[:foo])', false, [56 => 'foo']],
62 63 64 65
            ['SELECT table.field1, ARRAY[\'3\'] FROM schema.table table WHERE table.f1 = :foo AND ARRAY[\'3\']', false, [73 => 'foo']],
            ['SELECT table.field1, ARRAY[\'3\']::integer[] FROM schema.table table WHERE table.f1 = :foo AND ARRAY[\'3\']::integer[]', false, [84 => 'foo']],
            ['SELECT table.field1, ARRAY[:foo] FROM schema.table table WHERE table.f1 = :bar AND ARRAY[\'3\']', false, [27 => 'foo', 74 => 'bar']],
            ['SELECT table.field1, ARRAY[:foo]::integer[] FROM schema.table table WHERE table.f1 = :bar AND ARRAY[\'3\']::integer[]', false, [27 => 'foo', 85 => 'bar']],
Sergei Morozov's avatar
Sergei Morozov committed
66 67
            [
                <<<'SQLDATA'
68 69 70 71 72 73
SELECT * FROM foo WHERE 
bar = ':not_a_param1 ''":not_a_param2"'''
OR bar=:a_param1
OR bar=:a_param2||':not_a_param3'
OR bar=':not_a_param4 '':not_a_param5'' :not_a_param6'
OR bar=''
andreas's avatar
andreas committed
74
OR bar=:a_param3
75
SQLDATA
Sergei Morozov's avatar
Sergei Morozov committed
76 77 78 79 80 81 82 83 84 85 86 87 88
                ,
                false,
                [
                    74 => 'a_param1',
                    91 => 'a_param2',
                    190 => 'a_param3',
                ],
            ],
            ["SELECT data.age AS age, data.id AS id, data.name AS name, data.id AS id FROM test_data data WHERE (data.description LIKE :condition_0 ESCAPE '\\\\') AND (data.description LIKE :condition_1 ESCAPE '\\\\') ORDER BY id ASC", false, [121 => 'condition_0', 174 => 'condition_1']],
            ['SELECT data.age AS age, data.id AS id, data.name AS name, data.id AS id FROM test_data data WHERE (data.description LIKE :condition_0 ESCAPE "\\\\") AND (data.description LIKE :condition_1 ESCAPE "\\\\") ORDER BY id ASC', false, [121 => 'condition_0', 174 => 'condition_1']],
            ['SELECT data.age AS age, data.id AS id, data.name AS name, data.id AS id FROM test_data data WHERE (data.description LIKE :condition_0 ESCAPE "\\\\") AND (data.description LIKE :condition_1 ESCAPE \'\\\\\') ORDER BY id ASC', false, [121 => 'condition_0', 174 => 'condition_1']],
            ['SELECT data.age AS age, data.id AS id, data.name AS name, data.id AS id FROM test_data data WHERE (data.description LIKE :condition_0 ESCAPE `\\\\`) AND (data.description LIKE :condition_1 ESCAPE `\\\\`) ORDER BY id ASC', false, [121 => 'condition_0', 174 => 'condition_1']],
            ['SELECT data.age AS age, data.id AS id, data.name AS name, data.id AS id FROM test_data data WHERE (data.description LIKE :condition_0 ESCAPE \'\\\\\') AND (data.description LIKE :condition_1 ESCAPE `\\\\`) ORDER BY id ASC', false, [121 => 'condition_0', 174 => 'condition_1']],
Luís Cobucci's avatar
Luís Cobucci committed
89

Sergei Morozov's avatar
Sergei Morozov committed
90
        ];
91
    }
92

93 94 95 96 97 98
    /**
     * @dataProvider dataGetPlaceholderPositions
     */
    public function testGetPlaceholderPositions($query, $isPositional, $expectedParamPos)
    {
        $actualParamPos = SQLParserUtils::getPlaceholderPositions($query, $isPositional);
99
        self::assertEquals($expectedParamPos, $actualParamPos);
100
    }
101

102
    public function dataExpandListParameters()
103
    {
Sergei Morozov's avatar
Sergei Morozov committed
104
        return [
105
            // Positional: Very simple with one needle
Sergei Morozov's avatar
Sergei Morozov committed
106 107 108 109
            [
                'SELECT * FROM Foo WHERE foo IN (?)',
                [[1, 2, 3]],
                [Connection::PARAM_INT_ARRAY],
110
                'SELECT * FROM Foo WHERE foo IN (?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
111 112 113
                [1, 2, 3],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
114
            // Positional: One non-list before d one after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
115 116 117 118
            [
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?)',
                ['string', [1, 2, 3]],
                [ParameterType::STRING, Connection::PARAM_INT_ARRAY],
119
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
120 121 122
                ['string', 1, 2, 3],
                [ParameterType::STRING, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
123
            // Positional: One non-list after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
124 125 126 127
            [
                'SELECT * FROM Foo WHERE bar IN (?) AND baz = ?',
                [[1, 2, 3], 'foo'],
                [Connection::PARAM_INT_ARRAY, ParameterType::STRING],
128
                'SELECT * FROM Foo WHERE bar IN (?, ?, ?) AND baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
129 130 131
                [1, 2, 3, 'foo'],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::STRING],
            ],
132
            // Positional: One non-list before and one after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
133 134 135 136
            [
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?) AND baz = ?',
                [1, [1, 2, 3], 4],
                [ParameterType::INTEGER, Connection::PARAM_INT_ARRAY, ParameterType::INTEGER],
137
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?, ?, ?) AND baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
138 139
                [1, 1, 2, 3, 4],
                [
140 141 142 143 144
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
Sergei Morozov's avatar
Sergei Morozov committed
145 146
                ],
            ],
147
            // Positional: Two lists
Sergei Morozov's avatar
Sergei Morozov committed
148 149 150 151
            [
                'SELECT * FROM Foo WHERE foo IN (?, ?)',
                [[1, 2, 3], [4, 5]],
                [Connection::PARAM_INT_ARRAY, Connection::PARAM_INT_ARRAY],
152
                'SELECT * FROM Foo WHERE foo IN (?, ?, ?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
153 154
                [1, 2, 3, 4, 5],
                [
155 156 157 158 159
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
Sergei Morozov's avatar
Sergei Morozov committed
160 161
                ],
            ],
162
            // Positional: Empty "integer" array DDC-1978
Sergei Morozov's avatar
Sergei Morozov committed
163 164 165 166
            [
                'SELECT * FROM Foo WHERE foo IN (?)',
                [[]],
                [Connection::PARAM_INT_ARRAY],
167
                'SELECT * FROM Foo WHERE foo IN (NULL)',
Sergei Morozov's avatar
Sergei Morozov committed
168 169 170
                [],
                [],
            ],
171
            // Positional: Empty "str" array DDC-1978
Sergei Morozov's avatar
Sergei Morozov committed
172 173 174 175
            [
                'SELECT * FROM Foo WHERE foo IN (?)',
                [[]],
                [Connection::PARAM_STR_ARRAY],
176
                'SELECT * FROM Foo WHERE foo IN (NULL)',
Sergei Morozov's avatar
Sergei Morozov committed
177 178 179
                [],
                [],
            ],
180
            // Positional: explicit keys for params and types
Sergei Morozov's avatar
Sergei Morozov committed
181
            [
182
                'SELECT * FROM Foo WHERE foo = ? AND bar = ? AND baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
183 184 185 186 187 188
                [1 => 'bar', 2 => 'baz', 0 => 1],
                [2 => ParameterType::STRING, 1 => ParameterType::STRING],
                'SELECT * FROM Foo WHERE foo = ? AND bar = ? AND baz = ?',
                [1 => 'bar', 0 => 1, 2 => 'baz'],
                [1 => ParameterType::STRING, 2 => ParameterType::STRING],
            ],
189
            // Positional: explicit keys for array params and array types
Sergei Morozov's avatar
Sergei Morozov committed
190 191 192 193
            [
                'SELECT * FROM Foo WHERE foo IN (?) AND bar IN (?) AND baz = ?',
                [1 => ['bar1', 'bar2'], 2 => true, 0 => [1, 2, 3]],
                [2 => ParameterType::BOOLEAN, 1 => Connection::PARAM_STR_ARRAY, 0 => Connection::PARAM_INT_ARRAY],
194
                'SELECT * FROM Foo WHERE foo IN (?, ?, ?) AND bar IN (?, ?) AND baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
195 196
                [1, 2, 3, 'bar1', 'bar2', true],
                [
197 198 199 200 201 202
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::STRING,
                    ParameterType::STRING,
                    ParameterType::BOOLEAN,
Sergei Morozov's avatar
Sergei Morozov committed
203 204
                ],
            ],
205
            // Positional starts from 1: One non-list before and one after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
206 207 208 209
            [
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?) AND baz = ? AND foo IN (?)',
                [1 => 1, 2 => [1, 2, 3], 3 => 4, 4 => [5, 6]],
                [
210 211 212 213
                    1 => ParameterType::INTEGER,
                    2 => Connection::PARAM_INT_ARRAY,
                    3 => ParameterType::INTEGER,
                    4 => Connection::PARAM_INT_ARRAY,
Sergei Morozov's avatar
Sergei Morozov committed
214
                ],
215
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?, ?, ?) AND baz = ? AND foo IN (?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
216 217
                [1, 1, 2, 3, 4, 5, 6],
                [
218 219 220 221 222 223 224
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
                    ParameterType::INTEGER,
Sergei Morozov's avatar
Sergei Morozov committed
225 226
                ],
            ],
227
            //  Named parameters : Very simple with param int
Sergei Morozov's avatar
Sergei Morozov committed
228 229 230 231
            [
                'SELECT * FROM Foo WHERE foo = :foo',
                ['foo' => 1],
                ['foo' => ParameterType::INTEGER],
232
                'SELECT * FROM Foo WHERE foo = ?',
Sergei Morozov's avatar
Sergei Morozov committed
233 234 235
                [1],
                [ParameterType::INTEGER],
            ],
236

237
             //  Named parameters : Very simple with param int and string
Sergei Morozov's avatar
Sergei Morozov committed
238 239 240 241
            [
                'SELECT * FROM Foo WHERE foo = :foo AND bar = :bar',
                ['bar' => 'Some String','foo' => 1],
                ['foo' => ParameterType::INTEGER, 'bar' => ParameterType::STRING],
242
                'SELECT * FROM Foo WHERE foo = ? AND bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
243 244 245
                [1,'Some String'],
                [ParameterType::INTEGER, ParameterType::STRING],
            ],
246
            //  Named parameters : Very simple with one needle
Sergei Morozov's avatar
Sergei Morozov committed
247 248 249 250
            [
                'SELECT * FROM Foo WHERE foo IN (:foo)',
                ['foo' => [1, 2, 3]],
                ['foo' => Connection::PARAM_INT_ARRAY],
251
                'SELECT * FROM Foo WHERE foo IN (?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
252 253 254
                [1, 2, 3],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
255
            // Named parameters: One non-list before d one after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
256 257 258 259
            [
                'SELECT * FROM Foo WHERE foo = :foo AND bar IN (:bar)',
                ['foo' => 'string', 'bar' => [1, 2, 3]],
                ['foo' => ParameterType::STRING, 'bar' => Connection::PARAM_INT_ARRAY],
260
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
261 262 263
                ['string', 1, 2, 3],
                [ParameterType::STRING, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
264
            // Named parameters: One non-list after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
265 266 267 268
            [
                'SELECT * FROM Foo WHERE bar IN (:bar) AND baz = :baz',
                ['bar' => [1, 2, 3], 'baz' => 'foo'],
                ['bar' => Connection::PARAM_INT_ARRAY, 'baz' => ParameterType::STRING],
269
                'SELECT * FROM Foo WHERE bar IN (?, ?, ?) AND baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
270 271 272
                [1, 2, 3, 'foo'],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::STRING],
            ],
273
            // Named parameters: One non-list before and one after list-needle
Sergei Morozov's avatar
Sergei Morozov committed
274 275 276 277
            [
                'SELECT * FROM Foo WHERE foo = :foo AND bar IN (:bar) AND baz = :baz',
                ['bar' => [1, 2, 3],'foo' => 1, 'baz' => 4],
                ['bar' => Connection::PARAM_INT_ARRAY, 'foo' => ParameterType::INTEGER, 'baz' => ParameterType::INTEGER],
278
                'SELECT * FROM Foo WHERE foo = ? AND bar IN (?, ?, ?) AND baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
279 280 281
                [1, 1, 2, 3, 4],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
282
            // Named parameters: Two lists
Sergei Morozov's avatar
Sergei Morozov committed
283 284 285 286
            [
                'SELECT * FROM Foo WHERE foo IN (:a, :b)',
                ['b' => [4, 5],'a' => [1, 2, 3]],
                ['a' => Connection::PARAM_INT_ARRAY, 'b' => Connection::PARAM_INT_ARRAY],
287
                'SELECT * FROM Foo WHERE foo IN (?, ?, ?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
288 289 290
                [1, 2, 3, 4, 5],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
291
            //  Named parameters : With the same name arg type string
Sergei Morozov's avatar
Sergei Morozov committed
292 293 294 295
            [
                'SELECT * FROM Foo WHERE foo <> :arg AND bar = :arg',
                ['arg' => 'Some String'],
                ['arg' => ParameterType::STRING],
296
                'SELECT * FROM Foo WHERE foo <> ? AND bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
297 298 299
                ['Some String','Some String'],
                [ParameterType::STRING,ParameterType::STRING],
            ],
300
             //  Named parameters : With the same name arg
Sergei Morozov's avatar
Sergei Morozov committed
301 302 303 304
            [
                'SELECT * FROM Foo WHERE foo IN (:arg) AND NOT bar IN (:arg)',
                ['arg' => [1, 2, 3]],
                ['arg' => Connection::PARAM_INT_ARRAY],
305
                'SELECT * FROM Foo WHERE foo IN (?, ?, ?) AND NOT bar IN (?, ?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
306 307 308
                [1, 2, 3, 1, 2, 3],
                [ParameterType::INTEGER,ParameterType::INTEGER, ParameterType::INTEGER,ParameterType::INTEGER,ParameterType::INTEGER, ParameterType::INTEGER],
            ],
309 310

             //  Named parameters : Same name, other name in between DBAL-299
Sergei Morozov's avatar
Sergei Morozov committed
311 312 313 314
            [
                'SELECT * FROM Foo WHERE (:foo = 2) AND (:bar = 3) AND (:foo = 2)',
                ['foo' => 2,'bar' => 3],
                ['foo' => ParameterType::INTEGER,'bar' => ParameterType::INTEGER],
315
                'SELECT * FROM Foo WHERE (? = 2) AND (? = 3) AND (? = 2)',
Sergei Morozov's avatar
Sergei Morozov committed
316 317 318
                [2, 3, 2],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::INTEGER],
            ],
Fabio B. Silva's avatar
Fabio B. Silva committed
319
             //  Named parameters : Empty "integer" array DDC-1978
Sergei Morozov's avatar
Sergei Morozov committed
320 321 322 323
            [
                'SELECT * FROM Foo WHERE foo IN (:foo)',
                ['foo' => []],
                ['foo' => Connection::PARAM_INT_ARRAY],
324
                'SELECT * FROM Foo WHERE foo IN (NULL)',
Sergei Morozov's avatar
Sergei Morozov committed
325 326 327
                [],
                [],
            ],
Fabio B. Silva's avatar
Fabio B. Silva committed
328
             //  Named parameters : Two empty "str" array DDC-1978
Sergei Morozov's avatar
Sergei Morozov committed
329 330 331 332
            [
                'SELECT * FROM Foo WHERE foo IN (:foo) OR bar IN (:bar)',
                ['foo' => [], 'bar' => []],
                ['foo' => Connection::PARAM_STR_ARRAY, 'bar' => Connection::PARAM_STR_ARRAY],
333
                'SELECT * FROM Foo WHERE foo IN (NULL) OR bar IN (NULL)',
Sergei Morozov's avatar
Sergei Morozov committed
334 335 336 337 338 339 340
                [],
                [],
            ],
            [
                'SELECT * FROM Foo WHERE foo IN (:foo) OR bar = :bar OR baz = :baz',
                ['foo' => [1, 2], 'bar' => 'bar', 'baz' => 'baz'],
                ['foo' => Connection::PARAM_INT_ARRAY, 'baz' => 'string'],
341
                'SELECT * FROM Foo WHERE foo IN (?, ?) OR bar = ? OR baz = ?',
Sergei Morozov's avatar
Sergei Morozov committed
342 343 344 345 346 347 348
                [1, 2, 'bar', 'baz'],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::STRING, 'string'],
            ],
            [
                'SELECT * FROM Foo WHERE foo IN (:foo) OR bar = :bar',
                ['foo' => [1, 2], 'bar' => 'bar'],
                ['foo' => Connection::PARAM_INT_ARRAY],
349
                'SELECT * FROM Foo WHERE foo IN (?, ?) OR bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
350 351 352
                [1, 2, 'bar'],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::STRING],
            ],
353
            // Params/types with colons
Sergei Morozov's avatar
Sergei Morozov committed
354 355 356 357
            [
                'SELECT * FROM Foo WHERE foo = :foo OR bar = :bar',
                [':foo' => 'foo', ':bar' => 'bar'],
                [':foo' => ParameterType::INTEGER],
358
                'SELECT * FROM Foo WHERE foo = ? OR bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
359 360 361 362 363 364 365
                ['foo', 'bar'],
                [ParameterType::INTEGER, ParameterType::STRING],
            ],
            [
                'SELECT * FROM Foo WHERE foo = :foo OR bar = :bar',
                [':foo' => 'foo', ':bar' => 'bar'],
                [':foo' => ParameterType::INTEGER, 'bar' => ParameterType::INTEGER],
366
                'SELECT * FROM Foo WHERE foo = ? OR bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
367 368 369 370 371 372 373
                ['foo', 'bar'],
                [ParameterType::INTEGER, ParameterType::INTEGER],
            ],
            [
                'SELECT * FROM Foo WHERE foo IN (:foo) OR bar = :bar',
                [':foo' => [1, 2], ':bar' => 'bar'],
                ['foo' => Connection::PARAM_INT_ARRAY],
374
                'SELECT * FROM Foo WHERE foo IN (?, ?) OR bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
375 376 377 378 379 380 381
                [1, 2, 'bar'],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::STRING],
            ],
            [
                'SELECT * FROM Foo WHERE foo IN (:foo) OR bar = :bar',
                ['foo' => [1, 2], 'bar' => 'bar'],
                [':foo' => Connection::PARAM_INT_ARRAY],
382
                'SELECT * FROM Foo WHERE foo IN (?, ?) OR bar = ?',
Sergei Morozov's avatar
Sergei Morozov committed
383 384 385
                [1, 2, 'bar'],
                [ParameterType::INTEGER, ParameterType::INTEGER, ParameterType::STRING],
            ],
386
            // DBAL-522 - null valued parameters are not considered
Sergei Morozov's avatar
Sergei Morozov committed
387
            [
388
                'INSERT INTO Foo (foo, bar) values (:foo, :bar)',
Sergei Morozov's avatar
Sergei Morozov committed
389 390
                ['foo' => 1, 'bar' => null],
                [':foo' => ParameterType::INTEGER, ':bar' => ParameterType::NULL],
391
                'INSERT INTO Foo (foo, bar) values (?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
392 393 394 395
                [1, null],
                [ParameterType::INTEGER, ParameterType::NULL],
            ],
            [
396
                'INSERT INTO Foo (foo, bar) values (?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
397 398
                [1, null],
                [ParameterType::INTEGER, ParameterType::NULL],
399
                'INSERT INTO Foo (foo, bar) values (?, ?)',
Sergei Morozov's avatar
Sergei Morozov committed
400 401 402
                [1, null],
                [ParameterType::INTEGER, ParameterType::NULL],
            ],
403
            // DBAL-1205 - Escaped single quotes SQL- and C-Style
Sergei Morozov's avatar
Sergei Morozov committed
404
            [
405
                "SELECT * FROM Foo WHERE foo = :foo||''':not_a_param''\\'' OR bar = ''':not_a_param''\\'':bar",
Sergei Morozov's avatar
Sergei Morozov committed
406 407
                [':foo' => 1, ':bar' => 2],
                [':foo' => ParameterType::INTEGER, 'bar' => ParameterType::INTEGER],
408
                'SELECT * FROM Foo WHERE foo = ?||\'\'\':not_a_param\'\'\\\'\' OR bar = \'\'\':not_a_param\'\'\\\'\'?',
Sergei Morozov's avatar
Sergei Morozov committed
409 410 411 412
                [1, 2],
                [ParameterType::INTEGER, ParameterType::INTEGER],
            ],
        ];
413
    }
414

415 416 417 418 419
    /**
     * @dataProvider dataExpandListParameters
     */
    public function testExpandListParameters($q, $p, $t, $expectedQuery, $expectedParams, $expectedTypes)
    {
Sergei Morozov's avatar
Sergei Morozov committed
420
        [$query, $params, $types] = SQLParserUtils::expandListParameters($q, $p, $t);
421

Sergei Morozov's avatar
Sergei Morozov committed
422 423 424
        self::assertEquals($expectedQuery, $query, 'Query was not rewritten correctly.');
        self::assertEquals($expectedParams, $params, 'Params dont match');
        self::assertEquals($expectedTypes, $types, 'Types dont match');
425
    }
426

427
    public function dataQueryWithMissingParameters()
428
    {
Sergei Morozov's avatar
Sergei Morozov committed
429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
        return [
            [
                'SELECT * FROM foo WHERE bar = :param',
                ['other' => 'val'],
                [],
            ],
            [
                'SELECT * FROM foo WHERE bar = :param',
                [],
                [],
            ],
            [
                'SELECT * FROM foo WHERE bar = :param',
                [],
                ['param' => Connection::PARAM_INT_ARRAY],
            ],
            [
                'SELECT * FROM foo WHERE bar = :param',
                [],
                [':param' => Connection::PARAM_INT_ARRAY],
            ],
            [
                'SELECT * FROM foo WHERE bar = :param',
                [],
                ['bar' => Connection::PARAM_INT_ARRAY],
            ],
            [
                'SELECT * FROM foo WHERE bar = :param',
                ['bar' => 'value'],
                ['bar' => Connection::PARAM_INT_ARRAY],
            ],
        ];
461 462 463 464 465
    }

    /**
     * @dataProvider dataQueryWithMissingParameters
     */
Sergei Morozov's avatar
Sergei Morozov committed
466
    public function testExceptionIsThrownForMissingParam($query, $params, $types = [])
467
    {
Sergei Morozov's avatar
Sergei Morozov committed
468
        $this->expectException(SQLParserUtilsException::class);
469
        $this->expectExceptionMessage('Value for :param not found in params array. Params array key should be "param"');
470 471 472

        SQLParserUtils::expandListParameters($query, $params, $types);
    }
473
}