require_relative "helper"

module Vool
  class TestAssignMom < MiniTest::Test
    include MomCompile

    def setup
      Risc.machine.boot
      @stats = compile_first_method( "local = 5")
      @first = @stats.first
    end

    def test_class_compiles
      assert_equal Mom::SlotConstant , @first.class , @stats
    end
    def test_slot_is_set
      assert @first.left
    end
    def test_slot_starts_at_message
      assert_equal :message , @first.left.known_object
    end
    def test_slot_gets_self
      assert_equal :frame , @first.left.slots[0]
    end
    def test_slot_assigns_to_local
      assert_equal :local , @first.left.slots[-1]
    end
    def test_slot_assigns_something
      assert @stats.first.right
    end
    def test_slot_assigns_int
      assert_equal Mom::IntegerConstant ,  @first.right.class
    end
  end

  #otherwise as above, but assigning instance, so should get a SlotMove
  class TestAssignMomInstanceToLocal < MiniTest::Test
    include MomCompile
    def setup
      Risc.machine.boot
      @stats = compile_first_method( "local = @a")
    end
    def test_class_compiles
      assert_equal Mom::SlotMove , @stats.first.class , @stats
    end
  end

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

    def setup
      Risc.machine.boot
      @stats = compile_first_method( "arg = 5")
      @first = @stats.first
    end

    def test_class_compiles
      assert_equal Mom::SlotConstant , @first.class , @stats
    end
    def test_slot_is_set
      assert @first.left
    end
    def test_slot_starts_at_message
      assert_equal :message , @first.left.known_object
    end
    def test_slot_gets_self
      assert_equal :arguments , @first.left.slots[0]
    end
    def test_slot_assigns_to_local
      assert_equal :arg , @first.left.slots[-1]
    end
  end

  class TestAssignMomToInstance < MiniTest::Test
    include MomCompile
    def setup
      Risc.machine.boot
    end
    def test_assigns_const
      @stats = compile_first_method( "@a = 5")
      assert_equal Mom::SlotConstant , @stats.first.class , @stats
    end
    def test_assigns_move
      @stats = compile_first_method( "@a = arg")
      assert_equal Mom::SlotMove , @stats.first.class , @stats
    end
  end

end