core/identity: add data store for thread-safe storage of sessions and users

internal/identity/manager/datastore_test.go

@@ -0,0 +1,78 @@
+package manager
+
+import (
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+	"github.com/stretchr/testify/assert"
+	"google.golang.org/protobuf/testing/protocmp"
+
+	"github.com/pomerium/pomerium/pkg/grpc/session"
+	"github.com/pomerium/pomerium/pkg/grpc/user"
+)
+
+func TestDataStore(t *testing.T) {
+	t.Parallel()
+
+	ds := newDataStore()
+	s, u := ds.GetSessionAndUser("S1")
+	assert.Nil(t, s, "should return a nil session when none exists")
+	assert.Nil(t, u, "should return a nil user when none exists")
+
+	u, ss := ds.GetUserAndSessions("U1")
+	assert.Nil(t, u, "should return a nil user when none exists")
+	assert.Empty(t, ss, "should return an empty list of sessions when no user exists")
+
+	s = &session.Session{Id: "S1", UserId: "U1"}
+	ds.PutSession(s)
+	s1, u1 := ds.GetSessionAndUser("S1")
+	assert.NotNil(t, s1, "should return a non-nil session")
+	assert.False(t, s == s1, "should return different pointers")
+	assert.Empty(t, cmp.Diff(s, s1, protocmp.Transform()), "should be the same as was entered")
+	assert.Nil(t, u1, "should return a nil user when only the session exists")
+
+	ds.PutUser(&user.User{
+		Id: "U1",
+	})
+	_, u1 = ds.GetSessionAndUser("S1")
+	assert.NotNil(t, u1, "should return a user now that it has been added")
+
+	ds.PutSession(&session.Session{Id: "S4", UserId: "U1"})
+	ds.PutSession(&session.Session{Id: "S3", UserId: "U1"})
+	ds.PutSession(&session.Session{Id: "S2", UserId: "U1"})
+	u, ss = ds.GetUserAndSessions("U1")
+	assert.NotNil(t, u)
+	assert.Empty(t, cmp.Diff(ss, []*session.Session{
+		{Id: "S1", UserId: "U1"},
+		{Id: "S2", UserId: "U1"},
+		{Id: "S3", UserId: "U1"},
+		{Id: "S4", UserId: "U1"},
+	}, protocmp.Transform()), "should return all sessions in id order")
+
+	ds.DeleteSession("S4")
+
+	u, ss = ds.GetUserAndSessions("U1")
+	assert.NotNil(t, u)
+	assert.Empty(t, cmp.Diff(ss, []*session.Session{
+		{Id: "S1", UserId: "U1"},
+		{Id: "S2", UserId: "U1"},
+		{Id: "S3", UserId: "U1"},
+	}, protocmp.Transform()), "should return all sessions in id order")
+
+	ds.DeleteUser("U1")
+	u, ss = ds.GetUserAndSessions("U1")
+	assert.Nil(t, u)
+	assert.Empty(t, cmp.Diff(ss, []*session.Session{
+		{Id: "S1", UserId: "U1"},
+		{Id: "S2", UserId: "U1"},
+		{Id: "S3", UserId: "U1"},
+	}, protocmp.Transform()), "should still return all sessions in id order")
+
+	ds.DeleteSession("S1")
+	ds.DeleteSession("S2")
+	ds.DeleteSession("S3")
+
+	u, ss = ds.GetUserAndSessions("U1")
+	assert.Nil(t, u)
+	assert.Empty(t, ss)
+}