193372ca95
Add the untracked SMS service package and its security-focused tests, plus database-backed IAM service unit coverage. This keeps the verification surface small and brings previously validated test/runtime support sources into version control as an isolated change.
81 lines
2.5 KiB
Go
81 lines
2.5 KiB
Go
package sms
|
||
|
||
import (
|
||
"context"
|
||
"testing"
|
||
|
||
"github.com/stretchr/testify/assert"
|
||
)
|
||
|
||
// TestMockSMSService_SendWhenDisabled_ShouldReturnError
|
||
// TDD: 当SMS服务禁用时，SendVerificationCode应返回错误而非成功
|
||
func TestMockSMSService_SendWhenDisabled_ShouldReturnError(t *testing.T) {
|
||
config := &Config{Enabled: false}
|
||
svc := NewMockSMSService(config)
|
||
|
||
codeID, err := svc.SendVerificationCode(context.Background(), "13800138000")
|
||
|
||
// 安全要求：禁用时应返回错误
|
||
assert.Error(t, err)
|
||
assert.Equal(t, ErrSMSServiceDisabled, err)
|
||
assert.Empty(t, codeID)
|
||
}
|
||
|
||
// TestMockSMSService_VerifyWhenDisabled_ShouldReturnError
|
||
// TDD: 当SMS服务禁用时，VerifyCode应返回错误而非验证
|
||
func TestMockSMSService_VerifyWhenDisabled_ShouldReturnError(t *testing.T) {
|
||
config := &Config{Enabled: false}
|
||
svc := NewMockSMSService(config)
|
||
|
||
// 不应接受任何验证码
|
||
valid, err := svc.VerifyCode(context.Background(), "any-id", "13800138000", "123456")
|
||
|
||
// 安全要求：禁用时应返回错误
|
||
assert.Error(t, err)
|
||
assert.Equal(t, ErrSMSServiceDisabled, err)
|
||
assert.False(t, valid)
|
||
}
|
||
|
||
// TestTencentSMSService_SendWhenDisabled_ShouldReturnError
|
||
// TDD: 腾讯云SMS服务禁用时应返回错误
|
||
func TestTencentSMSService_SendWhenDisabled_ShouldReturnError(t *testing.T) {
|
||
config := &Config{Enabled: false}
|
||
svc := NewTencentSMSService(config)
|
||
|
||
codeID, err := svc.SendVerificationCode(context.Background(), "13800138000")
|
||
|
||
assert.Error(t, err)
|
||
assert.Equal(t, ErrSMSServiceDisabled, err)
|
||
assert.Empty(t, codeID)
|
||
}
|
||
|
||
// TestAliyunSMSService_SendWhenDisabled_ShouldReturnError
|
||
// TDD: 阿里云SMS服务禁用时应返回错误
|
||
func TestAliyunSMSService_SendWhenDisabled_ShouldReturnError(t *testing.T) {
|
||
config := &Config{Enabled: false}
|
||
svc, err := NewAliyunSMSService(config)
|
||
assert.NoError(t, err)
|
||
|
||
codeID, err := svc.SendVerificationCode(context.Background(), "13800138000")
|
||
|
||
assert.Error(t, err)
|
||
assert.Equal(t, ErrSMSServiceDisabled, err)
|
||
assert.Empty(t, codeID)
|
||
}
|
||
|
||
// TestDisabledServiceAcceptsNoCodes
|
||
// TDD: 禁用服务时不应接受任何验证码
|
||
func TestDisabledServiceAcceptsNoCodes(t *testing.T) {
|
||
config := &Config{Enabled: false}
|
||
svc := NewMockSMSService(config)
|
||
ctx := context.Background()
|
||
|
||
testCases := []string{"123456", "000000", "999999", "abcdef"}
|
||
|
||
for _, code := range testCases {
|
||
valid, err := svc.VerifyCode(ctx, "test-id", "13800138000", code)
|
||
assert.Error(t, err, "code %s should be rejected when service is disabled", code)
|
||
assert.False(t, valid)
|
||
}
|
||
}
|