6bb08ffee4
* Easier way to dnstap? * Remove unnecessary function parameter from Tapper * golint * golint 2 * Proxy dnstap tests * README.md & doc * net.IP * Proxy test was incorrect * Small changes * Update README.md * Was not reporting dnstap errors + test * Wasn't working at all, now it's ok * Thanks Travis
39 lines
949 B
Go
39 lines
949 B
Go
package msg
|
|
|
|
import (
|
|
"net"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/coredns/coredns/plugin/test"
|
|
"github.com/coredns/coredns/request"
|
|
|
|
tap "github.com/dnstap/golang-dnstap"
|
|
"github.com/miekg/dns"
|
|
)
|
|
|
|
func testRequest(t *testing.T, expected Builder, r request.Request) {
|
|
d := Builder{}
|
|
d.Addr(r.W.RemoteAddr())
|
|
if d.SocketProto != expected.SocketProto ||
|
|
d.SocketFam != expected.SocketFam ||
|
|
!reflect.DeepEqual(d.Address, expected.Address) ||
|
|
d.Port != expected.Port {
|
|
t.Fatalf("expected: %v, have: %v", expected, d)
|
|
return
|
|
}
|
|
}
|
|
func TestRequest(t *testing.T) {
|
|
testRequest(t, Builder{
|
|
SocketProto: tap.SocketProtocol_UDP,
|
|
SocketFam: tap.SocketFamily_INET,
|
|
Address: net.ParseIP("10.240.0.1"),
|
|
Port: 40212,
|
|
}, testingRequest())
|
|
}
|
|
func testingRequest() request.Request {
|
|
m := new(dns.Msg)
|
|
m.SetQuestion("example.com.", dns.TypeA)
|
|
m.SetEdns0(4097, true)
|
|
return request.Request{W: &test.ResponseWriter{}, Req: m}
|
|
}
|