mirror of https://github.com/hashicorp/consul
74 lines
1.2 KiB
Go
74 lines
1.2 KiB
Go
package watch
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func init() {
|
|
watchFuncFactory["noop"] = noopWatch
|
|
}
|
|
|
|
func noopWatch(params map[string]interface{}) (WatchFunc, error) {
|
|
fn := func(p *WatchPlan) (uint64, interface{}, error) {
|
|
idx := p.lastIndex + 1
|
|
return idx, idx, nil
|
|
}
|
|
return fn, nil
|
|
}
|
|
|
|
func mustParse(t *testing.T, q string) *WatchPlan {
|
|
params := makeParams(t, q)
|
|
plan, err := Parse(params)
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
return plan
|
|
}
|
|
|
|
func TestRun_Stop(t *testing.T) {
|
|
plan := mustParse(t, `{"type":"noop"}`)
|
|
|
|
var expect uint64 = 1
|
|
doneCh := make(chan struct{})
|
|
plan.Handler = func(idx uint64, val interface{}) {
|
|
if idx != expect {
|
|
t.Fatalf("Bad: %d %d", expect, idx)
|
|
}
|
|
if val != expect {
|
|
t.Fatalf("Bad: %d %d", expect, val)
|
|
}
|
|
if expect == 1 {
|
|
close(doneCh)
|
|
}
|
|
expect++
|
|
}
|
|
|
|
errCh := make(chan error, 1)
|
|
go func() {
|
|
errCh <- plan.Run("127.0.0.1:8500")
|
|
}()
|
|
|
|
select {
|
|
case <-doneCh:
|
|
plan.Stop()
|
|
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatalf("handler never ran")
|
|
}
|
|
|
|
select {
|
|
case err := <-errCh:
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
case <-time.After(1 * time.Second):
|
|
t.Fatalf("watcher didn't exit")
|
|
}
|
|
|
|
if expect == 1 {
|
|
t.Fatalf("Bad: %d", expect)
|
|
}
|
|
}
|