40 lines
1.1 KiB
Ruby
40 lines
1.1 KiB
Ruby
require_relative "helper"
|
|
|
|
module Melon
|
|
module Compilers
|
|
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_equal 2 , 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
|
|
end
|