github.com/xmplusdev/xmcore@v1.8.11-0.20240412132628-5518b55526af/app/dns/nameserver_doh_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/xmplusdev/xmcore/app/dns"
    11  	"github.com/xmplusdev/xmcore/common"
    12  	"github.com/xmplusdev/xmcore/common/net"
    13  	dns_feature "github.com/xmplusdev/xmcore/features/dns"
    14  )
    15  
    16  func TestDOHNameServer(t *testing.T) {
    17  	url, err := url.Parse("https+local://1.1.1.1/dns-query")
    18  	common.Must(err)
    19  
    20  	s := NewDoHLocalNameServer(url, QueryStrategy_USE_IP)
    21  	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
    22  	ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.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  
    33  func TestDOHNameServerWithCache(t *testing.T) {
    34  	url, err := url.Parse("https+local://1.1.1.1/dns-query")
    35  	common.Must(err)
    36  
    37  	s := NewDoHLocalNameServer(url, QueryStrategy_USE_IP)
    38  	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
    39  	ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{
    40  		IPv4Enable: true,
    41  		IPv6Enable: true,
    42  	}, false)
    43  	cancel()
    44  	common.Must(err)
    45  	if len(ips) == 0 {
    46  		t.Error("expect some ips, but got 0")
    47  	}
    48  
    49  	ctx2, cancel := context.WithTimeout(context.Background(), time.Second*5)
    50  	ips2, err := s.QueryIP(ctx2, "google.com", net.IP(nil), dns_feature.IPOption{
    51  		IPv4Enable: true,
    52  		IPv6Enable: true,
    53  	}, true)
    54  	cancel()
    55  	common.Must(err)
    56  	if r := cmp.Diff(ips2, ips); r != "" {
    57  		t.Fatal(r)
    58  	}
    59  }
    60  
    61  func TestDOHNameServerWithIPv4Override(t *testing.T) {
    62  	url, err := url.Parse("https+local://1.1.1.1/dns-query")
    63  	common.Must(err)
    64  
    65  	s := NewDoHLocalNameServer(url, QueryStrategy_USE_IP4)
    66  	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
    67  	ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{
    68  		IPv4Enable: true,
    69  		IPv6Enable: true,
    70  	}, false)
    71  	cancel()
    72  	common.Must(err)
    73  	if len(ips) == 0 {
    74  		t.Error("expect some ips, but got 0")
    75  	}
    76  
    77  	for _, ip := range ips {
    78  		if len(ip) != net.IPv4len {
    79  			t.Error("expect only IPv4 response from DNS query")
    80  		}
    81  	}
    82  }
    83  
    84  func TestDOHNameServerWithIPv6Override(t *testing.T) {
    85  	url, err := url.Parse("https+local://1.1.1.1/dns-query")
    86  	common.Must(err)
    87  
    88  	s := NewDoHLocalNameServer(url, QueryStrategy_USE_IP6)
    89  	ctx, cancel := context.WithTimeout(context.Background(), time.Second*5)
    90  	ips, err := s.QueryIP(ctx, "google.com", net.IP(nil), dns_feature.IPOption{
    91  		IPv4Enable: true,
    92  		IPv6Enable: true,
    93  	}, false)
    94  	cancel()
    95  	common.Must(err)
    96  	if len(ips) == 0 {
    97  		t.Error("expect some ips, but got 0")
    98  	}
    99  
   100  	for _, ip := range ips {
   101  		if len(ip) != net.IPv6len {
   102  			t.Error("expect only IPv6 response from DNS query")
   103  		}
   104  	}
   105  }