54 lines
1.4 KiB
Go
54 lines
1.4 KiB
Go
package api
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"syscall"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/jackc/pgx/v5/pgconn"
|
|
"github.com/jackc/puddle/v2"
|
|
|
|
"booklib/internal/config"
|
|
"booklib/internal/redispkg"
|
|
)
|
|
|
|
func testCfg() *config.Config {
|
|
return &config.Config{Addr: ":8080", JWTSecret: []byte("s3cret"), ScanInterval: time.Minute, UploadMaxMB: 200,
|
|
TrustedProxies: []string{"172.16.0.0/12"}} // 与 prod 默认一致: 只有 compose 网段内代理才可信
|
|
}
|
|
|
|
func TestHealthz(t *testing.T) {
|
|
r := NewRouter(testCfg(), nil, redispkg.New(""), nil)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/healthz", nil)
|
|
w := httptest.NewRecorder()
|
|
r.ServeHTTP(w, req)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("healthz = %d, want 200", w.Code)
|
|
}
|
|
}
|
|
|
|
func TestDBErrStatus(t *testing.T) {
|
|
for _, tc := range []struct {
|
|
e error
|
|
want int
|
|
}{
|
|
{&pgconn.PgError{Code: "57P01"}, http.StatusServiceUnavailable}, // PG 停机:池内连接被服务端断
|
|
{&pgconn.PgError{Code: "08006"}, http.StatusServiceUnavailable},
|
|
{fmt.Errorf("dial: %w", syscall.ECONNREFUSED), http.StatusServiceUnavailable},
|
|
{puddle.ErrClosedPool, http.StatusServiceUnavailable},
|
|
{errors.New("boom"), http.StatusInternalServerError},
|
|
} {
|
|
w := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(w)
|
|
dbErr(c, tc.e)
|
|
if w.Code != tc.want {
|
|
t.Errorf("dbErr(%v) = %d, want %d", tc.e, w.Code, tc.want)
|
|
}
|
|
}
|
|
}
|