require_relative "helper"

module Risc
  class TestMethodCompiler < MiniTest::Test
    include ScopeHelper

    def setup
      Parfait.boot!
    end

    def in_test_vool(str)
      input = in_Test(str)
      vool = RubyX::RubyXCompiler.new(input).ruby_to_vool
      vool.to_mom(nil)
      vool
    end
    def create_method
      vool = in_test_vool("def meth; @ivar = 5;end")
      test = Parfait.object_space.get_class_by_name(:Test)
      test.get_method(:meth)
    end

    def test_method_has_source
      method = create_method
      assert_equal Vool::IvarAssignment ,  method.source.class
    end

    def test_method_has_no_locals
      method = create_method
      assert_equal 1 , method.frame_type.instance_length
    end

    def test_method_has_no_args
      method = create_method
      assert_equal 1 , method.args_type.instance_length
    end

    def test_creates_method_in_class
      method = create_method
      assert method , "No method created"
      assert_equal Parfait::VoolMethod , method.class
    end

    def test_creates_method_statement_in_class
      clazz = in_test_vool("def meth; @ivar = 5 ;end")
      assert_equal Vool::Statements , clazz.body.class
      assert_equal Vool::MethodStatement , clazz.body.first.class
    end

    def test_method_statement_has_class
      input = in_Test("def meth; @ivar = 5;end")
      vool = RubyX::RubyXCompiler.new(input).ruby_to_vool
      clazz = vool.to_mom(nil)
      assert vool.body.first.clazz
    end

    def test_parfait_class_creation
      input = in_Test("def meth; @ivar = 5;end")
      vool = RubyX::RubyXCompiler.new(input).ruby_to_vool
      clazz = vool.to_mom(nil)
      assert_equal Parfait::Class , vool.body.first.clazz.class
    end

    def test_typed_method_instance_type
      vool = in_test_vool("def meth; @ivar = 5; @ibar = 4;end")
      test = Parfait.object_space.get_class_by_name(:Test)
      method = test.instance_type.get_method(:meth)
      assert_equal 1, method.for_type.variable_index(:ivar)
      assert_equal 2, method.for_type.variable_index(:ibar)
    end

    def test_vool_method_has_one_local
      vool = in_test_vool("def meth; local = 5 ; a = 6;end")
      test = Parfait.object_space.get_class_by_name(:Test)
      method = test.get_method(:meth)
      assert_equal 3 , method.frame_type.instance_length
      assert_equal 1 , method.frame_type.variable_index(:local)
      assert_equal 2 , method.frame_type.variable_index(:a)
    end

    def test_typed_method_has_one_local
      vool = in_test_vool("def meth; local = 5 ; a = 6;end")
      test = Parfait.object_space.get_class_by_name(:Test)
      method = test.instance_type.get_method(:meth)
      assert_equal 3 , method.frame_type.instance_length
      assert_equal 1 , method.frame_type.variable_index(:local)
    end

  end
end