27 lines
675 B
Go
27 lines
675 B
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
|
|
"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)
|
|
}
|
|
}
|