require_relative "helper"

module Vool
  class TestAssignMom < MiniTest::Test
    include VoolCompile

    def setup
      Parfait.boot!(Parfait.default_test_options)
      @compiler = compile_first_method( "local = 5")
      @ins = @compiler.mom_instructions.next
    end

    def test_class_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_slot_gets_self
      assert_equal :frame , @ins.left.slots[0]
    end
    def test_slot_assigns_to_local
      assert_equal :local , @ins.left.slots[-1]
    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

  #otherwise as above, but assigning instance, so should get a SlotLoad
  class TestAssignMomInstanceToLocal < MiniTest::Test
    include VoolCompile
    def setup
      Parfait.boot!(Parfait.default_test_options)
      @compiler = compile_first_method( "@a = 5 ; local = @a")
      @ins = @compiler.mom_instructions.next
    end
    def test_class_compiles
      assert_equal Mom::SlotLoad , @ins.next.class , @ins
    end
  end

  #compiling to an argument should result in different second parameter in the slot array
  class TestAssignToArg < MiniTest::Test
    include VoolCompile

    def setup
      Parfait.boot!(Parfait.default_test_options)
      @compiler = compile_first_method( "arg = 5")
      @ins = @compiler.mom_instructions.next
    end

    def test_class_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_slot_gets_self
      assert_equal :arguments , @ins.left.slots[0]
    end
    def test_slot_assigns_to_local
      assert_equal :arg , @ins.left.slots[-1]
    end
  end

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

end