backend/internal/service/verification_service_test.go

package service

import (
	"testing"
	"time"
)

// TestGenerateVerificationCode 测试生成验证码函数
func TestGenerateVerificationCode(t *testing.T) {
	// 创建服务实例（使用 nil，因为这个测试不需要依赖）
	svc := &verificationService{}

	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 := svc.generateCode()
			if (err != nil) != tt.wantErr {
				t.Errorf("generateCode() error = %v, wantErr %v", err, tt.wantErr)
				return
			}
			if !tt.wantErr && len(code) != tt.wantLen {
				t.Errorf("generateCode() code length = %v, want %v", len(code), tt.wantLen)
			}
			// 验证验证码只包含数字
			for _, c := range code {
				if c < '0' || c > '9' {
					t.Errorf("generateCode() code contains non-digit: %c", c)
				}
			}
		})
	}

	// 测试多次生成，验证码应该不同（概率上）
	codes := make(map[string]bool)
	for i := 0; i < 100; i++ {
		code, err := svc.generateCode()
		if err != nil {
			t.Fatalf("generateCode() 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) {
	svc := &verificationService{}
	code, err := svc.generateCode()
	if err != nil {
		t.Fatalf("generateCode() 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("验证码类型不能为空字符串")
		}
	}
}