viml/parser/expressions,tests: Add AST freeing, with sanity checks

3 files changed, 5 insertions, 2 deletions

diff --git a/test/helpers.lua b/test/helpers.lua
index 6a42963d7f..83e78ba059 100644
--- a/test/helpers.lua
+++ b/test/helpers.lua
@@ -358,6 +358,8 @@ format_luav = function(v, indent)
     else
       ret = ('%e'):format(v)
     end
+  elseif type(v) == 'nil' then
+    ret = 'nil'
   else
     print(type(v))
     -- Not implemented yet
diff --git a/test/symbolic/klee/viml_expressions_parser.c b/test/symbolic/klee/viml_expressions_parser.c
index a4fbdb6bf4..c0cedceb21 100644
--- a/test/symbolic/klee/viml_expressions_parser.c
+++ b/test/symbolic/klee/viml_expressions_parser.c
@@ -92,6 +92,6 @@ int main(const int argc, const char *const *const argv,
   assert(ast.root != NULL
          || plines[0].size == 0);
   assert(ast.root != NULL || ast.err.msg);
-  // FIXME: check for AST recursiveness
-  // FIXME: free memory and assert no memory leaks
+  viml_pexpr_free_ast(ast);
+  assert(allocated_memory == 0);
 }
diff --git a/test/unit/viml/expressions/parser_spec.lua b/test/unit/viml/expressions/parser_spec.lua
index 407114ff33..9624ced022 100644
--- a/test/unit/viml/expressions/parser_spec.lua
+++ b/test/unit/viml/expressions/parser_spec.lua
@@ -253,6 +253,7 @@ describe('Expressions parser', function()
       end
       eq(exp_highlighting, hls)
     end
+    lib.viml_pexpr_free_ast(east)
   end
   local function hl(group, str, shift)
     return function(next_col)