87 lines
1.9 KiB
Go
87 lines
1.9 KiB
Go
package auth
|
||
|
||
import (
|
||
"carrotskin/pkg/config"
|
||
"testing"
|
||
)
|
||
|
||
// TestGetJWTService_NotInitialized 测试未初始化时获取JWT服务
|
||
func TestGetJWTService_NotInitialized(t *testing.T) {
|
||
_, err := GetJWTService()
|
||
if err == nil {
|
||
t.Error("未初始化时应该返回错误")
|
||
}
|
||
|
||
expectedError := "JWT服务未初始化，请先调用 auth.Init()"
|
||
if err.Error() != expectedError {
|
||
t.Errorf("错误消息 = %q, want %q", err.Error(), expectedError)
|
||
}
|
||
}
|
||
|
||
// TestMustGetJWTService_Panic 测试MustGetJWTService在未初始化时panic
|
||
func TestMustGetJWTService_Panic(t *testing.T) {
|
||
defer func() {
|
||
if r := recover(); r == nil {
|
||
t.Error("MustGetJWTService 应该在未初始化时panic")
|
||
}
|
||
}()
|
||
|
||
_ = MustGetJWTService()
|
||
}
|
||
|
||
// TestInit_JWTService 测试JWT服务初始化
|
||
func TestInit_JWTService(t *testing.T) {
|
||
cfg := config.JWTConfig{
|
||
Secret: "test-secret-key",
|
||
ExpireHours: 24,
|
||
}
|
||
|
||
err := Init(cfg)
|
||
if err != nil {
|
||
t.Errorf("Init() 错误 = %v, want nil", err)
|
||
}
|
||
|
||
// 验证可以获取服务
|
||
service, err := GetJWTService()
|
||
if err != nil {
|
||
t.Errorf("GetJWTService() 错误 = %v, want nil", err)
|
||
}
|
||
if service == nil {
|
||
t.Error("GetJWTService() 返回的服务不应为nil")
|
||
}
|
||
}
|
||
|
||
// TestInit_JWTService_Once 测试Init只执行一次
|
||
func TestInit_JWTService_Once(t *testing.T) {
|
||
cfg := config.JWTConfig{
|
||
Secret: "test-secret-key-1",
|
||
ExpireHours: 24,
|
||
}
|
||
|
||
// 第一次初始化
|
||
err1 := Init(cfg)
|
||
if err1 != nil {
|
||
t.Fatalf("第一次Init() 错误 = %v", err1)
|
||
}
|
||
|
||
service1, _ := GetJWTService()
|
||
|
||
// 第二次初始化（应该不会改变服务）
|
||
cfg2 := config.JWTConfig{
|
||
Secret: "test-secret-key-2",
|
||
ExpireHours: 48,
|
||
}
|
||
err2 := Init(cfg2)
|
||
if err2 != nil {
|
||
t.Fatalf("第二次Init() 错误 = %v", err2)
|
||
}
|
||
|
||
service2, _ := GetJWTService()
|
||
|
||
// 验证是同一个实例（sync.Once保证）
|
||
if service1 != service2 {
|
||
t.Error("Init应该只执行一次，返回同一个实例")
|
||
}
|
||
}
|
||
|