// Copyright 2016 The etcd Authors // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License. package integration import ( "context" "reflect" "testing" "github.com/coreos/etcd/integration" "github.com/coreos/etcd/pkg/testutil" "github.com/coreos/etcd/pkg/types" ) func TestMemberList(t *testing.T) { defer testutil.AfterTest(t) clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 3}) defer clus.Terminate(t) capi := clus.RandClient() resp, err := capi.MemberList(context.Background()) if err != nil { t.Fatalf("failed to list member %v", err) } if len(resp.Members) != 3 { t.Errorf("number of members = %d, want %d", len(resp.Members), 3) } } func TestMemberAdd(t *testing.T) { defer testutil.AfterTest(t) clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 3}) defer clus.Terminate(t) capi := clus.RandClient() urls := []string{"http://127.0.0.1:1234"} resp, err := capi.MemberAdd(context.Background(), urls) if err != nil { t.Fatalf("failed to add member %v", err) } if !reflect.DeepEqual(resp.Member.PeerURLs, urls) { t.Errorf("urls = %v, want %v", urls, resp.Member.PeerURLs) } } func TestMemberRemove(t *testing.T) { defer testutil.AfterTest(t) clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 3}) defer clus.Terminate(t) capi := clus.Client(1) resp, err := capi.MemberList(context.Background()) if err != nil { t.Fatalf("failed to list member %v", err) } rmvID := resp.Members[0].ID // indexes in capi member list don't necessarily match cluster member list; // find member that is not the client to remove for _, m := range resp.Members { mURLs, _ := types.NewURLs(m.PeerURLs) if !reflect.DeepEqual(mURLs, clus.Members[1].ServerConfig.PeerURLs) { rmvID = m.ID break } } _, err = capi.MemberRemove(context.Background(), rmvID) if err != nil { t.Fatalf("failed to remove member %v", err) } resp, err = capi.MemberList(context.Background()) if err != nil { t.Fatalf("failed to list member %v", err) } if len(resp.Members) != 2 { t.Errorf("number of members = %d, want %d", len(resp.Members), 2) } } func TestMemberUpdate(t *testing.T) { defer testutil.AfterTest(t) clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 3}) defer clus.Terminate(t) capi := clus.RandClient() resp, err := capi.MemberList(context.Background()) if err != nil { t.Fatalf("failed to list member %v", err) } urls := []string{"http://127.0.0.1:1234"} _, err = capi.MemberUpdate(context.Background(), resp.Members[0].ID, urls) if err != nil { t.Fatalf("failed to update member %v", err) } resp, err = capi.MemberList(context.Background()) if err != nil { t.Fatalf("failed to list member %v", err) } if !reflect.DeepEqual(resp.Members[0].PeerURLs, urls) { t.Errorf("urls = %v, want %v", urls, resp.Members[0].PeerURLs) } } func TestMemberAddUpdateWrongURLs(t *testing.T) { defer testutil.AfterTest(t) clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 1}) defer clus.Terminate(t) capi := clus.RandClient() tt := [][]string{ // missing protocol scheme {"://127.0.0.1:2379"}, // unsupported scheme {"mailto://127.0.0.1:2379"}, // not conform to host:port {"http://127.0.0.1"}, // contain a path {"http://127.0.0.1:2379/path"}, // first path segment in URL cannot contain colon {"127.0.0.1:1234"}, // URL scheme must be http, https, unix, or unixs {"localhost:1234"}, } for i := range tt { _, err := capi.MemberAdd(context.Background(), tt[i]) if err == nil { t.Errorf("#%d: MemberAdd err = nil, but error", i) } _, err = capi.MemberUpdate(context.Background(), 0, tt[i]) if err == nil { t.Errorf("#%d: MemberUpdate err = nil, but error", i) } } }