From 82105b91c0670483be6fa4178319cb7fb557ffb8 Mon Sep 17 00:00:00 2001
From: Jim Lambert <jlambert@hashicorp.com>
Date: Fri, 4 Sep 2020 14:20:17 -0400
Subject: [PATCH] add TestState with unit tests

---
 internal/session/testing.go      | 13 ++++++++++++-
 internal/session/testing_test.go | 14 ++++++++++++++
 2 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/internal/session/testing.go b/internal/session/testing.go
index be1cf6ec2e..106df7181d 100644
--- a/internal/session/testing.go
+++ b/internal/session/testing.go
@@ -19,6 +19,17 @@ import (
 	"github.com/stretchr/testify/require"
 )
 
+func TestState(t *testing.T, conn *gorm.DB, sessionId string, state Status) *State {
+	t.Helper()
+	require := require.New(t)
+	rw := db.New(conn)
+	s, err := NewState(sessionId, state)
+	require.NoError(err)
+	err = rw.Create(context.Background(), s)
+	require.NoError(err)
+	return s
+}
+
 func TestSession(
 	t *testing.T,
 	conn *gorm.DB,
@@ -148,7 +159,7 @@ func TestSessionParams(t *testing.T, conn *gorm.DB, wrapper wrapping.Wrapper, ia
 		tcpTarget.PublicId,
 		sets[0].PublicId,
 		at.PublicId,
-		user.ScopeId, // for now, we pick the user scope as the session scope
+		tcpTarget.ScopeId,
 		"127.0.0.1",
 		"22"
 }
diff --git a/internal/session/testing_test.go b/internal/session/testing_test.go
index 85dcf8e3d6..9f9391ddfb 100644
--- a/internal/session/testing_test.go
+++ b/internal/session/testing_test.go
@@ -19,3 +19,17 @@ func Test_TestSession(t *testing.T) {
 	require.NotNil(s)
 	assert.NotEmpty(s.PublicId)
 }
+
+func Test_TestState(t *testing.T) {
+	t.Helper()
+	assert, require := assert.New(t), require.New(t)
+	conn, _ := db.TestSetup(t, "postgres")
+	wrapper := db.TestWrapper(t)
+	iamRepo := iam.TestRepo(t, conn, wrapper)
+	s := TestDefaultSession(t, conn, wrapper, iamRepo)
+	require.NotNil(s)
+	assert.NotEmpty(s.PublicId)
+
+	state := TestState(t, conn, s.PublicId, Pending)
+	require.NotNil(state)
+}