-
Notifications
You must be signed in to change notification settings - Fork 0
/
helper_test.go
86 lines (79 loc) · 1.67 KB
/
helper_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
80
81
82
83
84
85
86
package main
import (
"crypto/rsa"
"io/ioutil"
"testing"
"github.com/golang-jwt/jwt"
)
func makePrivateKey() (pkey *rsa.PrivateKey, err error) {
private, err := ioutil.ReadFile("test/key/private.pem")
if err != nil {
return
}
pkey, err = jwt.ParseRSAPrivateKeyFromPEM(private)
if err != nil {
return
}
return
}
func makePublicKey() (pkey *rsa.PublicKey, err error) {
public, err := ioutil.ReadFile("test/key/public.pem")
if err != nil {
return
}
pkey, err = jwt.ParseRSAPublicKeyFromPEM(public)
if err != nil {
return
}
return
}
func getJwtToken() (t string, err error) {
privateKey, err := makePrivateKey()
if err != nil {
return
}
type GusherData struct {
UserId string `json:"user_id"`
Channels []string `json:"channels"`
AppKey string `json:"app_key"`
}
type MyCustomClaims struct {
Gusher GusherData `json:"gusher"`
jwt.StandardClaims
}
gd := GusherData{
"test", []string{"AA", "BB"}, "TEST1",
}
claims := MyCustomClaims{
gd, jwt.StandardClaims{},
}
token := jwt.NewWithClaims(jwt.SigningMethodRS256, claims)
t, err = token.SignedString(privateKey)
if err != nil {
return
}
return
}
func TestDecode(t *testing.T) {
token, err := getJwtToken()
if err != nil {
t.Error(err)
}
publicKey, err := makePublicKey()
if err != nil {
t.Error(err)
}
a, err := Decode(publicKey, token)
if err != nil {
t.Error(err)
}
if a.Gusher.UserId != "test" {
t.Errorf("user_id parse error %s", a.Gusher.UserId)
}
if a.Gusher.AppKey != "TEST1" {
t.Errorf("app_key parse error %s", a.Gusher.AppKey)
}
if a.Gusher.Channels[0] != "AA" && a.Gusher.Channels[1] != "BB" {
t.Errorf("channels parse error %v", a.Gusher.Channels)
}
}