Initial commit

pipe/test_helpers/test_helpers.go

@@ -0,0 +1,98 @@
+package test_helpers
+
+import(
+	"fmt"
+	"github.com/BytemarkHosting/go-pdns/pipe/backend"
+	"testing"
+)
+
+func EmptyDispatch(b *backend.Backend, q *backend.Query)([]*backend.Response,error) {
+	return nil, nil
+}
+
+
+func AssertEqualString(t *testing.T, a, b, msg string) {
+	if a != b {
+		t.Logf(fmt.Sprintf("%s:'\n%s\n'\nshould be the same as:'\n%s\n'", msg, a, b))
+		t.FailNow()
+	}
+}
+
+func AssertEqualInt(t *testing.T, a, b int, msg string) {
+	if a != b {
+		t.Logf(fmt.Sprintf("%s: %d should == %d", msg, a, b))
+		t.FailNow()
+	}
+}
+
+func RefuteError(t *testing.T, err error, msg string) {
+	if err != nil {
+		t.Logf("Error: %s was expected to be nil", msg, err)
+		t.FailNow()
+	}
+}
+
+func Assert(t *testing.T, condition bool, msg string) {
+	if !condition {
+		t.Log(msg)
+		t.FailNow()
+	}
+}
+
+func FakeQuery(protoVersion int) *backend.Query {
+	if protoVersion < 1 || protoVersion > 3 {
+		panic("Invalid protoVersion") 
+	}
+
+	q := backend.Query{
+		ProtocolVersion: protoVersion,
+		QClass: "IN",
+		QType: "ANY",
+		QName: "example.com",
+		Id: "-1",
+		RemoteIpAddress: "127.0.0.2",
+	}
+	if protoVersion > 1 {
+		q.LocalIpAddress = "127.0.0.1"
+	}
+	if protoVersion > 2 {
+		q.EdnsSubnetAddress = "127.0.0.3"
+	}
+
+	return &q
+}
+
+func FakeResponse(protoVersion int) *backend.Response {
+	if protoVersion < 1 || protoVersion > 3 {
+		panic("Invalid protoVersion") 
+	}
+
+	r := backend.Response{
+		ProtocolVersion: protoVersion,
+		QClass: "IN",
+		QType: "ANY",
+		QName: "example.com",
+		Id: "-1",
+		TTL: "3600",
+		Content: "foo",
+	}
+	if protoVersion > 2 {
+		r.ScopeBits = "24"
+		r.Auth = "auth"
+	}
+
+	return &r
+}
+
+func FakeQueryString(t *testing.T, protoVersion int) string {
+	str, err := FakeQuery(protoVersion).String()
+	RefuteError(t, err, "Failed to serialise test query")
+	return str
+}
+
+func FakeResponseString(t *testing.T, protoVersion int) string {
+	str, err := FakeResponse(protoVersion).String()
+	RefuteError(t, err, "Failed to serialise test response")
+	return str
+}
+