File tree Expand file tree Collapse file tree 4 files changed +22
-31
lines changed
ppl/src/test/java/org/opensearch/sql/ppl Expand file tree Collapse file tree 4 files changed +22
-31
lines changed Original file line number Diff line number Diff line change 1111import org .opensearch .sql .ppl .antlr .PPLSyntaxParser ;
1212import org .opensearch .sql .ppl .parser .AstBuilder ;
1313
14- /**
15- * Base class for tests for the AST query planner.
16- */
14+ /** Base class for tests for the AST query planner. */
1715public class AstPlanningTest {
18- public final Settings settings = Mockito .mock (Settings .class );
19- public final PPLSyntaxParser parser = new PPLSyntaxParser ();
16+ protected final Settings settings = Mockito .mock (Settings .class );
17+ protected final PPLSyntaxParser parser = new PPLSyntaxParser ();
2018
21- public Node plan (String query ) {
22- AstBuilder astBuilder = new AstBuilder (query , settings );
23- return astBuilder .visit (parser .parse (query ));
24- }
19+ protected Node plan (String query ) {
20+ AstBuilder astBuilder = new AstBuilder (query , settings );
21+ return astBuilder .visit (parser .parse (query ));
22+ }
2523}
Original file line number Diff line number Diff line change 6060import org .junit .Rule ;
6161import org .junit .Test ;
6262import org .junit .rules .ExpectedException ;
63- import org .mockito .Mockito ;
6463import org .opensearch .sql .ast .Node ;
6564import org .opensearch .sql .ast .dsl .AstDSL ;
6665import org .opensearch .sql .ast .expression .DataType ;
7473import org .opensearch .sql .ast .tree .ML ;
7574import org .opensearch .sql .ast .tree .RareTopN .CommandType ;
7675import org .opensearch .sql .common .antlr .SyntaxCheckException ;
77- import org .opensearch .sql .common .setting .Settings ;
7876import org .opensearch .sql .common .setting .Settings .Key ;
7977import org .opensearch .sql .ppl .AstPlanningTest ;
80- import org .opensearch .sql .ppl .antlr .PPLSyntaxParser ;
8178import org .opensearch .sql .utils .SystemIndexUtils ;
8279
8380public class AstBuilderTest extends AstPlanningTest {
Original file line number Diff line number Diff line change 55
66package org .opensearch .sql .ppl .parser ;
77
8+ import static org .junit .Assert .assertEquals ;
9+
810import org .junit .Test ;
911import org .opensearch .sql .ppl .AstPlanningTest ;
1012
11- import static org .junit .Assert .assertEquals ;
12-
1313public class AstEquivalenceTest extends AstPlanningTest {
14- @ Test
15- public void testSpathArgumentDeshuffle () {
16- assertEquals (plan ("source = t | spath path=a input=a" ), plan ("source = t | spath input=a a" ));
17- }
14+ @ Test
15+ public void testSpathArgumentDeshuffle () {
16+ assertEquals (plan ("source = t | spath path=a input=a" ), plan ("source = t | spath input=a a" ));
17+ }
1818
19- @ Test
20- public void testHeadLimitEquivalent () {
21- assertEquals (plan ("source = t | head limit=50" ), plan ("source = t | head 50" ));
22- }
19+ @ Test
20+ public void testHeadLimitEquivalent () {
21+ assertEquals (plan ("source = t | head limit=50" ), plan ("source = t | head 50" ));
22+ }
2323
24- @ Test
25- public void testTopLimitEquivalent () {
26- assertEquals (plan ("source = t | top limit=50 field_name" ), plan ("source = t | top 50 field_name" ));
27- }
24+ @ Test
25+ public void testTopLimitEquivalent () {
26+ assertEquals (
27+ plan ("source = t | top limit=50 field_name" ), plan ("source = t | top 50 field_name" ));
28+ }
2829}
Original file line number Diff line number Diff line change 1515import static org .opensearch .sql .ast .dsl .AstDSL .stringLiteral ;
1616
1717import org .junit .Test ;
18- import org .mockito .Mockito ;
19- import org .opensearch .sql .ast .Node ;
2018import org .opensearch .sql .ast .tree .Eval ;
2119import org .opensearch .sql .ast .tree .SPath ;
22- import org .opensearch .sql .common .setting .Settings ;
2320import org .opensearch .sql .ppl .AstPlanningTest ;
24- import org .opensearch .sql .ppl .antlr .PPLSyntaxParser ;
25- import org .opensearch .sql .ppl .parser .AstBuilder ;
2621
2722public class SPathRewriteTest extends AstPlanningTest {
2823 // Control test to make sure something fundamental hasn't changed about the json_extract parsing
You can’t perform that action at this time.
0 commit comments