120 lines
2.8 KiB
Go
120 lines
2.8 KiB
Go
package service
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
// TestGenerateVerificationCode 测试生成验证码函数
|
||
func TestGenerateVerificationCode(t *testing.T) {
|
||
tests := []struct {
|
||
name string
|
||
wantLen int
|
||
wantErr bool
|
||
}{
|
||
{
|
||
name: "生成6位验证码",
|
||
wantLen: CodeLength,
|
||
wantErr: false,
|
||
},
|
||
}
|
||
|
||
for _, tt := range tests {
|
||
t.Run(tt.name, func(t *testing.T) {
|
||
code, err := GenerateVerificationCode()
|
||
if (err != nil) != tt.wantErr {
|
||
t.Errorf("GenerateVerificationCode() error = %v, wantErr %v", err, tt.wantErr)
|
||
return
|
||
}
|
||
if !tt.wantErr && len(code) != tt.wantLen {
|
||
t.Errorf("GenerateVerificationCode() code length = %v, want %v", len(code), tt.wantLen)
|
||
}
|
||
// 验证验证码只包含数字
|
||
for _, c := range code {
|
||
if c < '0' || c > '9' {
|
||
t.Errorf("GenerateVerificationCode() code contains non-digit: %c", c)
|
||
}
|
||
}
|
||
})
|
||
}
|
||
|
||
// 测试多次生成，验证码应该不同（概率上）
|
||
codes := make(map[string]bool)
|
||
for i := 0; i < 100; i++ {
|
||
code, err := GenerateVerificationCode()
|
||
if err != nil {
|
||
t.Fatalf("GenerateVerificationCode() failed: %v", err)
|
||
}
|
||
if codes[code] {
|
||
t.Logf("发现重复验证码（这是正常的，因为只有6位数字）: %s", code)
|
||
}
|
||
codes[code] = true
|
||
}
|
||
}
|
||
|
||
// TestVerificationConstants 测试验证码相关常量
|
||
func TestVerificationConstants(t *testing.T) {
|
||
if CodeLength != 6 {
|
||
t.Errorf("CodeLength = %d, want 6", CodeLength)
|
||
}
|
||
|
||
if CodeExpiration != 10*time.Minute {
|
||
t.Errorf("CodeExpiration = %v, want 10 minutes", CodeExpiration)
|
||
}
|
||
|
||
if CodeRateLimit != 1*time.Minute {
|
||
t.Errorf("CodeRateLimit = %v, want 1 minute", CodeRateLimit)
|
||
}
|
||
|
||
// 验证验证码类型常量
|
||
types := []string{
|
||
VerificationTypeRegister,
|
||
VerificationTypeResetPassword,
|
||
VerificationTypeChangeEmail,
|
||
}
|
||
|
||
for _, vType := range types {
|
||
if vType == "" {
|
||
t.Error("验证码类型不能为空")
|
||
}
|
||
}
|
||
}
|
||
|
||
// TestVerificationCodeFormat 测试验证码格式
|
||
func TestVerificationCodeFormat(t *testing.T) {
|
||
code, err := GenerateVerificationCode()
|
||
if err != nil {
|
||
t.Fatalf("GenerateVerificationCode() failed: %v", err)
|
||
}
|
||
|
||
// 验证长度
|
||
if len(code) != 6 {
|
||
t.Errorf("验证码长度应为6位，实际为%d位", len(code))
|
||
}
|
||
|
||
// 验证只包含数字
|
||
for i, c := range code {
|
||
if c < '0' || c > '9' {
|
||
t.Errorf("验证码第%d位包含非数字字符: %c", i+1, c)
|
||
}
|
||
}
|
||
}
|
||
|
||
// TestVerificationTypes 测试验证码类型
|
||
func TestVerificationTypes(t *testing.T) {
|
||
validTypes := map[string]bool{
|
||
VerificationTypeRegister: true,
|
||
VerificationTypeResetPassword: true,
|
||
VerificationTypeChangeEmail: true,
|
||
}
|
||
|
||
for vType, isValid := range validTypes {
|
||
if !isValid {
|
||
t.Errorf("验证码类型 %s 应该是有效的", vType)
|
||
}
|
||
if vType == "" {
|
||
t.Error("验证码类型不能为空字符串")
|
||
}
|
||
}
|
||
}
|