|
@@ -15,6 +15,7 @@
|
|
|
package client
|
|
package client
|
|
|
|
|
|
|
|
import (
|
|
import (
|
|
|
|
|
+ "encoding/json"
|
|
|
"net/http"
|
|
"net/http"
|
|
|
"net/url"
|
|
"net/url"
|
|
|
"reflect"
|
|
"reflect"
|
|
@@ -109,3 +110,156 @@ func TestV2MembersURL(t *testing.T) {
|
|
|
t.Fatalf("v2MembersURL got %#v, want %#v", got, want)
|
|
t.Fatalf("v2MembersURL got %#v, want %#v", got, want)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
+
|
|
|
|
|
+func TestMemberUnmarshal(t *testing.T) {
|
|
|
|
|
+ tests := []struct {
|
|
|
|
|
+ body []byte
|
|
|
|
|
+ wantMember Member
|
|
|
|
|
+ wantError bool
|
|
|
|
|
+ }{
|
|
|
|
|
+ // no URLs, just check ID & Name
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"id": "c", "name": "dungarees"}`),
|
|
|
|
|
+ wantMember: Member{ID: "c", Name: "dungarees", PeerURLs: nil, ClientURLs: nil},
|
|
|
|
|
+ },
|
|
|
|
|
+
|
|
|
|
|
+ // both client and peer URLs
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"peerURLs": ["http://127.0.0.1:4001"], "clientURLs": ["http://127.0.0.1:4001"]}`),
|
|
|
|
|
+ wantMember: Member{
|
|
|
|
|
+ PeerURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:4001",
|
|
|
|
|
+ },
|
|
|
|
|
+ ClientURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:4001",
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+
|
|
|
|
|
+ // multiple peer URLs
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"peerURLs": ["http://127.0.0.1:4001", "https://example.com"]}`),
|
|
|
|
|
+ wantMember: Member{
|
|
|
|
|
+ PeerURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:4001",
|
|
|
|
|
+ "https://example.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ ClientURLs: nil,
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+
|
|
|
|
|
+ // multiple client URLs
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"clientURLs": ["http://127.0.0.1:4001", "https://example.com"]}`),
|
|
|
|
|
+ wantMember: Member{
|
|
|
|
|
+ PeerURLs: nil,
|
|
|
|
|
+ ClientURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:4001",
|
|
|
|
|
+ "https://example.com",
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+
|
|
|
|
|
+ // invalid JSON
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"peerU`),
|
|
|
|
|
+ wantError: true,
|
|
|
|
|
+ },
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ for i, tt := range tests {
|
|
|
|
|
+ got := Member{}
|
|
|
|
|
+ err := json.Unmarshal(tt.body, &got)
|
|
|
|
|
+ if tt.wantError != (err != nil) {
|
|
|
|
|
+ t.Errorf("#%d: want error %t, got %v", i, tt.wantError, err)
|
|
|
|
|
+ continue
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ if !reflect.DeepEqual(tt.wantMember, got) {
|
|
|
|
|
+ t.Errorf("#%d: incorrect output: want=%#v, got=%#v", i, tt.wantMember, got)
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
|
|
+func TestMemberCollectionUnmarshal(t *testing.T) {
|
|
|
|
|
+ tests := []struct {
|
|
|
|
|
+ body []byte
|
|
|
|
|
+ want memberCollection
|
|
|
|
|
+ }{
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"members":[]}`),
|
|
|
|
|
+ want: memberCollection([]Member{}),
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ body: []byte(`{"members":[{"id":"2745e2525fce8fe","peerURLs":["http://127.0.0.1:7003"],"name":"node3","clientURLs":["http://127.0.0.1:4003"]},{"id":"42134f434382925","peerURLs":["http://127.0.0.1:2380","http://127.0.0.1:7001"],"name":"node1","clientURLs":["http://127.0.0.1:2379","http://127.0.0.1:4001"]},{"id":"94088180e21eb87b","peerURLs":["http://127.0.0.1:7002"],"name":"node2","clientURLs":["http://127.0.0.1:4002"]}]}`),
|
|
|
|
|
+ want: memberCollection(
|
|
|
|
|
+ []Member{
|
|
|
|
|
+ {
|
|
|
|
|
+ ID: "2745e2525fce8fe",
|
|
|
|
|
+ Name: "node3",
|
|
|
|
|
+ PeerURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:7003",
|
|
|
|
|
+ },
|
|
|
|
|
+ ClientURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:4003",
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ ID: "42134f434382925",
|
|
|
|
|
+ Name: "node1",
|
|
|
|
|
+ PeerURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:2380",
|
|
|
|
|
+ "http://127.0.0.1:7001",
|
|
|
|
|
+ },
|
|
|
|
|
+ ClientURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:2379",
|
|
|
|
|
+ "http://127.0.0.1:4001",
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+ {
|
|
|
|
|
+ ID: "94088180e21eb87b",
|
|
|
|
|
+ Name: "node2",
|
|
|
|
|
+ PeerURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:7002",
|
|
|
|
|
+ },
|
|
|
|
|
+ ClientURLs: []string{
|
|
|
|
|
+ "http://127.0.0.1:4002",
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+ },
|
|
|
|
|
+ ),
|
|
|
|
|
+ },
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ for i, tt := range tests {
|
|
|
|
|
+ var got memberCollection
|
|
|
|
|
+ err := json.Unmarshal(tt.body, &got)
|
|
|
|
|
+ if err != nil {
|
|
|
|
|
+ t.Errorf("#%d: unexpected error: %v", i, err)
|
|
|
|
|
+ continue
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ if !reflect.DeepEqual(tt.want, got) {
|
|
|
|
|
+ t.Errorf("#%d: incorrect output: want=%#v, got=%#v", i, tt.want, got)
|
|
|
|
|
+ }
|
|
|
|
|
+ }
|
|
|
|
|
+}
|
|
|
|
|
+
|
|
|
|
|
+func TestMemberCreateRequestMarshal(t *testing.T) {
|
|
|
|
|
+ req := memberCreateRequest{
|
|
|
|
|
+ PeerURLs: types.URLs([]url.URL{
|
|
|
|
|
+ url.URL{Scheme: "http", Host: "127.0.0.1:8081"},
|
|
|
|
|
+ url.URL{Scheme: "https", Host: "127.0.0.1:8080"},
|
|
|
|
|
+ }),
|
|
|
|
|
+ }
|
|
|
|
|
+ want := []byte(`{"peerURLs":["http://127.0.0.1:8081","https://127.0.0.1:8080"]}`)
|
|
|
|
|
+
|
|
|
|
|
+ got, err := json.Marshal(&req)
|
|
|
|
|
+ if err != nil {
|
|
|
|
|
+ t.Fatalf("Marshal returned unexpected err=%v", err)
|
|
|
|
|
+ }
|
|
|
|
|
+
|
|
|
|
|
+ if !reflect.DeepEqual(want, got) {
|
|
|
|
|
+ t.Fatalf("Failed to marshal memberCreateRequest: want=%s, got=%s", want, got)
|
|
|
|
|
+ }
|
|
|
|
|
+}
|