78 lines
1.5 KiB
Go
78 lines
1.5 KiB
Go
|
package consul
|
||
|
|
|
||
|
|
import (
|
||
|
|
"os"
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/hashicorp/consul/consul/structs"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Testing for GH-300 and GH-279
|
||
|
|
func TestHealthCheckRace(t *testing.T) {
|
||
|
|
fsm, err := NewFSM(os.Stderr)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
defer fsm.Close()
|
||
|
|
state := fsm.State()
|
||
|
|
|
||
|
|
req := structs.RegisterRequest{
|
||
|
|
Datacenter: "dc1",
|
||
|
|
Node: "foo",
|
||
|
|
Address: "127.0.0.1",
|
||
|
|
Service: &structs.NodeService{
|
||
|
|
ID: "db",
|
||
|
|
Service: "db",
|
||
|
|
},
|
||
|
|
Check: &structs.HealthCheck{
|
||
|
|
Node: "foo",
|
||
|
|
CheckID: "db",
|
||
|
|
Name: "db connectivity",
|
||
|
|
Status: structs.HealthPassing,
|
||
|
|
ServiceID: "db",
|
||
|
|
},
|
||
|
|
}
|
||
|
|
buf, err := structs.Encode(structs.RegisterRequestType, req)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
log := makeLog(buf)
|
||
|
|
log.Index = 10
|
||
|
|
resp := fsm.Apply(log)
|
||
|
|
if resp != nil {
|
||
|
|
t.Fatalf("resp: %v", resp)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Verify the index
|
||
|
|
idx, out1 := state.CheckServiceNodes("db")
|
||
|
|
if idx != 10 {
|
||
|
|
t.Fatalf("Bad index")
|
||
|
|
}
|
||
|
|
|
||
|
|
// Update the check state
|
||
|
|
req.Check.Status = structs.HealthCritical
|
||
|
|
buf, err = structs.Encode(structs.RegisterRequestType, req)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("err: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
log = makeLog(buf)
|
||
|
|
log.Index = 20
|
||
|
|
resp = fsm.Apply(log)
|
||
|
|
if resp != nil {
|
||
|
|
t.Fatalf("resp: %v", resp)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Verify the index changed
|
||
|
|
idx, out2 := state.CheckServiceNodes("db")
|
||
|
|
if idx != 20 {
|
||
|
|
t.Fatalf("Bad index")
|
||
|
|
}
|
||
|
|
|
||
|
|
if reflect.DeepEqual(out1, out2) {
|
||
|
|
t.Fatalf("match: %#v %#v", *out1[0].Checks[0], *out2[0].Checks[0])
|
||
|
|
}
|
||
|
|
}
|