Skip to content

Commit

Permalink
Add etcdsd getservers tests
Browse files Browse the repository at this point in the history
  • Loading branch information
gabrielerzinger authored and felipejfc committed Aug 15, 2018
1 parent 0ffa04b commit 62fdd53
Showing 1 changed file with 32 additions and 0 deletions.
32 changes: 32 additions & 0 deletions cluster/etcd_service_discovery_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -42,6 +42,22 @@ var etcdSDTables = []struct {
{NewServer("backend-2", "type3", false, nil)},
}

var etcdSDTablesMultipleServers = []struct {
servers []*Server
}{
{[]*Server{}},
{[]*Server{
NewServer("frontend-1", "type1", true, map[string]string{"k1": "v1"}),
NewServer("backend-1", "type2", false, map[string]string{"k2": "v2"}),
NewServer("backend-2", "type3", false, nil),
}},
{[]*Server{
NewServer("frontend-1", "type2", true, map[string]string{"k1": "v1"}),
NewServer("frontend-2", "type2", true, map[string]string{"k2": "v2"}),
NewServer("frontend-3", "type2", true, map[string]string{"k1": "v1"}),
}},
}

func getConfig(conf ...*viper.Viper) *config.Config {
config := config.NewConfig(conf...)
return config
Expand Down Expand Up @@ -199,6 +215,22 @@ func TestEtcdSDGetServer(t *testing.T) {
}
}

func TestEtcdSDGetServers(t *testing.T) {
t.Parallel()
for _, table := range etcdSDTablesMultipleServers {
config := getConfig()
c, cli := helpers.GetTestEtcd(t)
defer c.Terminate(t)
e := getEtcdSD(t, config, &Server{}, cli)
e.bootstrapLease()
for _, server := range table.servers {
e.bootstrapServer(server)
}
serverList := e.GetServers()
assert.ElementsMatch(t, table.servers, serverList)
}
}

func TestEtcdSDInit(t *testing.T) {
t.Parallel()
for _, table := range etcdSDTables {
Expand Down

0 comments on commit 62fdd53

Please sign in to comment.