Browse Source

Minor tweaks to logging and test running

Ask Bjørn Hansen 10 years ago
parent
commit
c86fbed741
2 changed files with 7 additions and 7 deletions
  1. 2 2
      serve.go
  2. 5 5
      serve_test.go

+ 2 - 2
serve.go

@@ -24,8 +24,8 @@ func serve(w dns.ResponseWriter, req *dns.Msg, z *Zone) {
 
 	qtype := req.Question[0].Qtype
 
-	logPrintf("[zone %s] incoming %s %s %d from %s\n", z.Origin, req.Question[0].Name,
-		dns.TypeToString[qtype], req.MsgHdr.Id, w.RemoteAddr())
+	logPrintf("[zone %s] incoming  %s %s (id %d) from %s\n", z.Origin, req.Question[0].Name,
+		dns.TypeToString[qtype], req.Id, w.RemoteAddr())
 
 	// Global meter
 	metrics.Get("queries").(metrics.Meter).Mark(1)

+ 5 - 5
serve_test.go

@@ -4,7 +4,6 @@ import (
 	"net"
 	"strings"
 	"sync"
-	"time"
 
 	"github.com/abh/geodns/Godeps/_workspace/src/github.com/miekg/dns"
 	. "github.com/abh/geodns/Godeps/_workspace/src/gopkg.in/check.v1"
@@ -30,9 +29,9 @@ func (s *ServeSuite) SetUpSuite(c *C) {
 	setupRootZone()
 	zonesReadDir("dns", Zones)
 
-	go listenAndServe(PORT)
-
-	time.Sleep(200 * time.Millisecond)
+	// listenAndServe returns after listning on udp + tcp, so just
+	// wait for it before continuing
+	listenAndServe(PORT)
 }
 
 func (s *ServeSuite) TestServing(c *C) {
@@ -258,9 +257,10 @@ func exchange(c *C, name string, dnstype uint16) *dns.Msg {
 
 func dorequest(c *C, msg *dns.Msg) *dns.Msg {
 	cli := new(dns.Client)
+	// cli.ReadTimeout = 2 * time.Second
 	r, _, err := cli.Exchange(msg, "127.0.0.1"+PORT)
 	if err != nil {
-		c.Log("err", err)
+		c.Logf("request err '%s': %s", msg.String(), err)
 		c.Fail()
 	}
 	return r