50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package operand
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/mmcloughlin/avo/reg"
|
|
)
|
|
|
|
func TestMemAsm(t *testing.T) {
|
|
cases := []struct {
|
|
Mem Mem
|
|
Expect string
|
|
}{
|
|
{Mem{Base: reg.EAX}, "(AX)"},
|
|
{Mem{Disp: 16, Base: reg.RAX}, "16(AX)"},
|
|
{Mem{Base: reg.R11, Index: reg.RAX, Scale: 4}, "(R11)(AX*4)"},
|
|
{Mem{Base: reg.R11, Index: reg.RAX, Scale: 1}, "(R11)(AX*1)"},
|
|
{Mem{Base: reg.R11, Index: reg.RAX}, "(R11)"},
|
|
{Mem{Base: reg.R11, Scale: 8}, "(R11)"},
|
|
{Mem{Disp: 2048, Base: reg.R11, Index: reg.RAX, Scale: 8}, "2048(R11)(AX*8)"},
|
|
}
|
|
for _, c := range cases {
|
|
got := c.Mem.Asm()
|
|
if got != c.Expect {
|
|
t.Errorf("%#v.Asm() = %s expected %s", c.Mem, got, c.Expect)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestRegisters(t *testing.T) {
|
|
cases := []struct {
|
|
Op Op
|
|
Expect []reg.Register
|
|
}{
|
|
{reg.R11, []reg.Register{reg.R11}},
|
|
{Mem{Base: reg.EAX}, []reg.Register{reg.EAX}},
|
|
{Mem{Base: reg.RBX, Index: reg.R10}, []reg.Register{reg.RBX, reg.R10}},
|
|
{Imm(42), nil},
|
|
{Rel(42), nil},
|
|
{LabelRef("idk"), nil},
|
|
}
|
|
for _, c := range cases {
|
|
got := Registers(c.Op)
|
|
if !reflect.DeepEqual(got, c.Expect) {
|
|
t.Errorf("Registers(%#v) = %#v expected %#v", c.Op, got, c.Expect)
|
|
}
|
|
}
|
|
}
|