73 lines
2.0 KiB
Go
73 lines
2.0 KiB
Go
package operand
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/mmcloughlin/avo/reg"
|
|
)
|
|
|
|
func TestSymbolString(t *testing.T) {
|
|
cases := []struct {
|
|
Symbol Symbol
|
|
Expect string
|
|
}{
|
|
{Symbol{}, ""},
|
|
{Symbol{Name: "name"}, "name"},
|
|
{Symbol{Name: "static", Static: true}, "static<>"},
|
|
}
|
|
for _, c := range cases {
|
|
got := c.Symbol.String()
|
|
if got != c.Expect {
|
|
t.Errorf("%#v.String() = %s expected %s", c.Symbol, got, c.Expect)
|
|
}
|
|
}
|
|
}
|
|
|
|
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{Disp: -7, Base: reg.RAX}, "-7(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)"},
|
|
{Mem{Symbol: Symbol{Name: "foo"}, Base: reg.StaticBase}, "foo(SB)"},
|
|
{Mem{Symbol: Symbol{Name: "foo"}, Base: reg.StaticBase, Disp: 4}, "foo+4(SB)"},
|
|
{Mem{Symbol: Symbol{Name: "foo"}, Base: reg.StaticBase, Disp: -7}, "foo-7(SB)"},
|
|
{Mem{Symbol: Symbol{Name: "bar", Static: true}, Base: reg.StaticBase, Disp: 4, Index: reg.R11, Scale: 4}, "bar<>+4(SB)(R11*4)"},
|
|
{NewParamAddr("param", 16), "param+16(FP)"},
|
|
}
|
|
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)
|
|
}
|
|
}
|
|
}
|