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()
}
}
