Fix the testcode

This commit is contained in:
Miek Gieben 2012-05-20 12:44:49 +02:00
parent 983d979698
commit e11c077112

View File

@ -70,23 +70,19 @@ func TestClientEDNS0(t *testing.T) {
func TestClientTsigAXFR(t *testing.T) {
m := new(Msg)
m.SetAxfr("miek.nl.")
m.SetTsig("axfr.", HmacMD5, 300, m.MsgHdr.Id, time.Now().Unix())
m.SetTsig("axfr.", HmacMD5, 300, time.Now().Unix())
c := NewClient()
c.TsigSecret = map[string]string{"axfr.": "so6ZGir4GPAqINNh9U5c3A=="}
c.Net = "tcp"
if err := c.XfrReceive(m, "85.223.71.124:53"); err != nil {
if a, err := c.XfrReceive(m, "85.223.71.124:53"); err != nil {
t.Log("Failed to setup axfr" + err.Error())
t.Fail()
return
}
for {
ex := <-c.ReplyChan
} else {
for ex := range a {
t.Log(ex.Reply.String())
if ex.Error == ErrXfrLast {
break
}
if ex.Error != nil {
t.Logf("Error %s\n", ex.Error.Error())
t.Fail()
@ -98,6 +94,7 @@ func TestClientTsigAXFR(t *testing.T) {
t.Logf("%s\n", ex.Reply.String())
}
}
}
func TestClientAXFRMultipleMessages(t *testing.T) {
m := new(Msg)
@ -107,17 +104,13 @@ func TestClientAXFRMultipleMessages(t *testing.T) {
c.Net = "tcp"
// timeout?
if err := c.XfrReceive(m, "85.223.71.124:53"); err != nil {
if a, err := c.XfrReceive(m, "85.223.71.124:53"); err != nil {
t.Log("Failed to setup axfr" + err.Error())
t.Fail()
return
}
for {
ex := <-c.ReplyChan
} else {
for ex := range a {
t.Log(ex.Reply.String())
if ex.Error == ErrXfrLast {
break
}
if ex.Error != nil {
t.Logf("Error %s\n", ex.Error.Error())
t.Fail()
@ -128,3 +121,4 @@ func TestClientAXFRMultipleMessages(t *testing.T) {
}
}
}
}