38 lines
1,006 B
Go
38 lines
1,006 B
Go
package dnstest
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func TestMultiWriteMsg(t *testing.T) {
|
|
w := &responseWriter{}
|
|
record := NewMultiRecorder(w)
|
|
|
|
responseTestName := "testmsg.example.org."
|
|
responseTestMsg := new(dns.Msg)
|
|
responseTestMsg.SetQuestion(responseTestName, dns.TypeA)
|
|
|
|
record.WriteMsg(responseTestMsg)
|
|
record.WriteMsg(responseTestMsg)
|
|
|
|
if len(record.Msgs) != 2 {
|
|
t.Fatalf("Expected 2 messages to be written, but instead found %d\n", len(record.Msgs))
|
|
}
|
|
if record.Len != responseTestMsg.Len()*2 {
|
|
t.Fatalf("Expected the bytes written counter to be %d, but instead found %d\n", responseTestMsg.Len()*2, record.Len)
|
|
}
|
|
}
|
|
|
|
func TestMultiWrite(t *testing.T) {
|
|
w := &responseWriter{}
|
|
record := NewRecorder(w)
|
|
responseTest := []byte("testmsg.example.org.")
|
|
|
|
record.Write(responseTest)
|
|
record.Write(responseTest)
|
|
if record.Len != len(responseTest)*2 {
|
|
t.Fatalf("Expected the bytes written counter to be %d, but instead found %d\n", len(responseTest)*2, record.Len)
|
|
}
|
|
}
|