@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());
}
