package dns_test import ( "context" "net/url" "testing" "time" "github.com/google/go-cmp/cmp" . "github.com/xtls/xray-core/app/dns" "github.com/xtls/xray-core/common" "github.com/xtls/xray-core/common/net" "github.com/xtls/xray-core/features/dns" ) func TestQUICNameServer(t *testing.T) { url, err := url.Parse("quic://dns.adguard.com") common.Must(err) s, err := NewQUICNameServer(url, QueryStrategy_USE_IP) common.Must(err) ctx, cancel := context.WithTimeout(context.Background(), time.Second*2) ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns.IPOption{ IPv4Enable: true, IPv6Enable: true, }, false) cancel() common.Must(err) if len(ips) == 0 { t.Error("expect some ips, but got 0") } ctx2, cancel := context.WithTimeout(context.Background(), time.Second*5) ips2, err := s.QueryIP(ctx2, "google.com", net.IP(nil), dns.IPOption{ IPv4Enable: true, IPv6Enable: true, }, true) cancel() common.Must(err) if r := cmp.Diff(ips2, ips); r != "" { t.Fatal(r) } } func TestQUICNameServerWithIPv4Override(t *testing.T) { url, err := url.Parse("quic://dns.adguard.com") common.Must(err) s, err := NewQUICNameServer(url, QueryStrategy_USE_IP4) common.Must(err) ctx, cancel := context.WithTimeout(context.Background(), time.Second*2) ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns.IPOption{ IPv4Enable: true, IPv6Enable: true, }, false) cancel() common.Must(err) if len(ips) == 0 { t.Error("expect some ips, but got 0") } for _, ip := range ips { if len(ip) != net.IPv4len { t.Error("expect only IPv4 response from DNS query") } } } func TestQUICNameServerWithIPv6Override(t *testing.T) { url, err := url.Parse("quic://dns.adguard.com") common.Must(err) s, err := NewQUICNameServer(url, QueryStrategy_USE_IP6) common.Must(err) ctx, cancel := context.WithTimeout(context.Background(), time.Second*2) ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns.IPOption{ IPv4Enable: true, IPv6Enable: true, }, false) cancel() common.Must(err) if len(ips) == 0 { t.Error("expect some ips, but got 0") } for _, ip := range ips { if len(ip) != net.IPv6len { t.Error("expect only IPv6 response from DNS query") } } }