我有一个名为 CreateObject 的处理函数。此函数同时包装了对我无法控制的外部 API 的 POST 请求。如果我想对其进行单元测试，我面临的问题是我无法在每次运行测试时都将新对象发布到外部服务。所以我想知道是否有办法用 Go 或任何解决方法来模拟它。

非常感谢。

主包

func main() {

router := mux.NewRouter()

router.HandleFunc("/groups", services.CreateObject).Methods("POST")
c := cors.New(cors.Options{
    AllowedOrigins:   []string{"*"},
    AllowCredentials: true,
    AllowedMethods:   []string{"GET", "POST", "PUT", "DELETE", "PATCH", "OPTIONS"},
    AllowedHeaders:   []string{"*"},
    ExposedHeaders:   []string{"*"},
})

handler := c.Handler(router)

http.ListenAndServe(":3000", handler)

包对象

  func CreateObject(w http.ResponseWriter, r *http.Request) {



        var newobject Object
        _ = json.NewDecoder(r.Body).Decode(&newobject )

        //Do things

        jsonStr, err := json.Marshal(newobject)
        if err != nil {
            fmt.Println(err)
            return
        }

        req, err := http.NewRequest("POST", ExternalURL+"/object", bytes.NewBuffer(jsonStr))


        if err != nil {
            fmt.Println(err)
            return
        }

        client := &http.Client{}
        resp, err := client.Do(req)
        if err != nil {
            panic(err)
        }
        defer resp.Body.Close()

        body, _ := ioutil.ReadAll(resp.Body)


        if resp.StatusCode < 200 || resp.StatusCode > 299 {
            w.WriteHeader(resp.StatusCode)
            w.Header().Set("Content-Type", "application/json")
            w.Write(body)

        } else {
            w.WriteHeader(201)
        }



}

最佳答案

你有几个选择:

1) 您可以定义一个接口(interface)，该接口(interface)具有 http.Client 的导出方法集。然后你可以创建一个这种类型的包级变量，默认为 *http.Client。您可以使用此变量，而不是在 CreateObject 中使用 *http.Client。由于它是一个接口(interface)，因此您可以轻松模拟客户端。界面看起来像这样:

type HTTPClient interface {
    Do(req *http.Request) (*http.Response, error)
    Get(url string) (resp *http.Response, err error)
    Post(url string, contentType string, body io.Reader) (resp *http.Response, err error)
    PostForm(url string, data url.Values) (resp *http.Response, err error)
    Head(url string) (resp *http.Response, err error)
}

由于您只调用 Do()，因此您的模拟只需要为 Do 定义一个实际的测试实现。我们经常为此使用函数字段样式:

type MockClient struct {
    DoFunc func(req *http.Request) (*http.Response, error)
    // other function fields, if you need them
}

func (m MockClient) Do(req *http.Request) (r *http.Response, err error) {
    if m.DoFunc != nil {
        r, err = m.DoFunc(req)
    }
    return
}

// Define the other 4 methods of the HTTPclient as trivial returns

var mockClient HTTPClient = MockClient{
    DoFunc: func(req *http.Request) (*http.Response, error) {
        return nil, nil
    },
}

var mockClientFail HTTPClient = MockClient{
    DoFunc: func(req *http.Request) (*http.Response, error) {
        return nil, fmt.Errorf("failed")
    },
}

2) 在本地主机端口上建立您自己的 HTTP 模拟服务器，并在您的测试中更改您的 ExternalURL 变量以指向它。这允许您实际测试拨号(这使其更像是功能测试而不是单元测试)，同时仍然有效地“模拟”外部端点。

无论哪种情况，请确保您还编写了一些回归测试以确保外部端点仍按预期工作。

编辑:根据 dm03514，Go 已经内置了一个模拟 HTTP 服务器:https://golang.org/pkg/net/http/httptest/

关于unit-testing - 如何对包装对外部服务的 http 调用的处理程序进行单元测试，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/44727059/