require_relative 'helper'

class TestMemory < MiniTest::Test
  include ArmHelper

  def test_ldr
    code = @machine.ldr  :r0, :r0
    assert_code code, :ldr ,  [0x00,0x00,0x90,0xe5] #e5 90 00 00
  end
  def test_ldr_const_offset
    code = @machine.ldr  :r0, :r0 ,  4
    assert_code code, :ldr ,  [0x04,0x00,0x90,0xe5] #e5 90 00 04
  end
  def test_ldr_reg_offset
    code = @machine.ldr  :r3, :r4 , :r5
    assert_code code, :ldr ,  [0x05,0x30,0x94,0xe7] #e7 94 30 05
  end
  def test_ldr_reg_shift2
    code = @machine.ldr  :r3, :r4 , :r5 , :shift_lsl => 2
    assert_code code, :ldr ,  [0x05,0x31,0x94,0xe7] #e7 94 31 05
  end
  def test_ldr_reg_shift3
    code = @machine.ldr  :r3, :r4 , :r5 , :shift_lsl => 3
    assert_code code, :ldr ,  [0x85,0x31,0x94,0xe7] #e7 94 31 85
  end
  def test_ldrb
    code = @machine.ldrb  :r0, :r0
    assert_code code, :ldrb ,  [0x00,0x00,0xd0,0xe5] #e5 d0 00 00
  end
  def test_ldrb_const
    code = @machine.ldrb  :r0, :r0 , 12
    assert_code code, :ldrb ,  [0x0c,0x00,0xd0,0xe5] #e5 d0 00 0c
  end
  def test_ldrb_reg_offset
    code = @machine.ldrb  :r0, :r1 , :r2
    assert_code code, :ldrb ,  [0x02,0x00,0xd1,0xe7] #e7 d1 00 02
  end

  def test_str
    code = @machine.str  :r0, :r1
    assert_code code, :str ,  [0x00,0x00,0x81,0xe5] #e5 81 00 00
  end
  def test_str_const_offset
    code = @machine.str  :r1, :r2 ,  4
    assert_code code, :str ,  [0x04,0x10,0x82,0xe5] #e5 82 10 04
  end
  def test_str_reg_offset
    code = @machine.str  :r3, :r4 , :r5
    assert_code code, :str ,  [0x05,0x30,0x84,0xe7] #e7 84 30 05
  end
  def test_str_reg_shift
    code = @machine.str  :r3, :r4 , :r5 , :shift_lsl => 2
    assert_code code, :str ,  [0x05,0x31,0x84,0xe7] #e7 84 31 05
  end

  def test_strb_inc   #THIS IS BROKEN, but so is gnu as, as it produces same output for ! or not
    code = @machine.strb  :r0, :r2 , 1 , pre_post_index: 1  # pre_post_index is autoinc (r2 here)
    assert_code code, :strb ,  [0x01,0x00,0xc2,0xe5] #e5 e2 00 01
  end

  def test_strb
    code = @machine.strb  :r0, :r0
    assert_code code, :strb ,  [0x00,0x00,0xc0,0xe5] #e5 c0 00 00
  end

  def test_strb_const
    code = @machine.strb  :r1, :r2 , 4
    assert_code code, :strb ,  [0x04,0x10,0xc2,0xe5] #e5 c2 10 04
  end
  def test_strb_reg_offset
    code = @machine.strb  :r1, :r2 , :r3
    assert_code code, :strb ,  [0x03,0x10,0xc2,0xe7] #e7 c2 10 03
  end
end