Fiber框架怎么为测试环境编写Mock中间件-Golang-PHP中文网

mock中间件是测试时临时替换真实中间件行为的函数，用于隔离依赖、控制上下文或跳过非核心逻辑；fiber无原生支持，需通过变量替换、gock拦截或完整链模拟实现，且必须在app.test()前完成替换。

fiber框架怎么为测试环境编写mock中间件

Mock中间件不是 Fiber 原生概念，得自己造

Fiber 没有内置 “Mock 中间件” 类型或注册机制——它只认 func(*fiber.Ctx) error。所谓 Mock 中间件，本质是**在测试时临时替换真实中间件行为的函数**，用于隔离依赖、控制上下文状态或跳过鉴权/日志等非核心逻辑。

常见错误是试图在测试里调用 app.Use() 动态覆盖已注册中间件，这不行：Fiber 的中间件链在启动时就固化了，app.Use() 只能追加，不能替换或删除。

真实中间件（如 JWT 验证）必须被绕过或重写，不能靠“禁用开关”参数控制
Mock 中间件必须在测试前、app.Test() 调用前完成替换，否则无效
别在 handler 里用 c.Locals 手动塞 mock 数据——那是补丁，不是可复用的中间件

用函数变量替换中间件实现最直接

把要 Mock 的中间件定义成包级变量，测试时直接赋值新函数：

// auth.go
var AuthMiddleware = func(c *fiber.Ctx) error {
    token := c.Get("Authorization")
    if token == "" {
        return c.Status(401).SendString("unauthorized")
    }
    return c.Next()
}

// main.go
app.Use(AuthMiddleware)

测试中替换：

// auth_test.go
func TestWithMockAuth(t *testing.T) {
    // 保存原值，测试后恢复（避免影响其他 test）
    original := AuthMiddleware
    defer func() { AuthMiddleware = original }()

    AuthMiddleware = func(c *fiber.Ctx) error {
        c.Locals("user_id", 123)
        return c.Next()
    }

    app := fiber.New()
    setupRoutes(app) // 此时用的是新 AuthMiddleware

    req, _ := http.NewRequest("GET", "/api/profile", nil)
    resp, _ := app.Test(req)
    // 断言返回 200，而非 401
}

适用于单个中间件、无跨包引用的场景
必须确保业务代码引用的是该变量，而不是内联匿名函数
若中间件在第三方包里（如 fiber/jwt），此法失效，得换方案

用 httptest.NewServer + gock 拦截外部依赖更可靠

当你的 Fiber 服务调用了外部 API（比如用户中心、支付网关），而你想在测试中 Mock 它们的响应，httptest.NewServer 不适用——它只 mock 本服务的 HTTP 入口，不处理 outbound 请求。

Fiber框架 3.0.0

Fiber框架 3.0.0 版本源码包下载，适合从 Fiber v2 迁移到 v3、体验新绑定系统、HTTP 客户端、状态管理和迁移工具的开发者。

下载

此时用 gock 拦截 http.DefaultClient 或自定义 client 发出的请求：

func TestExternalAPIFailure(t *testing.T) {
    defer gock.Off() // 必须清理，否则污染后续测试

    gock.New("https://api.example.com").
        Get("/users/123").
        Reply(503).
        JSON(map[string]string{"error": "service unavailable"})

    app := fiber.New()
    setupRoutes(app)

    req, _ := http.NewRequest("GET", "/profile/123", nil)
    resp, _ := app.Test(req)

    assert.Equal(t, 503, resp.StatusCode)
}