test: WIP extend right nullable test

igordejanovic · Oct 13, 2023 · b45ce82 · b45ce82
1 parent fb8085f
commit b45ce82
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/tests/src/glr/special/right_nullable/mod.rs b/tests/src/glr/special/right_nullable/mod.rs
@@ -1,5 +1,5 @@
+use rustemo::{lr::builder::TreeBuilder, rustemo_mod, glr::gss::GssHead};
 use rustemo::parser::Parser;
-use rustemo::rustemo_mod;
 use rustemo_compiler::output_cmp;
 
 rustemo_mod!(lang, "/src/glr/special/right_nullable");
@@ -10,4 +10,22 @@ use self::lang::LangParser;
 fn glr_special_right_nullable_g2() {
     let forest = LangParser::new().parse("aa").unwrap();
     assert_eq!(forest.solutions(), 1);
+    (1..=forest.solutions()).for_each(|i| {
+        let tree = forest.get_tree(i - 1);
+        let mut builder = TreeBuilder::new();
+        output_cmp!(
+            &format!("src/glr/special/right_nullable/tree_{}.ast", i),
+            format!("{:#?}", tree.unwrap().build::<TreeBuilder<
+                '_,
+                str,
+                lang::ProdKind,
+                lang::TokenKind,
+            >, GssHead<
+                '_,
+                str,
+                lang::State,
+                lang::TokenKind,
+            >, lang::State>(&mut builder))
+        );
+    })
 }