643550eabe
Another thing we can test automatically, we sorta settled on using an uppercase letter in in t.Log and t.Fatal calls. Let's just check for this.
115 lines
2.2 KiB
Go
115 lines
2.2 KiB
Go
package taprw
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/plugin/dnstap/test"
|
|
mwtest "github.com/coredns/coredns/plugin/test"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func testingMsg() (m *dns.Msg) {
|
|
m = new(dns.Msg)
|
|
m.SetQuestion("example.com.", dns.TypeA)
|
|
m.SetEdns0(4097, true)
|
|
return
|
|
}
|
|
|
|
func TestClientQueryResponse(t *testing.T) {
|
|
trapper := test.TrapTapper{Full: true}
|
|
m := testingMsg()
|
|
rw := ResponseWriter{
|
|
Query: m,
|
|
Tapper: &trapper,
|
|
ResponseWriter: &mwtest.ResponseWriter{},
|
|
}
|
|
d := test.TestingData()
|
|
|
|
// will the wire-format msg be reported?
|
|
bin, err := m.Pack()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
return
|
|
}
|
|
d.Packed = bin
|
|
|
|
if err := rw.WriteMsg(m); err != nil {
|
|
t.Fatal(err)
|
|
return
|
|
}
|
|
if l := len(trapper.Trap); l != 2 {
|
|
t.Fatalf("Mmsg %d trapped", l)
|
|
return
|
|
}
|
|
want, err := d.ToClientQuery()
|
|
if err != nil {
|
|
t.Fatal("Testing data must build", err)
|
|
}
|
|
have := trapper.Trap[0]
|
|
if !test.MsgEqual(want, have) {
|
|
t.Fatalf("Query: want: %v\nhave: %v", want, have)
|
|
}
|
|
want, err = d.ToClientResponse()
|
|
if err != nil {
|
|
t.Fatal("Testing data must build", err)
|
|
}
|
|
have = trapper.Trap[1]
|
|
if !test.MsgEqual(want, have) {
|
|
t.Fatalf("Response: want: %v\nhave: %v", want, have)
|
|
}
|
|
}
|
|
|
|
func TestClientQueryResponseWithSendOption(t *testing.T) {
|
|
trapper := test.TrapTapper{Full: true}
|
|
m := testingMsg()
|
|
rw := ResponseWriter{
|
|
Query: m,
|
|
Tapper: &trapper,
|
|
ResponseWriter: &mwtest.ResponseWriter{},
|
|
}
|
|
d := test.TestingData()
|
|
bin, err := m.Pack()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
return
|
|
}
|
|
d.Packed = bin
|
|
|
|
// Do not send both CQ and CR
|
|
o := SendOption{Cq: false, Cr: false}
|
|
rw.Send = &o
|
|
|
|
if err := rw.WriteMsg(m); err != nil {
|
|
t.Fatal(err)
|
|
return
|
|
}
|
|
if l := len(trapper.Trap); l != 0 {
|
|
t.Fatalf("%d msg trapped", l)
|
|
return
|
|
}
|
|
|
|
//Send CQ
|
|
o.Cq = true
|
|
if err := rw.WriteMsg(m); err != nil {
|
|
t.Fatal(err)
|
|
return
|
|
}
|
|
if l := len(trapper.Trap); l != 1 {
|
|
t.Fatalf("%d msg trapped", l)
|
|
return
|
|
}
|
|
|
|
//Send CR
|
|
trapper.Trap = trapper.Trap[:0]
|
|
o.Cq = false
|
|
o.Cr = true
|
|
if err := rw.WriteMsg(m); err != nil {
|
|
t.Fatal(err)
|
|
return
|
|
}
|
|
if l := len(trapper.Trap); l != 1 {
|
|
t.Fatalf("%d msg trapped", l)
|
|
return
|
|
}
|
|
}
|