李建琦
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
137 lines
4.0 KiB
Go
137 lines
4.0 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestFinalizeProxyQualityResult_ScoreAndGrade(t *testing.T) {
|
|
result := &ProxyQualityCheckResult{
|
|
PassedCount: 2,
|
|
WarnCount: 1,
|
|
FailedCount: 1,
|
|
ChallengeCount: 1,
|
|
}
|
|
|
|
finalizeProxyQualityResult(result)
|
|
|
|
require.Equal(t, 38, result.Score)
|
|
require.Equal(t, "F", result.Grade)
|
|
require.Contains(t, result.Summary, "通过 2 项")
|
|
require.Contains(t, result.Summary, "告警 1 项")
|
|
require.Contains(t, result.Summary, "失败 1 项")
|
|
require.Contains(t, result.Summary, "挑战 1 项")
|
|
}
|
|
|
|
func TestRunProxyQualityTarget_CloudflareChallenge(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.Header().Set("Content-Type", "text/html")
|
|
w.Header().Set("cf-ray", "test-ray-123")
|
|
w.WriteHeader(http.StatusForbidden)
|
|
_, _ = w.Write([]byte("<!DOCTYPE html><title>Just a moment...</title><script>window._cf_chl_opt={};</script>"))
|
|
}))
|
|
defer server.Close()
|
|
|
|
target := proxyQualityTarget{
|
|
Target: "openai",
|
|
URL: server.URL,
|
|
Method: http.MethodGet,
|
|
AllowedStatuses: map[int]struct{}{
|
|
http.StatusUnauthorized: {},
|
|
},
|
|
}
|
|
|
|
item := runProxyQualityTarget(context.Background(), server.Client(), target)
|
|
require.Equal(t, "challenge", item.Status)
|
|
require.Equal(t, http.StatusForbidden, item.HTTPStatus)
|
|
require.Equal(t, "test-ray-123", item.CFRay)
|
|
}
|
|
|
|
func TestRunProxyQualityTarget_AllowedStatusPass(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
_, _ = w.Write([]byte(`{"models":[]}`))
|
|
}))
|
|
defer server.Close()
|
|
|
|
target := proxyQualityTarget{
|
|
Target: "gemini",
|
|
URL: server.URL,
|
|
Method: http.MethodGet,
|
|
AllowedStatuses: map[int]struct{}{
|
|
http.StatusOK: {},
|
|
},
|
|
}
|
|
|
|
item := runProxyQualityTarget(context.Background(), server.Client(), target)
|
|
require.Equal(t, "pass", item.Status)
|
|
require.Equal(t, http.StatusOK, item.HTTPStatus)
|
|
}
|
|
|
|
func TestRunProxyQualityTarget_AllowedStatusPassForUnauthorized(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
_, _ = w.Write([]byte(`{"error":"unauthorized"}`))
|
|
}))
|
|
defer server.Close()
|
|
|
|
target := proxyQualityTarget{
|
|
Target: "openai",
|
|
URL: server.URL,
|
|
Method: http.MethodGet,
|
|
AllowedStatuses: map[int]struct{}{
|
|
http.StatusUnauthorized: {},
|
|
},
|
|
}
|
|
|
|
item := runProxyQualityTarget(context.Background(), server.Client(), target)
|
|
require.Equal(t, "pass", item.Status)
|
|
require.Equal(t, http.StatusUnauthorized, item.HTTPStatus)
|
|
require.Contains(t, item.Message, "目标可达")
|
|
}
|
|
|
|
func TestProxyQualityTargets_IncludesGrok(t *testing.T) {
|
|
var grokTarget *proxyQualityTarget
|
|
for i := range proxyQualityTargets {
|
|
if proxyQualityTargets[i].Target == "grok" {
|
|
grokTarget = &proxyQualityTargets[i]
|
|
break
|
|
}
|
|
}
|
|
|
|
require.NotNil(t, grokTarget)
|
|
require.Equal(t, "https://api.x.ai/v1/models", grokTarget.URL)
|
|
require.Equal(t, http.MethodGet, grokTarget.Method)
|
|
require.Contains(t, grokTarget.AllowedStatuses, http.StatusUnauthorized)
|
|
}
|
|
|
|
func TestRunProxyQualityTarget_GrokUnauthorizedPasses(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet {
|
|
t.Errorf("unexpected method: %s", r.Method)
|
|
}
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
_, _ = w.Write([]byte(`{"error":"unauthorized"}`))
|
|
}))
|
|
defer server.Close()
|
|
|
|
target := proxyQualityTarget{
|
|
Target: "grok",
|
|
URL: server.URL,
|
|
Method: http.MethodGet,
|
|
AllowedStatuses: map[int]struct{}{
|
|
http.StatusUnauthorized: {},
|
|
},
|
|
}
|
|
|
|
item := runProxyQualityTarget(context.Background(), server.Client(), target)
|
|
require.Equal(t, "grok", item.Target)
|
|
require.Equal(t, "pass", item.Status)
|
|
require.Equal(t, http.StatusUnauthorized, item.HTTPStatus)
|
|
require.Contains(t, item.Message, "目标可达")
|
|
}
|