snapshot of https://github.com/cortezaproject/corteza.git (2024.9.x, 2026-09-18; blobs > 131072B omitted)

server/store/tests/auth_oa2tokens.go

@@ -0,0 +1,88 @@
+package tests
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/cortezaproject/corteza/server/pkg/id"
+	"github.com/cortezaproject/corteza/server/pkg/rand"
+	"github.com/cortezaproject/corteza/server/store"
+	"github.com/cortezaproject/corteza/server/system/types"
+	"github.com/stretchr/testify/require"
+)
+
+func testAuthOa2tokens(t *testing.T, s store.AuthOa2tokens) {
+	var (
+		ctx = context.Background()
+		req = require.New(t)
+
+		makeNew = func() *types.AuthOa2token {
+			// minimum data set for new authOa2token
+			return &types.AuthOa2token{
+				ID:        id.Next(),
+				CreatedAt: time.Now(),
+				ExpiresAt: time.Now(),
+				Access:    string(rand.Bytes(5)),
+				Code:      string(rand.Bytes(5)),
+				Refresh:   string(rand.Bytes(5)),
+			}
+		}
+
+		truncAndCreate = func(t *testing.T) (*require.Assertions, *types.AuthOa2token) {
+			req := require.New(t)
+			req.NoError(s.TruncateAuthOa2tokens(ctx))
+			res := makeNew()
+			req.NoError(s.CreateAuthOa2token(ctx, res))
+			return req, res
+		}
+	)
+
+	t.Run("create", func(t *testing.T) {
+		authOa2token := makeNew()
+		req.NoError(s.CreateAuthOa2token(ctx, authOa2token))
+	})
+
+	t.Run("lookup by", func(t *testing.T) {
+		t.Run("access", func(t *testing.T) {
+			req, authOa2token := truncAndCreate(t)
+			fetched, err := s.LookupAuthOa2tokenByAccess(ctx, authOa2token.Access)
+			req.NoError(err)
+			req.NotNil(fetched)
+		})
+
+		t.Run("code", func(t *testing.T) {
+			req, authOa2token := truncAndCreate(t)
+			fetched, err := s.LookupAuthOa2tokenByCode(ctx, authOa2token.Code)
+			req.NoError(err)
+			req.NotNil(fetched)
+		})
+
+		t.Run("refresh", func(t *testing.T) {
+			req, authOa2token := truncAndCreate(t)
+			fetched, err := s.LookupAuthOa2tokenByRefresh(ctx, authOa2token.Refresh)
+			req.NoError(err)
+			req.NotNil(fetched)
+		})
+	})
+
+	t.Run("delete", func(t *testing.T) {
+		t.Run("access", func(t *testing.T) {
+			req, authOa2token := truncAndCreate(t)
+			err := s.DeleteAuthOA2TokenByRefresh(ctx, authOa2token.Access)
+			req.NoError(err)
+		})
+
+		t.Run("code", func(t *testing.T) {
+			req, authOa2token := truncAndCreate(t)
+			err := s.DeleteAuthOA2TokenByRefresh(ctx, authOa2token.Code)
+			req.NoError(err)
+		})
+
+		t.Run("refresh", func(t *testing.T) {
+			req, authOa2token := truncAndCreate(t)
+			err := s.DeleteAuthOA2TokenByRefresh(ctx, authOa2token.Refresh)
+			req.NoError(err)
+		})
+	})
+}