李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
94 lines
3.5 KiB
Go
94 lines
3.5 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
)
|
|
|
|
// newAliyunCaptchaTestTarget 起一个假的阿里云端点，让真实 SDK 走完整的签名/序列化链路。
|
|
func newAliyunCaptchaTestTarget(t *testing.T, handler http.HandlerFunc) (*aliyunCaptchaVerifier, service.AliyunCaptchaCredentials) {
|
|
t.Helper()
|
|
server := httptest.NewServer(handler)
|
|
t.Cleanup(server.Close)
|
|
|
|
verifier := &aliyunCaptchaVerifier{protocol: "HTTP", timeoutMillis: 2_000}
|
|
cred := service.AliyunCaptchaCredentials{
|
|
AccessKeyID: "test-ak-id",
|
|
AccessKeySecret: "test-ak-secret",
|
|
SceneID: "scene-1",
|
|
Endpoint: strings.TrimPrefix(server.URL, "http://"),
|
|
}
|
|
return verifier, cred
|
|
}
|
|
|
|
func TestAliyunCaptchaVerifier_VerifySuccess(t *testing.T) {
|
|
var capturedParam, capturedSceneID string
|
|
verifier, cred := newAliyunCaptchaTestTarget(t, func(w http.ResponseWriter, r *http.Request) {
|
|
require.NoError(t, r.ParseForm())
|
|
capturedParam = r.Form.Get("CaptchaVerifyParam")
|
|
capturedSceneID = r.Form.Get("SceneId")
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"Code":"Success","Message":"success","RequestId":"req-1","Success":true,"Result":{"VerifyResult":true,"VerifyCode":"T001"}}`))
|
|
})
|
|
|
|
result, err := verifier.VerifyCaptcha(context.Background(), cred, "the-verify-param")
|
|
require.NoError(t, err)
|
|
require.True(t, result.VerifyResult)
|
|
require.Equal(t, "T001", result.VerifyCode)
|
|
require.Equal(t, "the-verify-param", capturedParam)
|
|
require.Equal(t, "scene-1", capturedSceneID)
|
|
}
|
|
|
|
func TestAliyunCaptchaVerifier_VerifyResultFalse(t *testing.T) {
|
|
verifier, cred := newAliyunCaptchaTestTarget(t, func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte(`{"Code":"Success","RequestId":"req-2","Success":true,"Result":{"VerifyResult":false,"VerifyCode":"F002"}}`))
|
|
})
|
|
|
|
result, err := verifier.VerifyCaptcha(context.Background(), cred, "bad-param")
|
|
require.NoError(t, err)
|
|
require.False(t, result.VerifyResult)
|
|
require.Equal(t, "F002", result.VerifyCode)
|
|
}
|
|
|
|
func TestAliyunCaptchaVerifier_APIErrorNormalized(t *testing.T) {
|
|
verifier, cred := newAliyunCaptchaTestTarget(t, func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(http.StatusForbidden)
|
|
_, _ = w.Write([]byte(`{"Code":"SignatureDoesNotMatch","Message":"Specified signature is not matched with our calculation.","RequestId":"req-3"}`))
|
|
})
|
|
|
|
_, err := verifier.VerifyCaptcha(context.Background(), cred, "param")
|
|
require.Error(t, err)
|
|
var apiErr *service.AliyunCaptchaAPIError
|
|
require.ErrorAs(t, err, &apiErr)
|
|
require.Equal(t, "SignatureDoesNotMatch", apiErr.Code)
|
|
}
|
|
|
|
func TestAliyunCaptchaVerifier_TransportError(t *testing.T) {
|
|
server := httptest.NewServer(http.NotFoundHandler())
|
|
endpoint := strings.TrimPrefix(server.URL, "http://")
|
|
server.Close() // 立即关闭，制造连接失败
|
|
|
|
verifier := &aliyunCaptchaVerifier{protocol: "HTTP", timeoutMillis: 2_000}
|
|
cred := service.AliyunCaptchaCredentials{
|
|
AccessKeyID: "test-ak-id",
|
|
AccessKeySecret: "test-ak-secret",
|
|
SceneID: "scene-1",
|
|
Endpoint: endpoint,
|
|
}
|
|
|
|
_, err := verifier.VerifyCaptcha(context.Background(), cred, "param")
|
|
require.Error(t, err)
|
|
var apiErr *service.AliyunCaptchaAPIError
|
|
require.False(t, errors.As(err, &apiErr), "transport errors must not be normalized to API errors")
|
|
}
|