require_relative "../helper"

module VoolBlocks
  class TestAssignMom < MiniTest::Test
    include MomCompile

    def setup
      Parfait.boot!
      Risc::Builtin.boot_functions
      @ins = compile_first_block( "local = 5")
    end

    def test_block_compiles
      assert_equal Mom::SlotLoad , @ins.class , @ins
    end
    def test_slot_is_set
      assert @ins.left
    end
    def test_slot_starts_at_message
      assert_equal :message , @ins.left.known_object
    end
    def test_slots_left
      assert_equal [:frame , :local] , @ins.left.slots
    end
    def test_slot_assigns_something
      assert @ins.right
    end
    def test_slot_assigns_int
      assert_equal Mom::IntegerConstant ,  @ins.right.known_object.class
    end
  end

  class TestAssignMomInstanceToLocal < MiniTest::Test
    include MomCompile
    def setup
      Parfait.boot!
      @ins = compile_first_block( "local = @a" , "@a = 5") #second arg in method scope
    end
    def test_class_compiles
      assert_equal Mom::SlotLoad , @ins.class , @ins
    end
    def test_slots_left
      assert_equal [:frame, :local] , @ins.left.slots
    end
    def test_slots_right
      assert_equal [:receiver, :a] , @ins.right.slots
    end
  end

  class TestAssignToArg < MiniTest::Test
    include MomCompile

    def setup
      Parfait.boot!
      @ins = compile_first_block( "arg = 5")
    end

    def test_class_compiles
      assert_equal Mom::SlotLoad , @ins.class , @ins
    end
    def test_slot_is_set
      assert @ins.left
    end
    def test_slots_left
      assert_equal [:caller, :arguments, :arg] , @ins.left.slots
    end
  end

  class TestAssignMomToInstance < MiniTest::Test
    include MomCompile
    def setup
      Parfait.boot!
    end
    def test_assigns_const
      @ins = compile_first_method( "@a = 5")
      assert_equal Mom::SlotLoad , @ins.class , @ins
      assert_equal Mom::IntegerConstant , @ins.right.known_object.class , @ins
    end
    def test_assigns_move
      @ins = compile_first_method( "@a = arg")
      assert_equal Mom::SlotLoad , @ins.class , @ins
      assert_equal Mom::SlotDefinition , @ins.right.class , @ins
    end
  end

end