补充测试用例

internal/app/service/token/token_service.go

@@ -1,6 +1,7 @@
 package token
 
 import (
+	"fmt"
 	"time"
 
 	"github.com/golang-jwt/jwt/v5"
@@ -52,9 +53,16 @@ func (s *tokenService) ParseToken(tokenString string) (*Claims, error) {
 		return s.secret, nil
 	})
 
+	// 优先检查解析过程中是否发生错误
+	if err != nil {
+		return nil, err
+	}
+
+	// 只有当 token 对象有效时，才尝试获取 Claims 并验证
 	if claims, ok := token.Claims.(*Claims); ok && token.Valid {
 		return claims, nil
 	}
 
-	return nil, err
+	// 如果 token 无效（例如，过期但没有返回错误，或者 Claims 类型不匹配），则返回一个通用错误
+	return nil, fmt.Errorf("token is invalid")
 }

internal/app/service/token/token_service_test.go

@@ -1,6 +1,7 @@
 package token_test
 
 import (
+	"errors"
 	"testing"
 	"time"
 
@@ -45,36 +46,38 @@ func TestGenerateToken(t *testing.T) {
 }
 
 func TestParseToken(t *testing.T) {
-	// 使用一个测试密钥初始化 TokenService
-	testSecret := []byte("test_secret_key")
-	service := token.NewTokenService(testSecret)
+	// 使用两个不同的测试密钥
+	correctSecret := []byte("the_correct_secret")
+	wrongSecret := []byte("a_very_wrong_secret")
+
+	serviceWithCorrectKey := token.NewTokenService(correctSecret)
+	serviceWithWrongKey := token.NewTokenService(wrongSecret)
 
 	userID := uint(456)
 
-	// 生成一个有效的 token 用于解析测试
-	validToken, err := service.GenerateToken(userID)
+	// 1. 生成一个有效的 token
+	validToken, err := serviceWithCorrectKey.GenerateToken(userID)
 	if err != nil {
 		t.Fatalf("为解析测试生成有效令牌失败: %v", err)
 	}
 
-	// 测试用例 1: 有效 token
-	claims, err := service.ParseToken(validToken)
+	// 测试用例 1: 使用正确的密钥成功解析
+	claims, err := serviceWithCorrectKey.ParseToken(validToken)
 	if err != nil {
-		t.Errorf("解析有效令牌失败: %v", err)
+		t.Errorf("使用正确密钥解析有效令牌失败: %v", err)
 	}
-
 	if claims.UserID != userID {
 		t.Errorf("解析有效令牌时期望用户ID %d, 实际为 %d", userID, claims.UserID)
 	}
 
-	// 测试用例 2: 无效 token (例如, 错误的签名)
-	invalidToken := "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VyX2lkIjoxMjMsImV4cCI6MTY3ODkwNTYwMCwiaXNzIjoicGlnLWZhcm0tY29udHJvbGxlciJ9.invalid_signature_here"
-	_, err = service.ParseToken(invalidToken)
+	// 测试用例 2: 无效 token (例如, 格式错误的字符串)
+	invalidTokenString := "this.is.not.a.valid.jwt"
+	_, err = serviceWithCorrectKey.ParseToken(invalidTokenString)
 	if err == nil {
-		t.Error("解析无效令牌意外成功")
+		t.Error("解析格式错误的令牌意外成功")
 	}
 
-	// 测试用例 3: 过期 token (创建一个过期时间在过去的 token)
+	// 测试用C:\Users\divano\Desktop\work\AA-Pig\pig-farm-controller\internal\infra\repository\plan_repository_test.go例 3: 过期 token
 	expiredClaims := token.Claims{
 		UserID: userID,
 		RegisteredClaims: jwt.RegisteredClaims{
@@ -83,13 +86,22 @@ func TestParseToken(t *testing.T) {
 		},
 	}
 	expiredTokenClaims := jwt.NewWithClaims(jwt.SigningMethodHS256, expiredClaims)
-	expiredTokenString, err := expiredTokenClaims.SignedString(testSecret)
+	expiredTokenString, err := expiredTokenClaims.SignedString(correctSecret)
 	if err != nil {
 		t.Fatalf("生成过期令牌失败: %v", err)
 	}
-
-	_, err = service.ParseToken(expiredTokenString)
+	_, err = serviceWithCorrectKey.ParseToken(expiredTokenString)
 	if err == nil {
 		t.Error("解析过期令牌意外成功")
 	}
+
+	// 新增测试用例 4: 使用错误的密钥解析
+	_, err = serviceWithWrongKey.ParseToken(validToken)
+	if err == nil {
+		t.Error("使用错误密钥解析令牌意外成功")
+	}
+	// 我们可以更精确地检查错误类型，以确保它是签名错误
+	if !errors.Is(err, jwt.ErrTokenSignatureInvalid) {
+		t.Errorf("期望得到签名无效错误 (ErrTokenSignatureInvalid)，但得到了: %v", err)
+	}
 }