pass: test for liveness

pass/reg_test.go

@@ -0,0 +1,90 @@
+package pass_test
+
+import (
+	"testing"
+
+	"github.com/mmcloughlin/avo/reg"
+
+	"github.com/mmcloughlin/avo/pass"
+
+	"github.com/mmcloughlin/avo"
+	"github.com/mmcloughlin/avo/build"
+	"github.com/mmcloughlin/avo/operand"
+)
+
+func TestLivenessBasic(t *testing.T) {
+	// Build: a = 1, b = 2, a = a+b
+	ctx := build.NewContext()
+	ctx.Function("add")
+	a := ctx.GP64v()
+	b := ctx.GP64v()
+	ctx.MOVQ(operand.Imm(1), a)
+	ctx.MOVQ(operand.Imm(2), b)
+	ctx.ADDQ(a, b)
+
+	AssertLiveness(t, ctx,
+		[][]reg.Register{
+			{},
+			{a},
+			{a, b},
+		},
+		[][]reg.Register{
+			{a},
+			{a, b},
+			{},
+		},
+	)
+}
+
+func AssertLiveness(t *testing.T, ctx *build.Context, in, out [][]reg.Register) {
+	fn := ConstructLiveness(t, ctx)
+	is := fn.Instructions()
+
+	if len(in) != len(is) || len(out) != len(is) {
+		t.Fatalf("%d instructions: %d/%d in/out expectations", len(is), len(in), len(out))
+	}
+
+	for idx, i := range is {
+		AssertRegistersMatchSet(t, in[idx], i.LiveIn)
+		AssertRegistersMatchSet(t, out[idx], i.LiveOut)
+	}
+}
+
+func AssertRegistersMatchSet(t *testing.T, rs []reg.Register, s map[reg.ID]bool) {
+	if len(rs) != len(s) {
+		t.Fatalf("size mismatch")
+	}
+	for _, r := range rs {
+		if _, found := s[r.ID()]; !found {
+			t.Fatalf("missing register ID %v", r.ID())
+		}
+	}
+}
+
+func ConstructLiveness(t *testing.T, ctx *build.Context) *avo.Function {
+	f, errs := ctx.Result()
+	if errs != nil {
+		for _, err := range errs {
+			t.Error(err)
+		}
+		t.FailNow()
+	}
+
+	if len(f.Functions) != 1 {
+		t.Fatalf("expect 1 function")
+	}
+	fn := f.Functions[0]
+
+	passes := []func(*avo.Function) error{
+		pass.LabelTarget,
+		pass.CFG,
+		pass.Liveness,
+	}
+	for _, p := range passes {
+		if err := p(fn); err != nil {
+			t.Fatal(err)
+		}
+	}
+
+	return fn
+}