add package && add test

thrift-labs · Nov 21, 2024 · 2cee197 · 2cee197
1 parent 97ba721
commit 2cee197
Show file tree

Hide file tree

Showing 5 changed files with 15 additions and 2 deletions.
diff --git a/java/src/main/java/thriftlabs/thriftparser/ThriftBaseListener.java b/java/src/main/java/thriftlabs/thriftparser/ThriftBaseListener.java
@@ -1,3 +1,5 @@
+package thriftlabs.thriftparser;
+
 // Generated from Thrift.g4 by ANTLR 4.13.2
 
 import org.antlr.v4.runtime.ParserRuleContext;

diff --git a/java/src/main/java/thriftlabs/thriftparser/ThriftLexer.java b/java/src/main/java/thriftlabs/thriftparser/ThriftLexer.java
@@ -1,3 +1,5 @@
+package thriftlabs.thriftparser;
+
 // Generated from Thrift.g4 by ANTLR 4.13.2
 import org.antlr.v4.runtime.Lexer;
 import org.antlr.v4.runtime.CharStream;

diff --git a/java/src/main/java/thriftlabs/thriftparser/ThriftListener.java b/java/src/main/java/thriftlabs/thriftparser/ThriftListener.java
@@ -1,3 +1,5 @@
+package thriftlabs.thriftparser;
+
 // Generated from Thrift.g4 by ANTLR 4.13.2
 import org.antlr.v4.runtime.tree.ParseTreeListener;
 

diff --git a/java/src/main/java/thriftlabs/thriftparser/ThriftParser.java b/java/src/main/java/thriftlabs/thriftparser/ThriftParser.java
@@ -1,3 +1,5 @@
+package thriftlabs.thriftparser;
+
 // Generated from Thrift.g4 by ANTLR 4.13.2
 import org.antlr.v4.runtime.atn.*;
 import org.antlr.v4.runtime.dfa.DFA;

diff --git a/java/src/test/java/thriftlabs/thriftparser/ParserTest.java b/java/src/test/java/thriftlabs/thriftparser/ParserTest.java
@@ -27,7 +27,12 @@ public void testParse() {
         CommonTokenStream tokens = new CommonTokenStream(lexer);
         ThriftParser parser = new ThriftParser(tokens);
 
-        HeaderContext header = parser.header(0);
-        assertEquals("", header.String(), header.String());
+        ThriftParser.HeaderContext header = parser.header();
+        var n = header.include_();
+        TerminalNode tn = n.LITERAL();
+        Token t = tn.getSymbol();
+        String s = t.getText();
+
+        assertEquals("\"./other.thrift\"", s);
     }
 }