|
|
@@ -136,3 +136,108 @@ func TestResolveTCPAddrs(t *testing.T) {
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+
|
|
|
+func TestURLsEqual(t *testing.T) {
|
|
|
+ defer func() { resolveTCPAddr = net.ResolveTCPAddr }()
|
|
|
+ resolveTCPAddr = func(network, addr string) (*net.TCPAddr, error) {
|
|
|
+ host, port, err := net.SplitHostPort(addr)
|
|
|
+ if host != "example.com" {
|
|
|
+ return nil, errors.New("cannot resolve host.")
|
|
|
+ }
|
|
|
+ i, err := strconv.Atoi(port)
|
|
|
+ if err != nil {
|
|
|
+ return nil, err
|
|
|
+ }
|
|
|
+ return &net.TCPAddr{IP: net.ParseIP("10.0.10.1"), Port: i, Zone: ""}, nil
|
|
|
+ }
|
|
|
+
|
|
|
+ tests := []struct {
|
|
|
+ a []url.URL
|
|
|
+ b []url.URL
|
|
|
+ expect bool
|
|
|
+ }{
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}},
|
|
|
+ expect: true,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "example.com:2379"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.10.1:2379"}},
|
|
|
+ expect: true,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: true,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "example.com:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "example.com:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: true,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "10.0.10.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "example.com:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: true,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "example.com:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.10.1:2379"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.0.1:2379"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "example.com:2379"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.0.1:2379"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "127.0.0.1:2380"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "example.com:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "127.0.0.1:2380"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "127.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "example.com:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ {
|
|
|
+ a: []url.URL{{Scheme: "http", Host: "10.0.0.1:2379"}},
|
|
|
+ b: []url.URL{{Scheme: "http", Host: "10.0.0.1:2379"}, {Scheme: "http", Host: "127.0.0.1:2380"}},
|
|
|
+ expect: false,
|
|
|
+ },
|
|
|
+ }
|
|
|
+
|
|
|
+ for _, test := range tests {
|
|
|
+ result := URLsEqual(test.a, test.b)
|
|
|
+ if result != test.expect {
|
|
|
+ t.Errorf("a:%v b:%v, expected %v but %v", test.a, test.b, test.expect, result)
|
|
|
+ }
|
|
|
+ }
|
|
|
+}
|
|
|
+func TestURLStringsEqual(t *testing.T) {
|
|
|
+ result := URLStringsEqual([]string{"http://127.0.0.1:8080"}, []string{"http://127.0.0.1:8080"})
|
|
|
+ if !result {
|
|
|
+ t.Errorf("unexpected result %v", result)
|
|
|
+ }
|
|
|
+}
|