From 066dc257f92841b2a26cc7096dbb22990bff681f Mon Sep 17 00:00:00 2001
From: Brian Picciano <mediocregopher@gmail.com>
Date: Sun, 21 Aug 2016 10:36:19 -0600
Subject: [PATCH] wrote some basic tests for the things I care about so far

---
 expr/build.go      | 14 +++++++
 expr/build_test.go | 99 ++++++++++++++++++++++++++++++++++++++++++++++
 main.go            | 10 +----
 3 files changed, 114 insertions(+), 9 deletions(-)
 create mode 100644 expr/build_test.go

diff --git a/expr/build.go b/expr/build.go
index 80510a9..e517f19 100644
--- a/expr/build.go
+++ b/expr/build.go
@@ -7,11 +7,25 @@ import (
 	"llvm.org/llvm/bindings/go/llvm"
 )
 
+func init() {
+	log.Printf("initializing llvm")
+	llvm.LinkInMCJIT()
+	llvm.InitializeNativeTarget()
+	llvm.InitializeNativeAsmPrinter()
+}
+
 type BuildCtx struct {
 	B llvm.Builder
 	M llvm.Module
 }
 
+func NewBuildCtx(moduleName string) BuildCtx {
+	return BuildCtx{
+		B: llvm.NewBuilder(),
+		M: llvm.NewModule(moduleName),
+	}
+}
+
 func (bctx BuildCtx) Build(ctx Ctx, stmts ...Statement) llvm.Value {
 	var lastVal llvm.Value
 	for _, stmt := range stmts {
diff --git a/expr/build_test.go b/expr/build_test.go
new file mode 100644
index 0000000..fe5e6ac
--- /dev/null
+++ b/expr/build_test.go
@@ -0,0 +1,99 @@
+package expr
+
+import (
+	"fmt"
+	. "testing"
+
+	"llvm.org/llvm/bindings/go/llvm"
+)
+
+func buildTest(t *T, expected int64, stmts ...Statement) {
+	fmt.Println("-----------------------------------------")
+	ctx := NewCtx()
+	bctx := NewBuildCtx("")
+
+	fn := llvm.AddFunction(bctx.M, "", llvm.FunctionType(llvm.Int64Type(), []llvm.Type{}, false))
+	fnbl := llvm.AddBasicBlock(fn, "")
+	bctx.B.SetInsertPoint(fnbl, fnbl.FirstInstruction())
+	out := bctx.Build(ctx, stmts...)
+	bctx.B.CreateRet(out)
+
+	fmt.Println("######## dumping IR")
+	bctx.M.Dump()
+	fmt.Println("######## done dumping IR")
+
+	if err := llvm.VerifyModule(bctx.M, llvm.ReturnStatusAction); err != nil {
+		t.Fatal(err)
+	}
+
+	eng, err := llvm.NewExecutionEngine(bctx.M)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	res := eng.RunFunction(fn, []llvm.GenericValue{}).Int(false)
+	if int64(res) != expected {
+		t.Errorf("expected:[%T]%v actual:[%T]%v", expected, expected, res, res)
+	}
+}
+
+func TestAdd(t *T) {
+	buildTest(t, 2,
+		NewStatement(Macro("add"), Int(1), Int(1)))
+	buildTest(t, 4,
+		NewStatement(Macro("add"), Int(1),
+			NewStatement(Macro("add"), Int(1), Int(2))))
+	buildTest(t, 6,
+		NewStatement(Macro("add"),
+			NewStatement(Macro("add"), Int(1), Int(2)),
+			NewStatement(Macro("add"), Int(1), Int(2))))
+}
+
+func TestBind(t *T) {
+	buildTest(t, 2,
+		NewStatement(Macro("bind"), Identifier("A"), Int(1)),
+		NewStatement(Macro("add"), Identifier("A"), Int(1)))
+	buildTest(t, 2,
+		NewStatement(Macro("bind"), Identifier("A"), Int(1)),
+		NewStatement(Macro("add"), Identifier("A"), Identifier("A")))
+	buildTest(t, 2,
+		NewStatement(Macro("bind"), Identifier("A"), NewTuple(Int(1), Int(1))),
+		NewStatement(Macro("add"), Identifier("A")))
+	buildTest(t, 3,
+		NewStatement(Macro("bind"), Identifier("A"), NewTuple(Int(1), Int(1))),
+		NewStatement(Macro("add"), Int(1),
+			NewStatement(Macro("add"), Identifier("A"))))
+	buildTest(t, 4,
+		NewStatement(Macro("bind"), Identifier("A"), NewTuple(Int(1), Int(1))),
+		NewStatement(Macro("add"),
+			NewStatement(Macro("add"), Identifier("A")),
+			NewStatement(Macro("add"), Identifier("A"))))
+}
+
+func TestOp(t *T) {
+	incr := NewStatement(Macro("op"),
+		NewList(
+			NewStatement(Macro("add"), Int(1), NewStatement(Macro("in"))),
+		),
+	)
+
+	// bound op
+	buildTest(t, 2,
+		NewStatement(Macro("bind"), Identifier("incr"), incr),
+		NewStatement(Identifier("incr"), Int(1)))
+
+	// double bound op
+	buildTest(t, 3,
+		NewStatement(Macro("bind"), Identifier("incr"), incr),
+		NewStatement(Identifier("incr"),
+			NewStatement(Identifier("incr"), Int(1))))
+
+	// anon op
+	buildTest(t, 2,
+		NewStatement(incr, Int(1)))
+
+	// double anon op
+	buildTest(t, 3,
+		NewStatement(incr,
+			NewStatement(incr, Int(1))))
+}
diff --git a/main.go b/main.go
index 7d43474..3113565 100644
--- a/main.go
+++ b/main.go
@@ -18,17 +18,9 @@ func main() {
 	//	fmt.Println(e)
 	//}
 
-	log.Printf("initializing llvm")
-	llvm.LinkInMCJIT()
-	llvm.InitializeNativeTarget()
-	llvm.InitializeNativeAsmPrinter()
-
 	log.Printf("initializing build context")
 	ctx := expr.NewCtx()
-	bctx := expr.BuildCtx{
-		B: llvm.NewBuilder(),
-		M: llvm.NewModule("my_module"),
-	}
+	bctx := expr.NewBuildCtx("my_module")
 
 	log.Printf("making program")
 	add := expr.Macro("add")