102 lines
2.6 KiB
Go
102 lines
2.6 KiB
Go
|
package auth
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestTOTPManager_GenerateAndValidate(t *testing.T) {
|
|||
|
|
m := NewTOTPManager()
|
|||
|
|
|
|||
|
|
// 生成密钥
|
|||
|
|
setup, err := m.GenerateSecret("testuser@example.com")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GenerateSecret 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if setup.Secret == "" {
|
|||
|
|
t.Fatal("生成的 Secret 不应为空")
|
|||
|
|
}
|
|||
|
|
if setup.QRCodeBase64 == "" {
|
|||
|
|
t.Fatal("QRCode Base64 不应为空")
|
|||
|
|
}
|
|||
|
|
if len(setup.RecoveryCodes) != RecoveryCodeCount {
|
|||
|
|
t.Fatalf("恢复码数量期望 %d，实际 %d", RecoveryCodeCount, len(setup.RecoveryCodes))
|
|||
|
|
}
|
|||
|
|
t.Logf("生成 Secret: %s", setup.Secret)
|
|||
|
|
t.Logf("恢复码示例: %s", setup.RecoveryCodes[0])
|
|||
|
|
|
|||
|
|
// 用生成的密钥生成当前 TOTP 码，再验证
|
|||
|
|
code, err := m.GenerateCurrentCode(setup.Secret)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GenerateCurrentCode 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
if !m.ValidateCode(setup.Secret, code) {
|
|||
|
|
t.Fatalf("有效 TOTP 码应该通过验证，code=%s", code)
|
|||
|
|
}
|
|||
|
|
t.Logf("TOTP 验证通过，code=%s", code)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestTOTPManager_InvalidCode(t *testing.T) {
|
|||
|
|
m := NewTOTPManager()
|
|||
|
|
setup, err := m.GenerateSecret("user")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GenerateSecret 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 错误的验证码
|
|||
|
|
if m.ValidateCode(setup.Secret, "000000") {
|
|||
|
|
// 偶尔可能恰好正确，跳过而不是 fatal
|
|||
|
|
t.Skip("000000 碰巧是有效码，跳过测试")
|
|||
|
|
}
|
|||
|
|
t.Log("无效验证码正确拒绝")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestTOTPManager_RecoveryCodeFormat(t *testing.T) {
|
|||
|
|
m := NewTOTPManager()
|
|||
|
|
setup, err := m.GenerateSecret("user2")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("GenerateSecret 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for i, code := range setup.RecoveryCodes {
|
|||
|
|
parts := strings.Split(code, "-")
|
|||
|
|
if len(parts) != 2 {
|
|||
|
|
t.Errorf("恢复码 [%d] 格式错误（期望 XXXXX-XXXXX）: %s", i, code)
|
|||
|
|
}
|
|||
|
|
if len(parts[0]) != 5 || len(parts[1]) != 5 {
|
|||
|
|
t.Errorf("恢复码 [%d] 各部分长度应为 5: %s", i, code)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestValidateRecoveryCode(t *testing.T) {
|
|||
|
|
codes := []string{"ABCDE-FGHIJ", "KLMNO-PQRST", "UVWXY-ZABCD"}
|
|||
|
|
|
|||
|
|
// 正确匹配
|
|||
|
|
idx, ok := ValidateRecoveryCode("ABCDE-FGHIJ", codes)
|
|||
|
|
if !ok || idx != 0 {
|
|||
|
|
t.Fatalf("有效恢复码应该匹配，idx=%d ok=%v", idx, ok)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 大小写不敏感
|
|||
|
|
idx2, ok2 := ValidateRecoveryCode("klmno-pqrst", codes)
|
|||
|
|
if !ok2 || idx2 != 1 {
|
|||
|
|
t.Fatalf("大小写不敏感匹配失败，idx=%d ok=%v", idx2, ok2)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 去除空格
|
|||
|
|
idx3, ok3 := ValidateRecoveryCode(" UVWXY-ZABCD ", codes)
|
|||
|
|
if !ok3 || idx3 != 2 {
|
|||
|
|
t.Fatalf("去除空格匹配失败，idx=%d ok=%v", idx3, ok3)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 不匹配
|
|||
|
|
_, ok4 := ValidateRecoveryCode("XXXXX-YYYYY", codes)
|
|||
|
|
if ok4 {
|
|||
|
|
t.Fatal("无效恢复码不应该匹配")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
t.Log("恢复码验证全部通过")
|
|||
|
|
}
|