always list of methods in class

improve vool test dirs
2018-06-30 10:08:18 +03:00
parent 46d8f5002f
commit 4a7cc72732
21 changed files with 7 additions and 10 deletions
--- a/lib/vool/class_statement.rb
+++ b/lib/vool/class_statement.rb
@ -24,9 +24,10 @@ module Vool

    def to_mom( _ )
      create_class_object
-      mom = nil #return mom for test purpose
-      self.each do |node|
-        mom = node.to_mom(@clazz) if node.is_a?(MethodStatement)
+      mom = []
+      body.statements.each do |node|
+        raise "Only methods for now #{node}" unless node.is_a?(MethodStatement)
+        mom << node.to_mom(@clazz)
      end
      mom
    end