Add tests of querylog data

server/querylog_test.go

@@ -0,0 +1,56 @@
+package server
+
+import (
+	"testing"
+
+	"github.com/abh/geodns/v3/querylog"
+	"github.com/miekg/dns"
+)
+
+type testLogger struct {
+	lastLog querylog.Entry
+}
+
+func (l *testLogger) Close() error {
+	return nil
+}
+
+func (l *testLogger) Write(ql *querylog.Entry) error {
+	l.lastLog = *ql
+	return nil
+}
+
+func (l *testLogger) Last() querylog.Entry {
+	// l.logged = false
+	return l.lastLog
+}
+
+func testQueryLog(srv *Server) func(*testing.T) {
+
+	tlog := &testLogger{}
+
+	srv.SetQueryLogger(tlog)
+
+	return func(t *testing.T) {
+
+		r := exchange(t, "www-alias.example.com.", dns.TypeA)
+		expected := "geo.bitnames.com."
+		answer := r.Answer[0].(*dns.CNAME).Target
+		if answer != expected {
+			t.Logf("expected CNAME %s, got %s", expected, answer)
+			t.Fail()
+		}
+
+		last := tlog.Last()
+		// t.Logf("last log: %+v", last)
+
+		if last.Name != "www-alias.example.com." {
+			t.Logf("didn't get qname in Name querylog")
+			t.Fail()
+		}
+		if last.LabelName != "www" {
+			t.Logf("LabelName didn't contain resolved label")
+			t.Fail()
+		}
+	}
+}

server/serve_test.go

@@ -41,6 +41,8 @@ func TestServe(t *testing.T) {
 
 	t.Run("Serving", testServing)
 
+	t.Run("QueryLog", testQueryLog(srv))
+
 	// todo: run test queries?
 
 	cancel()