Nikolai Rodionov
33f48f2bfb
All checks were successful
ci/woodpecker/push/build Pipeline was successful
Signed-off-by: Nikolai Rodionov <allanger@badhouseplants.net>
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
package controllers_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"gitea.badhouseplants.net/softplayer/softplayer-backend/internal/controllers"
|
|
"github.com/google/uuid"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var (
|
|
testAccessTTL = time.Second * 5
|
|
testRefreshTTL = time.Second * 20
|
|
testUserID = uuid.New().String()
|
|
)
|
|
|
|
func TestGenerateInvalidTokenType(t *testing.T) {
|
|
authCtrl := controllers.NewAuthController([]byte("test"), testAccessTTL, testRefreshTTL, nil)
|
|
token, _, err := authCtrl.GenerateToken(testUserID, "invalid_type")
|
|
assert.Equal(t, "", token)
|
|
assert.ErrorIs(t, controllers.ErrUnknownTokenType, err)
|
|
}
|
|
|
|
func TestGenerateValidateAccessToken(t *testing.T) {
|
|
authCtrl := controllers.NewAuthController([]byte("test"), testAccessTTL, testRefreshTTL, nil)
|
|
now := time.Now()
|
|
token, _, err := authCtrl.GenerateToken(testUserID, controllers.TokenTypeAccess)
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, token)
|
|
|
|
claims, err := authCtrl.ParseToken(token)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, testUserID, claims.UserID)
|
|
assert.NotEmpty(t, claims.TokenID)
|
|
assert.Equal(t, controllers.TokenTypeAccess, claims.TokenType)
|
|
assert.Equal(t, now.Add(testAccessTTL).Unix(), claims.ExpiresAt.Unix())
|
|
assert.Equal(t, now.Unix(), claims.IssuedAt.Unix())
|
|
assert.Equal(t, now.Unix(), claims.NotBefore.Unix())
|
|
}
|
|
|
|
func TestGenerateValidateRefreshToken(t *testing.T) {
|
|
authCtrl := controllers.NewAuthController([]byte("test"), testAccessTTL, testRefreshTTL, nil)
|
|
now := time.Now()
|
|
token, _, err := authCtrl.GenerateToken(testUserID, controllers.TokenTypeRefresh)
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, token)
|
|
|
|
claims, err := authCtrl.ParseToken(token)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, testUserID, claims.UserID)
|
|
assert.NotEmpty(t, claims.TokenID)
|
|
assert.Equal(t, controllers.TokenTypeRefresh, claims.TokenType)
|
|
assert.Equal(t, now.Add(testRefreshTTL).Unix(), claims.ExpiresAt.Unix())
|
|
assert.Equal(t, now.Unix(), claims.IssuedAt.Unix())
|
|
assert.Equal(t, now.Unix(), claims.NotBefore.Unix())
|
|
}
|