/
DeleteQueryTest.java
66 lines (53 loc) · 2.36 KB
/
DeleteQueryTest.java
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
/*
* Copyright (c) 2018 Otávio Santana and others
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License v1.0
* and Apache License v2.0 which accompanies this distribution.
* The Eclipse Public License is available at http://www.eclipse.org/legal/epl-v10.html
* and the Apache License v2.0 is available at http://www.opensource.org/licenses/apache2.0.php.
* You may elect to redistribute this code under either of these licenses.
* Contributors:
* Otavio Santana
*/
package org.eclipse.jnosql.diana.query;
import jakarta.nosql.QueryException;
import org.antlr.v4.runtime.CharStream;
import org.antlr.v4.runtime.CharStreams;
import org.antlr.v4.runtime.CommonTokenStream;
import org.antlr.v4.runtime.tree.ParseTree;
import org.antlr.v4.runtime.tree.ParseTreeWalker;
import org.eclipse.jnosql.diana.query.provider.DeleteQueryArgumentProvider;
import org.eclipse.jnosql.diana.query.provider.WrongDeleteQueryArgumentProvider;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.ArgumentsSource;
public class DeleteQueryTest {
@ParameterizedTest
@ArgumentsSource(DeleteQueryArgumentProvider.class)
public void shouldExecuteQuery(String query) {
testQuery(query);
}
@Test
public void shouldIgnoreComments() {
testQuery("//ignore this line \n delete from Person");
}
@ParameterizedTest(name = "Should parser the query {0}")
@ArgumentsSource(WrongDeleteQueryArgumentProvider.class)
public void shouldNotExecute(String query) {
Assertions.assertThrows(QueryException.class, () -> testQuery(query));
}
private void testQuery(String query) {
CharStream stream = CharStreams.fromString(query);
QueryLexer lexer = new QueryLexer(stream);
CommonTokenStream tokens = new CommonTokenStream(lexer);
QueryParser parser = new QueryParser(tokens);
lexer.removeErrorListeners();
parser.removeErrorListeners();
lexer.addErrorListener(QueryErrorListener.INSTANCE);
parser.addErrorListener(QueryErrorListener.INSTANCE);
ParseTree tree = parser.delete();
ParseTreeWalker walker = new ParseTreeWalker();
walker.walk(new QueryBaseListener(), tree);
}
}