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

为了测试使用反向代理管理子域的 Go 应用程序，需要模拟子域或将测试请求通过 nginx 路由。模拟子域的方法包括硬编码子域为 newrequest 的参数或手动设置主机。将测试请求通过 nginx 路由的方法尚不清楚，但可能需要在测试套件中设置特定的环境变量或配置。

我有一个 go 应用程序，它使用 gin gonic 和 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 context 永远不会从上下文接收子域，并且表现得好像所有内容都来自 localhost:8000。

有没有办法：

• “模拟”子域，以便路由器认为调用来自 foo.localhost.com 而不是 localhost

• 设置我的测试套件，以便测试请求通过 nginx 进行路由。我更喜欢解决方案 1，因为这会导致设置/维护混乱。

编辑：

根据 httptest 文档，我尝试将 foo.localhost 硬编码为 newrequest 的参数，但它并没有真正按照我需要的方式运行：

newrequest 返回一个新的传入服务器请求，适合传递给 http.handler 进行测试。

目标是 rfc 7230“请求目标”：它可以是路径或绝对 url。如果目标是绝对 url，则使用 url 中的主机名。否则，将使用“example.com”。

当硬编码 http://foo.localhost.com/api/login 或 foo.localhost.com/api/login 作为请求目标时，它直接将其传递到我的路由器“foo.localhost.com/api/login”下" 而 nginx 会直接点击 /api/login 并从 c.request.host 进行解析

编辑2：

我目前正在探索使用手动设置主机：

req.header.set("host", "foo.localhost")

解决方案

http.newrequest 返回的请求不适合直接传递给 servehttp。使用 httptest.newrequest 返回的值来代替。

直接设置the Host field即可：

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)
}

终于介绍完啦！小伙伴们，这篇关于《使用反向代理管理子域的 Go 应用程序的端到端/集成测试方法》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！