@Test public void selectField() throws RecognitionException { String queryString = "select c where a = 1 or b > 2"; ANTLRStringStream in = new ANTLRStringStream(queryString); QueryFilterLexer lexer = new QueryFilterLexer(in); TokenRewriteStream tokens = new TokenRewriteStream(lexer); QueryFilterParser parser = new QueryFilterParser(tokens); Query query = parser.ql().query; Set<String> identifiers = query.getSelectSubjects(); assertTrue(identifiers.contains("c")); }
/** Test basic || expression */ @Test public void selectAll() throws RecognitionException { String queryString = "select * where a = 1 or b > 2"; ANTLRStringStream in = new ANTLRStringStream(queryString); QueryFilterLexer lexer = new QueryFilterLexer(in); TokenRewriteStream tokens = new TokenRewriteStream(lexer); QueryFilterParser parser = new QueryFilterParser(tokens); Query query = parser.ql().query; Set<String> identifiers = query.getSelectSubjects(); assertEquals(0, identifiers.size()); }