restrict the root, but expand for testing

lib/parser/salama.rb

@@ -31,8 +31,7 @@ module Parser
     include Operators
     include ModuleDef
 
-    rule(:root_body)    {( class_definition | function_definition |
+    rule(:root_body)    {( class_definition | function_definition  )}
-                          expression  | operator_expression | call_site  )}
     rule(:root)         { root_body.repeat.as(:expression_list) }
   end
 end

test/setup.rb

@@ -17,3 +17,10 @@ require 'salama-reader'
 
 require "minitest"
 require "minitest/autorun"
+
+# for testing add expression as allowed root
+# bit of a hack: the directory name should be passed as the root if the parse fails
+# but moving on (and coming back)
+Parser::Salama.class_eval do
+  rule(:root_body)    {( class_definition | function_definition | expression  )}
+end