unit-testing - httptest.NewRequest 设置上下文 stub

Question

我正在创建一个请求 stub ，以便将其传递给被测试的函数:

request := httptest.NewRequest("GET", "http://example.com/foo", nil)

问题:我还可以通过向其添加 request-uuid 值来为该请求 stub Context 对象吗？

Answer 1

你有请求，你可以在移交之前用它做任何你想做的事。

使用Request.Context()要访问其上下文，请使用 context.WithValue()派生一个新的context.Context在其中包含您的键值，并使用 Request.WithContext()获得新的 http.Request使用新上下文:

request := httptest.NewRequest("GET", "http://example.com/foo", nil)
ctx := request.Context()
ctx = context.WithValue(ctx, "request-uuid", "myvalue")
request = request.WithContext(ctx)

// now request's context contains the "request-uuid" key