mirror of https://github.com/hashicorp/consul
55 lines
960 B
Go
55 lines
960 B
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
|
|
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)
|
|
}
|
|
expect++
|
|
}
|
|
|
|
time.AfterFunc(10*time.Millisecond, func() {
|
|
plan.Stop()
|
|
})
|
|
|
|
err := plan.Run("127.0.0.1:8500")
|
|
if err != nil {
|
|
t.Fatalf("err: %v", err)
|
|
}
|
|
|
|
if expect == 1 {
|
|
t.Fatalf("Bad: %d", expect)
|
|
}
|
|
}
|