func TestUnregisterAll(t *testing.T) { client, cleanup := testutil.SetupDiscoverd(t) defer cleanup() serviceName := "unregisterAllTest" assert(client.Register(serviceName, ":1111"), t) assert(client.Register(serviceName, ":2222"), t) assert(client.Register(serviceName, ":3333"), t) services, err := client.Services(serviceName, 1) assert(err, t) if len(services) != 3 { t.Fatal("Wrong number of services") } assert(client.UnregisterAll(), t) set, err := client.NewServiceSet(serviceName) assert(err, t) if len(set.Services()) != 0 { t.Fatal("There should be no services") } assert(set.Close(), t) }
func setup(t *testing.T) (*discoverd.Client, func()) { killEtcd := runEtcdServer(t) client, killDiscoverd := bootDiscoverd(t, "") return client, func() { client.UnregisterAll() client.Close() killDiscoverd() killEtcd() } }
func SetupDiscoverdWithEtcd(t TestingT) (*discoverd.Client, string, func()) { etcdAddr, killEtcd := RunEtcdServer(t) client, killDiscoverd := BootDiscoverd(t, "", etcdAddr) return client, etcdAddr, func() { client.UnregisterAll() client.Close() killDiscoverd() killEtcd() } }
func SetupDiscoverd(t TestingT) (*discoverd.Client, func()) { killEtcd := RunEtcdServer(t) client, killDiscoverd := BootDiscoverd(t, "") return client, func() { client.UnregisterAll() client.Close() killDiscoverd() killEtcd() } }