Integrates session TTL tests with new state store.

This commit is contained in:
James Phillips 2015-10-12 20:56:17 -07:00
parent a15c24f771
commit 450886246d
1 changed files with 16 additions and 8 deletions

View File

@ -19,8 +19,10 @@ func TestInitializeSessionTimers(t *testing.T) {
testutil.WaitForLeader(t, s1.RPC, "dc1")
state := s1.fsm.State()
state.EnsureNode(1, structs.Node{Node: "foo", Address: "127.0.0.1"})
state := s1.fsm.StateNew()
if err := state.EnsureNode(1, &structs.Node{Node: "foo", Address: "127.0.0.1"}); err != nil {
t.Fatalf("err: %s", err)
}
session := &structs.Session{
ID: generateUUID(),
Node: "foo",
@ -57,8 +59,10 @@ func TestResetSessionTimer_Fault(t *testing.T) {
}
// Create a session
state := s1.fsm.State()
state.EnsureNode(1, structs.Node{Node: "foo", Address: "127.0.0.1"})
state := s1.fsm.StateNew()
if err := state.EnsureNode(1, &structs.Node{Node: "foo", Address: "127.0.0.1"}); err != nil {
t.Fatalf("err: %s", err)
}
session := &structs.Session{
ID: generateUUID(),
Node: "foo",
@ -89,8 +93,10 @@ func TestResetSessionTimer_NoTTL(t *testing.T) {
testutil.WaitForLeader(t, s1.RPC, "dc1")
// Create a session
state := s1.fsm.State()
state.EnsureNode(1, structs.Node{Node: "foo", Address: "127.0.0.1"})
state := s1.fsm.StateNew()
if err := state.EnsureNode(1, &structs.Node{Node: "foo", Address: "127.0.0.1"}); err != nil {
t.Fatalf("err: %s", err)
}
session := &structs.Session{
ID: generateUUID(),
Node: "foo",
@ -200,8 +206,10 @@ func TestInvalidateSession(t *testing.T) {
testutil.WaitForLeader(t, s1.RPC, "dc1")
// Create a session
state := s1.fsm.State()
state.EnsureNode(1, structs.Node{Node: "foo", Address: "127.0.0.1"})
state := s1.fsm.StateNew()
if err := state.EnsureNode(1, &structs.Node{Node: "foo", Address: "127.0.0.1"}); err != nil {
t.Fatalf("err: %s", err)
}
session := &structs.Session{
ID: generateUUID(),
Node: "foo",