95 lines
2.7 KiB
Go
95 lines
2.7 KiB
Go
package token
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/golang-jwt/jwt/v5"
|
|
)
|
|
|
|
func TestGenerateToken(t *testing.T) {
|
|
// 使用一个测试密钥初始化 TokenService
|
|
testSecret := []byte("test_secret_key")
|
|
service := NewTokenService(testSecret)
|
|
|
|
userID := uint(123)
|
|
tokenString, err := service.GenerateToken(userID)
|
|
|
|
if err != nil {
|
|
t.Fatalf("GenerateToken failed: %v", err)
|
|
}
|
|
|
|
if tokenString == "" {
|
|
t.Fatal("Generated token string is empty")
|
|
}
|
|
|
|
// 解析 token 以确保其有效性及声明
|
|
claims, err := service.ParseToken(tokenString)
|
|
if err != nil {
|
|
t.Fatalf("ParseToken failed after generation: %v", err)
|
|
}
|
|
|
|
if claims.UserID != userID {
|
|
t.Errorf("Expected UserID %d, got %d", userID, claims.UserID)
|
|
}
|
|
|
|
// 检查 token 是否未过期 (在合理范围内)
|
|
if claims.ExpiresAt == nil || claims.ExpiresAt.Time.Before(time.Now().Add(-time.Minute)) {
|
|
t.Errorf("Token expiration time is invalid or already expired")
|
|
}
|
|
|
|
if claims.Issuer != "pig-farm-controller" {
|
|
t.Errorf("Expected Issuer \"pig-farm-controller\", got \"%s\"", claims.Issuer)
|
|
}
|
|
}
|
|
|
|
func TestParseToken(t *testing.T) {
|
|
// 使用一个测试密钥初始化 TokenService
|
|
testSecret := []byte("test_secret_key")
|
|
service := NewTokenService(testSecret)
|
|
|
|
userID := uint(456)
|
|
|
|
// 生成一个有效的 token 用于解析测试
|
|
validToken, err := service.GenerateToken(userID)
|
|
if err != nil {
|
|
t.Fatalf("Failed to generate valid token for parsing test: %v", err)
|
|
}
|
|
|
|
// 测试用例 1: 有效 token
|
|
claims, err := service.ParseToken(validToken)
|
|
if err != nil {
|
|
t.Errorf("ParseToken failed for valid token: %v", err)
|
|
}
|
|
|
|
if claims.UserID != userID {
|
|
t.Errorf("Expected UserID %d, got %d for valid token", userID, claims.UserID)
|
|
}
|
|
|
|
// 测试用例 2: 无效 token (例如, 错误的签名)
|
|
invalidToken := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoxMjMsImV4cCI6MTY3ODkwNTYwMCwiaXNzIjoicGlnLWZhcm0tY29udHJvbGxlciJ9.invalid_signature_here"
|
|
_, err = service.ParseToken(invalidToken)
|
|
if err == nil {
|
|
t.Error("ParseToken unexpectedly succeeded for invalid token")
|
|
}
|
|
|
|
// 测试用例 3: 过期 token (创建一个过期时间在过去的 token)
|
|
expiredClaims := Claims{
|
|
UserID: userID,
|
|
RegisteredClaims: jwt.RegisteredClaims{
|
|
ExpiresAt: jwt.NewNumericDate(time.Now().Add(-time.Hour)), // 1 小时前
|
|
Issuer: "pig-farm-controller",
|
|
},
|
|
}
|
|
expiredTokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, expiredClaims)
|
|
expiredTokenString, err := expiredTokenClaims.SignedString(testSecret)
|
|
if err != nil {
|
|
t.Fatalf("Failed to generate expired token: %v", err)
|
|
}
|
|
|
|
_, err = service.ParseToken(expiredTokenString)
|
|
if err == nil {
|
|
t.Error("ParseToken unexpectedly succeeded for expired token")
|
|
}
|
|
}
|