v2ray-core/app/dns/dns_test.go

package dns_test

import (
	"net"
	"testing"

	"github.com/v2ray/v2ray-core/app/dns"
	netassert "github.com/v2ray/v2ray-core/common/net/testing/assert"
	v2testing "github.com/v2ray/v2ray-core/testing"
)

func TestDnsAdd(t *testing.T) {
	v2testing.Current(t)

	domain := "v2ray.com"
	cache := dns.NewCache()
	ip := cache.Get(domain)
	netassert.IP(ip).IsNil()

	cache.Add(domain, []byte{1, 2, 3, 4})
	ip = cache.Get(domain)
	netassert.IP(ip).Equals(net.IP([]byte{1, 2, 3, 4}))
}