mirror of
https://github.com/folbricht/routedns.git
synced 2026-05-08 00:49:18 -05:00
Frank Olbricht
27 lines
606 B
Go
27 lines
606 B
Go
package rdns
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/miekg/dns"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDNSClientSimpleTCP(t *testing.T) {
|
|
d := NewDNSClient("test-dns", "8.8.8.8:53", "tcp")
|
|
q := new(dns.Msg)
|
|
q.SetQuestion("google.com.", dns.TypeA)
|
|
r, err := d.Resolve(q, ClientInfo{})
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, r.Answer)
|
|
}
|
|
|
|
func TestDNSClientSimpleUDP(t *testing.T) {
|
|
d := NewDNSClient("test-dns", "8.8.8.8:53", "udp")
|
|
q := new(dns.Msg)
|
|
q.SetQuestion("google.com.", dns.TypeA)
|
|
r, err := d.Resolve(q, ClientInfo{})
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, r.Answer)
|
|
}
|