如何从* http.Request和* httptest.ResponseRecorder创建gin.Context?

时间:2018-10-09 09:22:23

标签: go testing gin-gonic

我想为gin创建一个类似于testify's HTTPBodyContain的测试助手。

我很难从gin.Context*http.Request创建*httptest.ResponseRecorder。我已经写过这样的东西:

func HTTPBodyContains(t *testing.T, handler gin.HandlerFunc, method, url string, values url.Values, str interface{}, msgAndArgs ...interface{}) bool {
    body := HTTPBody(handler, method, url, values)

    contains := strings.Contains(body, fmt.Sprint(str))
    if !contains {
        assert.Fail(t, fmt.Sprintf("Expected response body for \"%s\" to contain \"%s\" but found \"%s\"", url+"?"+values.Encode(), str, body))
    }

    return contains
}

func HTTPBody(handler gin.HandlerFunc, method, url string, values url.Values) string {
    w := httptest.NewRecorder()
    req, err := http.NewRequest(method, url+"?"+values.Encode(), nil)
    if err != nil {
        return ""
    }
    handler(&gin.Context{
        Request: req,
        Writer:  gin.ResponseWriter(w),
    })
    return w.Body.String()
}

但这是行不通的

Writer:  gin.ResponseWriter(w)

1 个答案:

答案 0 :(得分:0)

好吧,看来CreateTestContext可以满足我的需求。

func CreateTestContext(w http.ResponseWriter) (c *Context, r *Engine)
  

CreateTestContext返回用于测试目的的新引擎和上下文