394 lines
10 KiB
Go
394 lines
10 KiB
Go
|
package jwtx
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/golang-jwt/jwt/v4"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestGenerateJwtToken(t *testing.T) {
|
|||
|
|
// 测试数据
|
|||
|
|
testClaims := JwtClaims{
|
|||
|
|
UserId: 1,
|
|||
|
|
AgentId: 0,
|
|||
|
|
Platform: "wxh5",
|
|||
|
|
UserType: 0,
|
|||
|
|
IsAgent: 0,
|
|||
|
|
}
|
|||
|
|
testSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA"
|
|||
|
|
testExpire := int64(2592000) // 1小时
|
|||
|
|
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
claims JwtClaims
|
|||
|
|
secret string
|
|||
|
|
expire int64
|
|||
|
|
wantErr bool
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "正常生成token",
|
|||
|
|
claims: testClaims,
|
|||
|
|
secret: testSecret,
|
|||
|
|
expire: testExpire,
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "不同用户数据",
|
|||
|
|
claims: JwtClaims{
|
|||
|
|
UserId: 99999,
|
|||
|
|
AgentId: 11111,
|
|||
|
|
Platform: "mobile",
|
|||
|
|
UserType: 0,
|
|||
|
|
IsAgent: 1,
|
|||
|
|
},
|
|||
|
|
secret: testSecret,
|
|||
|
|
expire: testExpire,
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "空密钥",
|
|||
|
|
claims: testClaims,
|
|||
|
|
secret: "",
|
|||
|
|
expire: testExpire,
|
|||
|
|
wantErr: false, // 空密钥不会导致生成失败，但验证时会失败
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "零过期时间",
|
|||
|
|
claims: testClaims,
|
|||
|
|
secret: testSecret,
|
|||
|
|
expire: 0,
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "负数过期时间",
|
|||
|
|
claims: testClaims,
|
|||
|
|
secret: testSecret,
|
|||
|
|
expire: -3600,
|
|||
|
|
wantErr: false,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
token, err := GenerateJwtToken(tt.claims, tt.secret, tt.expire)
|
|||
|
|
|
|||
|
|
if (err != nil) != tt.wantErr {
|
|||
|
|
t.Errorf("GenerateJwtToken() error = %v, wantErr %v", err, tt.wantErr)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if !tt.wantErr {
|
|||
|
|
// 验证token不为空
|
|||
|
|
if token == "" {
|
|||
|
|
t.Error("GenerateJwtToken() 返回的token为空")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证token格式（JWT token应该包含两个点分隔符）
|
|||
|
|
parts := strings.Split(token, ".")
|
|||
|
|
if len(parts) != 3 {
|
|||
|
|
t.Errorf("GenerateJwtToken() 返回的token格式不正确，期望3部分，实际%d部分", len(parts))
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证token可以被解析（不验证签名，只验证格式）
|
|||
|
|
parsedToken, err := jwt.Parse(token, func(token *jwt.Token) (interface{}, error) {
|
|||
|
|
return []byte(tt.secret), nil
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if err == nil && parsedToken != nil {
|
|||
|
|
// 验证claims是否正确设置
|
|||
|
|
if claims, ok := parsedToken.Claims.(jwt.MapClaims); ok {
|
|||
|
|
// 验证userId
|
|||
|
|
if userId, exists := claims["userId"]; exists {
|
|||
|
|
if int64(userId.(float64)) != tt.claims.UserId {
|
|||
|
|
t.Errorf("token中的userId不匹配，期望%d，实际%v", tt.claims.UserId, userId)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
t.Error("token中缺少userId字段")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证extra字段存在
|
|||
|
|
if _, exists := claims[ExtraKey]; !exists {
|
|||
|
|
t.Error("token中缺少extra字段")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证exp字段
|
|||
|
|
if exp, exists := claims["exp"]; exists {
|
|||
|
|
expTime := int64(exp.(float64))
|
|||
|
|
now := time.Now().Unix()
|
|||
|
|
expectedExp := now + tt.expire
|
|||
|
|
// 允许5秒的时间差异
|
|||
|
|
if expTime < expectedExp-5 || expTime > expectedExp+5 {
|
|||
|
|
t.Errorf("token过期时间不正确，期望约%d，实际%d", expectedExp, expTime)
|
|||
|
|
}
|
|||
|
|
} else {
|
|||
|
|
t.Error("token中缺少exp字段")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证iat字段
|
|||
|
|
if _, exists := claims["iat"]; !exists {
|
|||
|
|
t.Error("token中缺少iat字段")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("生成的token: %s", token)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestGenerateJwtTokenAndParse(t *testing.T) {
|
|||
|
|
// 测试生成token后能够正确解析
|
|||
|
|
testClaims := JwtClaims{
|
|||
|
|
UserId: 12345,
|
|||
|
|
AgentId: 67890,
|
|||
|
|
Platform: "web",
|
|||
|
|
UserType: 1,
|
|||
|
|
IsAgent: 0,
|
|||
|
|
}
|
|||
|
|
testSecret := "test-secret-key"
|
|||
|
|
testExpire := int64(3600)
|
|||
|
|
|
|||
|
|
// 生成token
|
|||
|
|
token, err := GenerateJwtToken(testClaims, testSecret, testExpire)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GenerateJwtToken() failed: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 解析token
|
|||
|
|
parsedClaims, err := ParseJwtToken(token, testSecret)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("ParseJwtToken() failed: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证解析出的claims与原始claims一致
|
|||
|
|
if parsedClaims.UserId != testClaims.UserId {
|
|||
|
|
t.Errorf("UserId不匹配，期望%d，实际%d", testClaims.UserId, parsedClaims.UserId)
|
|||
|
|
}
|
|||
|
|
if parsedClaims.AgentId != testClaims.AgentId {
|
|||
|
|
t.Errorf("AgentId不匹配，期望%d，实际%d", testClaims.AgentId, parsedClaims.AgentId)
|
|||
|
|
}
|
|||
|
|
if parsedClaims.Platform != testClaims.Platform {
|
|||
|
|
t.Errorf("Platform不匹配，期望%s，实际%s", testClaims.Platform, parsedClaims.Platform)
|
|||
|
|
}
|
|||
|
|
if parsedClaims.UserType != testClaims.UserType {
|
|||
|
|
t.Errorf("UserType不匹配，期望%d，实际%d", testClaims.UserType, parsedClaims.UserType)
|
|||
|
|
}
|
|||
|
|
if parsedClaims.IsAgent != testClaims.IsAgent {
|
|||
|
|
t.Errorf("IsAgent不匹配，期望%d，实际%d", testClaims.IsAgent, parsedClaims.IsAgent)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("测试通过: 生成token并成功解析，claims数据一致")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func BenchmarkGenerateJwtToken(t *testing.B) {
|
|||
|
|
// 性能测试
|
|||
|
|
testClaims := JwtClaims{
|
|||
|
|
UserId: 12345,
|
|||
|
|
AgentId: 67890,
|
|||
|
|
Platform: "web",
|
|||
|
|
UserType: 1,
|
|||
|
|
IsAgent: 0,
|
|||
|
|
}
|
|||
|
|
testSecret := "test-secret-key"
|
|||
|
|
testExpire := int64(3600)
|
|||
|
|
|
|||
|
|
t.ResetTimer()
|
|||
|
|
for i := 0; i < t.N; i++ {
|
|||
|
|
_, err := GenerateJwtToken(testClaims, testSecret, testExpire)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GenerateJwtToken() failed: %v", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestParseJwtToken(t *testing.T) {
|
|||
|
|
// 使用你修改的测试数据
|
|||
|
|
testClaims := JwtClaims{
|
|||
|
|
UserId: 6,
|
|||
|
|
AgentId: 0,
|
|||
|
|
Platform: "wxh5",
|
|||
|
|
UserType: 0,
|
|||
|
|
IsAgent: 0,
|
|||
|
|
}
|
|||
|
|
testSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA"
|
|||
|
|
testExpire := int64(2592000) // 30天
|
|||
|
|
|
|||
|
|
// 先生成一个token用于测试
|
|||
|
|
token, err := GenerateJwtToken(testClaims, testSecret, testExpire)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("生成token失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("生成的测试token: %s", token)
|
|||
|
|
|
|||
|
|
tests := []struct {
|
|||
|
|
name string
|
|||
|
|
token string
|
|||
|
|
secret string
|
|||
|
|
wantErr bool
|
|||
|
|
wantClaims *JwtClaims
|
|||
|
|
}{
|
|||
|
|
{
|
|||
|
|
name: "正常解析token",
|
|||
|
|
token: token,
|
|||
|
|
secret: testSecret,
|
|||
|
|
wantErr: false,
|
|||
|
|
wantClaims: &testClaims,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "错误的密钥",
|
|||
|
|
token: token,
|
|||
|
|
secret: "wrong-secret",
|
|||
|
|
wantErr: true,
|
|||
|
|
wantClaims: nil,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "空token",
|
|||
|
|
token: "",
|
|||
|
|
secret: testSecret,
|
|||
|
|
wantErr: true,
|
|||
|
|
wantClaims: nil,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "无效token格式",
|
|||
|
|
token: "invalid.token.format",
|
|||
|
|
secret: testSecret,
|
|||
|
|
wantErr: true,
|
|||
|
|
wantClaims: nil,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "缺少点分隔符的token",
|
|||
|
|
token: "invalidtoken",
|
|||
|
|
secret: testSecret,
|
|||
|
|
wantErr: true,
|
|||
|
|
wantClaims: nil,
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
name: "自定义token",
|
|||
|
|
token: "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NTI5MDA5MTQsImV4dHJhIjp7ImFnZW50SWQiOjAsImlzQWdlbnQiOjAsInBsYXRmb3JtIjoid3hoNSIsInVzZXJJZCI6NiwidXNlclR5cGUiOjF9LCJpYXQiOjE3NTAzMDg5MTQsInVzZXJJZCI6Nn0.GPKgLOaALOIa1ft7Hipuo4YKFf5guYt0rz2MCDCSdCQ",
|
|||
|
|
secret: testSecret,
|
|||
|
|
wantErr: false,
|
|||
|
|
wantClaims: &testClaims,
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for _, tt := range tests {
|
|||
|
|
t.Run(tt.name, func(t *testing.T) {
|
|||
|
|
claims, err := ParseJwtToken(tt.token, tt.secret)
|
|||
|
|
|
|||
|
|
if (err != nil) != tt.wantErr {
|
|||
|
|
t.Errorf("ParseJwtToken() error = %v, wantErr %v", err, tt.wantErr)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if !tt.wantErr && tt.wantClaims != nil {
|
|||
|
|
if claims == nil {
|
|||
|
|
t.Error("ParseJwtToken() 返回的claims为nil")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 验证各个字段
|
|||
|
|
if claims.UserId != tt.wantClaims.UserId {
|
|||
|
|
t.Errorf("UserId不匹配，期望%d，实际%d", tt.wantClaims.UserId, claims.UserId)
|
|||
|
|
}
|
|||
|
|
if claims.AgentId != tt.wantClaims.AgentId {
|
|||
|
|
t.Errorf("AgentId不匹配，期望%d，实际%d", tt.wantClaims.AgentId, claims.AgentId)
|
|||
|
|
}
|
|||
|
|
if claims.Platform != tt.wantClaims.Platform {
|
|||
|
|
t.Errorf("Platform不匹配，期望%s，实际%s", tt.wantClaims.Platform, claims.Platform)
|
|||
|
|
}
|
|||
|
|
if claims.UserType != tt.wantClaims.UserType {
|
|||
|
|
t.Errorf("UserType不匹配，期望%d，实际%d", tt.wantClaims.UserType, claims.UserType)
|
|||
|
|
}
|
|||
|
|
if claims.IsAgent != tt.wantClaims.IsAgent {
|
|||
|
|
t.Errorf("IsAgent不匹配，期望%d，实际%d", tt.wantClaims.IsAgent, claims.IsAgent)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("解析成功的claims: UserId=%d, AgentId=%d, Platform=%s, UserType=%d, IsAgent=%d",
|
|||
|
|
claims.UserId, claims.AgentId, claims.Platform, claims.UserType, claims.IsAgent)
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestParseCustomJwtToken 测试解析自定义token - 你可以在这里传入你自己的token
|
|||
|
|
func TestParseCustomJwtToken(t *testing.T) {
|
|||
|
|
// 在这里修改你想要测试的token和secret
|
|||
|
|
customToken := "" // 在这里粘贴你的token
|
|||
|
|
customSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" // 你的密钥
|
|||
|
|
|
|||
|
|
// 如果没有提供自定义token，跳过测试
|
|||
|
|
if customToken == "" {
|
|||
|
|
t.Skip("跳过自定义token测试，请在代码中设置customToken值")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("解析自定义token: %s", customToken)
|
|||
|
|
|
|||
|
|
claims, err := ParseJwtToken(customToken, customSecret)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("解析自定义token失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("解析结果:")
|
|||
|
|
t.Logf(" UserId: %d", claims.UserId)
|
|||
|
|
t.Logf(" AgentId: %d", claims.AgentId)
|
|||
|
|
t.Logf(" Platform: %s", claims.Platform)
|
|||
|
|
t.Logf(" UserType: %d", claims.UserType)
|
|||
|
|
t.Logf(" IsAgent: %d", claims.IsAgent)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestGenerateAndParseWithRealData 生成一个真实的token并解析
|
|||
|
|
func TestGenerateAndParseWithRealData(t *testing.T) {
|
|||
|
|
// 使用真实数据生成token
|
|||
|
|
realClaims := JwtClaims{
|
|||
|
|
UserId: 1,
|
|||
|
|
AgentId: 0,
|
|||
|
|
Platform: "wxh5",
|
|||
|
|
UserType: 0,
|
|||
|
|
IsAgent: 0,
|
|||
|
|
}
|
|||
|
|
realSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA"
|
|||
|
|
realExpire := int64(2592000) // 30天
|
|||
|
|
|
|||
|
|
// 生成token
|
|||
|
|
token, err := GenerateJwtToken(realClaims, realSecret, realExpire)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("生成token失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("=== 生成的完整token ===")
|
|||
|
|
t.Logf("Token: %s", token)
|
|||
|
|
t.Logf("========================")
|
|||
|
|
|
|||
|
|
// 解析token
|
|||
|
|
parsedClaims, err := ParseJwtToken(token, realSecret)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("解析token失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Logf("=== 解析结果 ===")
|
|||
|
|
t.Logf("UserId: %d", parsedClaims.UserId)
|
|||
|
|
t.Logf("AgentId: %d", parsedClaims.AgentId)
|
|||
|
|
t.Logf("Platform: %s", parsedClaims.Platform)
|
|||
|
|
t.Logf("UserType: %d", parsedClaims.UserType)
|
|||
|
|
t.Logf("IsAgent: %d", parsedClaims.IsAgent)
|
|||
|
|
t.Logf("================")
|
|||
|
|
|
|||
|
|
// 验证数据一致性
|
|||
|
|
if parsedClaims.UserId != realClaims.UserId ||
|
|||
|
|
parsedClaims.AgentId != realClaims.AgentId ||
|
|||
|
|
parsedClaims.Platform != realClaims.Platform ||
|
|||
|
|
parsedClaims.UserType != realClaims.UserType ||
|
|||
|
|
parsedClaims.IsAgent != realClaims.IsAgent {
|
|||
|
|
t.Error("解析出的claims与原始数据不一致")
|
|||
|
|
} else {
|
|||
|
|
t.Log("✅ 数据一致性验证通过")
|
|||
|
|
}
|
|||
|
|
}
|