62 lines
2.2 KiB
Go
62 lines
2.2 KiB
Go
package service
|
|
|
|
import "testing"
|
|
|
|
func TestProviderFailureDetailsReadsTopLevelModerationError(t *testing.T) {
|
|
code, message := providerFailureDetails(map[string]any{
|
|
"code": contentModerationErrorCode,
|
|
"message": "prompt rejected",
|
|
})
|
|
if code != contentModerationErrorCode {
|
|
t.Fatalf("unexpected code: %q", code)
|
|
}
|
|
if message != "prompt rejected" {
|
|
t.Fatalf("unexpected message: %q", message)
|
|
}
|
|
}
|
|
|
|
func TestProviderFailureDetailsReadsNestedError(t *testing.T) {
|
|
code, message := providerFailureDetails(map[string]any{
|
|
"error": map[string]any{"code": "invalid_request", "message": "invalid size"},
|
|
})
|
|
if code != "invalid_request" || message != "invalid size" {
|
|
t.Fatalf("unexpected failure details: code=%q message=%q", code, message)
|
|
}
|
|
}
|
|
|
|
func TestProviderFailureDetailsPrefersNestedBusinessCode(t *testing.T) {
|
|
code, message := providerFailureDetails(map[string]any{
|
|
"code": float64(400),
|
|
"data": map[string]any{"code": contentModerationErrorCode, "message": "prompt rejected"},
|
|
})
|
|
if code != contentModerationErrorCode || message != "prompt rejected" {
|
|
t.Fatalf("unexpected wrapped failure details: code=%q message=%q", code, message)
|
|
}
|
|
}
|
|
|
|
func TestContentModerationFailureRequiresExactProviderCode(t *testing.T) {
|
|
if !isContentModerationFailure(`{"code":"sensitive_words_detected"}`) {
|
|
t.Fatal("expected moderation error to be detected")
|
|
}
|
|
if isContentModerationFailure("上游 HTTP 400") {
|
|
t.Fatal("generic HTTP 400 must remain retryable")
|
|
}
|
|
}
|
|
|
|
func TestProviderPayloadBusinessFailureRecognizesStringErrorCode(t *testing.T) {
|
|
code, message, failed := providerPayloadBusinessFailure(map[string]any{
|
|
"code": "RequestParameterIsWrong",
|
|
"data": nil,
|
|
"msg": "参数: prompt 的长度: 23142 大于最大长度 10000",
|
|
})
|
|
if !failed || code != "RequestParameterIsWrong" || message != "参数: prompt 的长度: 23142 大于最大长度 10000" {
|
|
t.Fatalf("business failure = (%q, %q, %v)", code, message, failed)
|
|
}
|
|
}
|
|
|
|
func TestProviderPayloadBusinessFailureAcceptsStringSuccessCode(t *testing.T) {
|
|
if code, message, failed := providerPayloadBusinessFailure(map[string]any{"code": "Success", "data": map[string]any{"task_id": "task-1"}}); failed {
|
|
t.Fatalf("success payload was marked failed: (%q, %q)", code, message)
|
|
}
|
|
}
|