require_relative "helper"

module Melon
  class TestMethodCollector < MiniTest::Test

    def setup
      Register.machine.boot unless Register.machine.booted
    end

    def parse_collect( input )
      ast = Parser::Ruby22.parse input
      MethodCollector.new.collect(ast)
    end

    def test_no_args
      methods = parse_collect "def meth; @ivar;end"
      assert methods.find{|m| m.name == :meth }
    end

    def test_one_arg
      method = parse_collect("def meth2(arg1); 1;end").first
      assert method.name == :meth2
      assert method.args_type.variable_index(:arg1) , method.args_type.inspect
    end

    def test_three_args
      method = parse_collect("def meth3(yksi,kaksi,kolme); 1;end").first
      assert method.args_type.variable_index(:kolme) , method.args_type.inspect
    end

    def test_one_local
      method = parse_collect("def meth2(arg1); foo = 2 ;end").first
      assert method.locals_type.variable_index(:foo) , method.locals_type.inspect
    end

  end
end