chengcheng
bfa17a3aba
ci / verify (pull_request) Successful in 11m22s
为 PostgreSQL 连接、就绪检查和本地登录设置分层超时，数据库不可用时返回稳定 503 错误码并记录无凭据的连接池统计。 前端登录在 10 秒后取消请求并兼容调用方 AbortSignal，Nginx 登录精确路由限制上游为 15 秒，同时更新 OpenAPI 和回归测试。
50 lines
1.6 KiB
Go
50 lines
1.6 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
)
|
|
|
|
func TestPostgresPoolConfigSetsDiagnosticAndConnectTimeout(t *testing.T) {
|
|
config, err := postgresPoolConfig("postgresql://gateway:password@localhost:5432/gateway?sslmode=disable")
|
|
if err != nil {
|
|
t.Fatalf("parse PostgreSQL pool config: %v", err)
|
|
}
|
|
if config.ConnConfig.ConnectTimeout != 5*time.Second {
|
|
t.Fatalf("connect timeout = %s, want 5s", config.ConnConfig.ConnectTimeout)
|
|
}
|
|
if applicationName := config.ConnConfig.RuntimeParams["application_name"]; applicationName != "easyai-ai-gateway" {
|
|
t.Fatalf("application_name = %q, want easyai-ai-gateway", applicationName)
|
|
}
|
|
}
|
|
|
|
func TestPostgresPoolConfigRejectsMalformedURL(t *testing.T) {
|
|
if _, err := postgresPoolConfig("://malformed"); err == nil {
|
|
t.Fatal("expected malformed PostgreSQL URL to fail")
|
|
}
|
|
}
|
|
|
|
func TestIsPostgresUnavailableClassifiesConnectivityFailures(t *testing.T) {
|
|
for _, testCase := range []struct {
|
|
name string
|
|
err error
|
|
}{
|
|
{name: "deadline", err: context.DeadlineExceeded},
|
|
{name: "connection exception", err: &pgconn.PgError{Code: "08006"}},
|
|
{name: "too many connections", err: &pgconn.PgError{Code: "53300"}},
|
|
{name: "cannot connect now", err: &pgconn.PgError{Code: "57P03"}},
|
|
} {
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
if !IsPostgresUnavailable(testCase.err) {
|
|
t.Fatalf("error %v was not classified as PostgreSQL unavailable", testCase.err)
|
|
}
|
|
})
|
|
}
|
|
if IsPostgresUnavailable(&pgconn.PgError{Code: "42601"}) {
|
|
t.Fatal("SQL syntax error was incorrectly classified as PostgreSQL unavailable")
|
|
}
|
|
}
|