-
Notifications
You must be signed in to change notification settings - Fork 0
/
jwt_maker_test.go
79 lines (61 loc) · 1.77 KB
/
jwt_maker_test.go
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package token
import (
"github.com/google/uuid"
"testing"
"time"
"github.com/dgrijalva/jwt-go"
"github.com/stretchr/testify/require"
)
func TestJWTMaker(t *testing.T) {
key = "sdfsfdffddsdjfisajdfasjdfamdksss"
maker, err := NewJWTMaker(key)
require.NoError(t, err)
id := uuid.New()
duration := time.Minute
issuedAt := time.Now()
expiredAt := issuedAt.Add(duration)
payload := &Payload{
ID: id,
IssuedAt: issuedAt,
ExpiredAt: expiredAt,
}
token, err := maker.CreateToken(payload)
require.NoError(t, err)
require.NotEmpty(t, token)
payload, err = maker.VerifyToken(token)
require.NoError(t, err)
require.NotEmpty(t, token)
require.NotZero(t, payload.ID)
require.Equal(t, id, payload.ID)
require.WithinDuration(t, issuedAt, payload.IssuedAt, time.Second)
require.WithinDuration(t, expiredAt, payload.ExpiredAt, time.Second)
}
func TestExpiredJWTToken(t *testing.T) {
maker, err := NewJWTMaker(key)
require.NoError(t, err)
payload := &Payload{
ID: uuid.New(),
IssuedAt: time.Now(),
ExpiredAt: time.Now().Add(-time.Minute),
}
token, err := maker.CreateToken(payload)
require.NoError(t, err)
require.NotEmpty(t, token)
payload, err = maker.VerifyToken(token)
require.Error(t, err)
require.EqualError(t, err, ErrExpiredToken.Error())
require.Nil(t, payload)
}
func TestInvalidJWTTokenAlgNone(t *testing.T) {
payload, err := NewPayload(time.Minute)
require.NoError(t, err)
jwtToken := jwt.NewWithClaims(jwt.SigningMethodNone, payload)
token, err := jwtToken.SignedString(jwt.UnsafeAllowNoneSignatureType)
require.NoError(t, err)
maker, err := NewJWTMaker(key)
require.NoError(t, err)
payload, err = maker.VerifyToken(token)
require.Error(t, err)
require.EqualError(t, err, ErrInvalidToken.Error())
require.Nil(t, payload)
}