github.com/moqsien/xraycore@v1.8.5/app/dns/nameserver_quic_test.go (about)

     1  package dns_test
     2  
     3  import (
     4  	"context"
     5  	"net/url"
     6  	"testing"
     7  	"time"
     8  
     9  	"github.com/google/go-cmp/cmp"
    10  	. "github.com/moqsien/xraycore/app/dns"
    11  	"github.com/moqsien/xraycore/common"
    12  	"github.com/moqsien/xraycore/common/net"
    13  	"github.com/moqsien/xraycore/features/dns"
    14  )
    15  
    16  func TestQUICNameServer(t *testing.T) {
    17  	url, err := url.Parse("quic://dns.adguard.com")
    18  	common.Must(err)
    19  	s, err := NewQUICNameServer(url)
    20  	common.Must(err)
    21  	ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
    22  	ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns.IPOption{
    23  		IPv4Enable: true,
    24  		IPv6Enable: true,
    25  	}, false)
    26  	cancel()
    27  	common.Must(err)
    28  	if len(ips) == 0 {
    29  		t.Error("expect some ips, but got 0")
    30  	}
    31  
    32  	ctx2, cancel := context.WithTimeout(context.Background(), time.Second*5)
    33  	ips2, err := s.QueryIP(ctx2, "google.com", net.IP(nil), dns.IPOption{
    34  		IPv4Enable: true,
    35  		IPv6Enable: true,
    36  	}, true)
    37  	cancel()
    38  	common.Must(err)
    39  	if r := cmp.Diff(ips2, ips); r != "" {
    40  		t.Fatal(r)
    41  	}
    42  }