add unit test

apache · Nov 6, 2020 · 97c2717 · 97c2717
1 parent 4cde13c
commit 97c2717
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/primitive/nsresolver_test.go b/primitive/nsresolver_test.go
@@ -131,3 +131,21 @@ func TesHttpReslverWithSnapshotFileOnce(t *testing.T) {
 		So(Diff(addrs1, srvs), ShouldBeFalse)
 	})
 }
+
+func TestDNSResolver(t *testing.T) {
+	Convey("Test UpdateNameServerAddress Use DNS", t, func() {
+		srvs := []string{
+			"examples.com:9876",
+		}
+
+		hosts, _ := net.LookupHost("examples.com")
+		expectedHosts := make([]string, 0)
+		for _, host:=range hosts{
+			expectedHosts = append(expectedHosts, host + ":9876")
+		}
+		resolver := NewDNSResolver(srvs)
+		addrs := resolver.Resolve()
+
+		So(Diff(expectedHosts, addrs), ShouldBeFalse)
+	})
+}