如何端到端/集成测试使用反向代理管理子域的 Go 应用程序？

Question

我有一个使用 Gin gonic 的 Go 应用程序和一个 Nginx 反向代理，它们将流量发送到 domain.com 上的另一个应用程序，并将所有 *.domain.com 子域流量直接发送到我的 Go 应用程序。然后我的 Go 应用程序有一个中间件，它将读取 nginx 从 Context 传递给它的主机名，并允许我的处理程序知道正在请求哪个子域，并为所述子域返回正确的数据和 cookie。这是一个非常简单的设置，从我在邮递员中的测试来看，它似乎工作正常，因为我所有的子域中的所有路由都是相同的，所以这样我只能为所有子域使用一个路由器，而不是每个子域使用一个路由器。现在，当我尝试进行端到端测试时，我的大问题就来了。我正在这样设置我的测试：  router := initRouter()  w := httptest.NewRecorder()  req, _ := http.NewRequest("POST", "/api/login", bytes.NewBuffer(jsonLogin))  req.Header.Set("Content-Type", "application/json")  router.ServeHTTP(w, req)  assert.Equal(t, 200, w.Code)返回initRouter()一个加载了我所有路由和中间件的 gin 引擎，其余的作为基本测试设置。显然，测试将失败，因为 gin 上下文永远不会从上下文中接收子域，并且表现得好像所有内容都来自 localhost:8000。有没有办法：“模拟”一个子域，以便路由器认为呼叫来自 foo.localhost.com 而不是 localhost设置我的测试套装，以便通过 nginx 路由测试请求。我更喜欢解决方案 1，因为这将是设置/维护的一团糟。

Answer 1

http.NewRequest 返回的请求不适合直接传递给ServeHTTP。请改用 httptest.NewRequest 返回的一个。

只需直接设置Host 字段：

package main

import (

    "net/http"

    "net/http/httptest"

    "testing"

)

func TestHelloWorld(t *testing.T) {

    mux := http.NewServeMux()

    mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {

        if r.Host != "foobar" {

            t.Errorf("Host is %q, want foobar", r.Host)

        }

    })

    w := httptest.NewRecorder()

    r := httptest.NewRequest("GET", "/api/login", nil)

    r.Host = "foobar"

    mux.ServeHTTP(w, r)

}