onasty/internal/jwtutil/jwtutil_test.go(view raw)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
package jwtutil
import (
"testing"
"testing/synctest"
"time"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestJWTUtil_AccessToken(t *testing.T) {
jwt := NewJWTUtil("key", time.Hour)
payload := Payload{UserID: "user.123"}
token, err := jwt.AccessToken(payload)
require.NoError(t, err)
assert.NotEmpty(t, token)
}
func TestJWTUtil_RefreshToken(t *testing.T) {
jwt := NewJWTUtil("key", time.Hour)
tok, err := jwt.RefreshToken()
require.NoError(t, err)
assert.Len(t, tok, 64)
secondTok, err := jwt.RefreshToken()
require.NoError(t, err)
// tokens should be unique
assert.NotEqual(t, tok, secondTok)
}
func TestJWTUtil_Parse(t *testing.T) {
jwt := NewJWTUtil("key", time.Hour)
payload := Payload{UserID: "qwerty"}
token, err := jwt.AccessToken(payload)
require.NoError(t, err)
assert.NotEmpty(t, token)
parsedPayload, err := jwt.Parse(token)
require.NoError(t, err)
assert.Equal(t, payload, parsedPayload)
}
func TestJWTUtil_Parse_expired(t *testing.T) {
ttl := 24 * time.Hour
synctest.Test(t, func(t *testing.T) {
jwt := NewJWTUtil("key", ttl)
payload := Payload{UserID: "qwerty"}
token, err := jwt.AccessToken(payload)
require.NoError(t, err)
assert.NotEmpty(t, token)
time.Sleep(2 * ttl)
_, err = jwt.Parse(token)
require.EqualError(t, err, ErrTokenExpired.Error())
})
}
|