Integrates session TTL tests with new state store.

consul/session_ttl_test.go

@@ -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",