watch: convert TestChecksWatch_Service to use channels
This commit is contained in:
parent
4677ca436a
commit
899d11f1c9
|
@ -406,28 +406,30 @@ func TestChecksWatch_Service(t *testing.T) {
|
||||||
a := agent.NewTestAgent(t.Name(), ``)
|
a := agent.NewTestAgent(t.Name(), ``)
|
||||||
defer a.Shutdown()
|
defer a.Shutdown()
|
||||||
|
|
||||||
|
invoke := make(chan error)
|
||||||
plan := mustParse(t, `{"type":"checks", "service":"foobar"}`)
|
plan := mustParse(t, `{"type":"checks", "service":"foobar"}`)
|
||||||
invoke := 0
|
|
||||||
plan.Handler = func(idx uint64, raw interface{}) {
|
plan.Handler = func(idx uint64, raw interface{}) {
|
||||||
if invoke == 0 {
|
|
||||||
if raw == nil {
|
if raw == nil {
|
||||||
return
|
return // ignore
|
||||||
}
|
}
|
||||||
v, ok := raw.([]*consulapi.HealthCheck)
|
v, ok := raw.([]*consulapi.HealthCheck)
|
||||||
if len(v) == 0 {
|
if !ok || len(v) == 0 {
|
||||||
|
return // ignore
|
||||||
|
}
|
||||||
|
if v[0].CheckID != "foobar" {
|
||||||
|
invoke <- errBadContent
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if !ok || v[0].CheckID != "foobar" {
|
invoke <- nil
|
||||||
t.Fatalf("Bad: %#v", raw)
|
|
||||||
}
|
|
||||||
invoke++
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
wg.Add(1)
|
||||||
go func() {
|
go func() {
|
||||||
time.Sleep(20 * time.Millisecond)
|
defer wg.Done()
|
||||||
|
|
||||||
catalog := a.Client().Catalog()
|
catalog := a.Client().Catalog()
|
||||||
|
|
||||||
|
time.Sleep(20 * time.Millisecond)
|
||||||
reg := &consulapi.CatalogRegistration{
|
reg := &consulapi.CatalogRegistration{
|
||||||
Node: "foobar",
|
Node: "foobar",
|
||||||
Address: "1.1.1.1",
|
Address: "1.1.1.1",
|
||||||
|
@ -444,30 +446,25 @@ func TestChecksWatch_Service(t *testing.T) {
|
||||||
ServiceID: "foobar",
|
ServiceID: "foobar",
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
_, err := catalog.Register(reg, nil)
|
if _, err := catalog.Register(reg, nil); err != nil {
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("err: %v", err)
|
t.Fatalf("err: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
time.Sleep(20 * time.Millisecond)
|
|
||||||
plan.Stop()
|
|
||||||
|
|
||||||
dereg := &consulapi.CatalogDeregistration{
|
|
||||||
Node: "foobar",
|
|
||||||
Address: "1.1.1.1",
|
|
||||||
Datacenter: "dc1",
|
|
||||||
}
|
|
||||||
catalog.Deregister(dereg, nil)
|
|
||||||
}()
|
}()
|
||||||
|
|
||||||
err := plan.Run(a.HTTPAddr())
|
wg.Add(1)
|
||||||
if err != nil {
|
go func() {
|
||||||
|
defer wg.Done()
|
||||||
|
if err := plan.Run(a.HTTPAddr()); err != nil {
|
||||||
|
t.Fatalf("err: %v", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
if err := <-invoke; err != nil {
|
||||||
t.Fatalf("err: %v", err)
|
t.Fatalf("err: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if invoke == 0 {
|
plan.Stop()
|
||||||
t.Fatalf("bad: %v", invoke)
|
wg.Wait()
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestEventWatch(t *testing.T) {
|
func TestEventWatch(t *testing.T) {
|
||||||
|
|
Loading…
Reference in New Issue