chore: add tests for internal/models (#740)

Author: leafty

internal/models/id_generators_test.go

@@ -0,0 +1,31 @@
+package models
+
+import (
+	"encoding/base64"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestULIDGenerator(t *testing.T) {
+	generator := ULIDGenerator{}
+
+	ulid, err := generator.ID()
+
+	require.NoError(t, err)
+	assert.NotEmpty(t, ulid)
+}
+
+func TestRandomGenerator(t *testing.T) {
+	generator := NewRandomGenerator(32)
+
+	id, err := generator.ID()
+
+	require.NoError(t, err)
+	assert.NotEmpty(t, id)
+
+	decoded, err := base64.RawURLEncoding.DecodeString(id)
+	require.NoError(t, err)
+	assert.Len(t, decoded, 32)
+}

internal/models/session_test.go

@@ -55,6 +55,17 @@ func TestSessionTouchNoExpiry(t *testing.T) {
 	assert.True(t, session.ExpiresAt.IsZero())
 }
 
+func TestSessionTouchNoIdleTTL(t *testing.T) {
+	session := Session{
+		CreatedAt:     time.Now(),
+		MaxTTLSeconds: 300,
+	}
+	session.Touch()
+	assert.False(t, session.Expired())
+	assert.True(t, session.ExpiresAt.After(session.CreatedAt))
+	assert.True(t, session.ExpiresAt.Equal(session.CreatedAt.Add(session.MaxTTL())))
+}
+
 func TestSessionLoginState(t *testing.T) {
 	session := Session{
 		CreatedAt: time.Now(),

internal/models/tokens_test.go renamed to internal/models/token_test.go

@@ -74,3 +74,64 @@ func TestNoEncryptor(t *testing.T) {
 	assert.Equal(t, token, encToken)
 	assert.Equal(t, token, decToken)
 }
+
+func TestTokenString(t *testing.T) {
+	token := AuthToken{
+		ID:         "123456",
+		Value:      "secretValue",
+		ProviderID: "providerId",
+		SessionID:  "sessionId",
+		Type:       AccessTokenType,
+	}
+	tokenString := token.String()
+	assert.Contains(t, tokenString, "ID: 123456")
+	assert.Contains(t, tokenString, "Value: redacted")
+	assert.NotContains(t, tokenString, "secretValue")
+	assert.Contains(t, tokenString, "ProviderID: providerId")
+	assert.Contains(t, tokenString, "SessionID: sessionId")
+	assert.Contains(t, tokenString, AccessTokenType)
+}
+
+func TestTokenNotExpired(t *testing.T) {
+	token := AuthToken{
+		ExpiresAt: time.Now().Add(time.Duration(5) * time.Minute),
+	}
+	assert.False(t, token.Expired())
+}
+
+func TestTokenExpired(t *testing.T) {
+	token := AuthToken{
+		ExpiresAt: time.Now().Add(-time.Duration(5) * time.Minute),
+	}
+	assert.True(t, token.Expired())
+}
+
+func TestTokenNoExpiry(t *testing.T) {
+	token := AuthToken{
+		ExpiresAt: time.Time{},
+	}
+	assert.True(t, token.ExpiresAt.IsZero())
+	assert.False(t, token.Expired())
+}
+
+func TestTokenNotExpiresSoon(t *testing.T) {
+	token := AuthToken{
+		ExpiresAt: time.Now().Add(time.Duration(5) * time.Minute),
+	}
+	assert.False(t, token.ExpiresSoon(time.Duration(3)*time.Minute))
+}
+
+func TestTokenExpiresSoon(t *testing.T) {
+	token := AuthToken{
+		ExpiresAt: time.Now().Add(time.Duration(2) * time.Minute),
+	}
+	assert.True(t, token.ExpiresSoon(time.Duration(3)*time.Minute))
+}
+
+func TestTokenExpiresSoonNoExpiry(t *testing.T) {
+	token := AuthToken{
+		ExpiresAt: time.Time{},
+	}
+	assert.True(t, token.ExpiresAt.IsZero())
+	assert.False(t, token.ExpiresSoon(time.Duration(3)*time.Minute))
+}