long-agent/user-system
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add missing PCE parameter to GenerateTokenPair calls in test files

Browse Source 
The JWT GenerateTokenPair functions were updated to require a PCE (Password
Changed Epoch) parameter for token invalidation. This commit updates test files
in concurrent and performance packages to include this parameter.

- internal/concurrent/concurrent_test.go: 2 call sites fixed
- internal/performance/benchmark_test.go: 3 call sites fixed
- internal/performance/performance_test.go: 4 call sites fixed
This commit is contained in:
long-agent
2026-04-18 20:16:45 +08:00
parent 61c19e54ac
commit a754545072
3 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/performance/benchmark_test.go
View File

@@ -68,13 +68,13 @@ func BenchmarkJWTGenerateToken(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
_, _, _ = jwtManager.GenerateTokenPair(int64(i), "testuser")
_, _, _ = jwtManager.GenerateTokenPair(int64(i), "testuser", 0)
}
}
func BenchmarkJWTValidateToken(b *testing.B) {
jwtManager := auth.NewJWT("benchmark-secret-key-32bytes!", 2*time.Hour, 7*24*time.Hour)
token, _, _ := jwtManager.GenerateTokenPair(1, "testuser")
token, _, _ := jwtManager.GenerateTokenPair(1, "testuser", 0)
b.ResetTimer()
for i := 0; i < b.N; i++ {
@@ -87,7 +87,7 @@ func BenchmarkJWTGenerateAndValidate(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
token, _, _ := jwtManager.GenerateTokenPair(int64(i), "testuser")
token, _, _ := jwtManager.GenerateTokenPair(int64(i), "testuser", 0)
jwtManager.ValidateAccessToken(token)
}
}

8
internal/performance/performance_test.go
View File

@@ -148,7 +148,7 @@ func BenchmarkTokenGeneration(b *testing.B) {
for i := 0; i < b.N; i++ {
start := time.Now()
_, _, err := jwtManager.GenerateTokenPair(1, "benchuser")
_, _, err := jwtManager.GenerateTokenPair(1, "benchuser", 0)
latency := time.Since(start).Nanoseconds()
metrics.RecordLatency(latency)
if err == nil {
@@ -165,7 +165,7 @@ func BenchmarkTokenGeneration(b *testing.B) {
// BenchmarkTokenValidation JWT验证性能测试
func BenchmarkTokenValidation(b *testing.B) {
jwtManager := auth.NewJWT("benchmark-secret", 2*time.Hour, 7*24*time.Hour)
accessToken, _, err := jwtManager.GenerateTokenPair(1, "benchuser")
accessToken, _, err := jwtManager.GenerateTokenPair(1, "benchuser", 0)
if err != nil {
b.Fatalf("生成Token失败: %v", err)
}
@@ -201,7 +201,7 @@ func TestP99LatencyThreshold(t *testing.T) {
operation: func() time.Duration {
jwtManager := auth.NewJWT("test-secret", 2*time.Hour, 7*24*time.Hour)
start := time.Now()
jwtManager.GenerateTokenPair(1, "testuser")
jwtManager.GenerateTokenPair(1, "testuser", 0)
return time.Since(start)
},
thresholdMs: 100,
@@ -322,7 +322,7 @@ func TestMemoryUsage(t *testing.T) {
jwtManager := auth.NewJWT("test-secret", 2*time.Hour, 7*24*time.Hour)
for i := 0; i < 10000; i++ {
accessToken, _, _ := jwtManager.GenerateTokenPair(int64(i%100), "testuser")
accessToken, _, _ := jwtManager.GenerateTokenPair(int64(i%100), "testuser", 0)
jwtManager.ValidateAccessToken(accessToken)
}