php小编小新今天为大家带来了关于Golang模拟强制改变函数定义的介绍。Golang是一种高效、简洁的编程语言,具有强大的类型系统。然而,在某些情况下,我们可能需要修改已有函数的定义,以满足特定的需求。本文将向大家介绍如何在Golang中模拟强制改变函数定义的方法,让我们一起来探索吧!
我有以下功能:
func getprice(date string) { url := (fmt.printf("http://endponint/%s", date)) resp, err := http.get(url) // unmarshall body and get price return price }
为了对该函数进行单元测试,我被迫重构为:
func getprice(client httpclient, date string) { url := (fmt.printf("http://endponint/%s", date)) resp, err := client.get(url) // unmarshall body and get price return price }
我的测试如下所示:
type MockClient struct { Response *http.Response Err error } func (m *MockClient) Get(url string) (*http.Response, error) { return m.Response, m.Err } mockResp := &http.Response{ StatusCode: http.StatusOK, Body: ioutil.NopCloser(strings.NewReader("mock response")), } client := &MockClient{ Response: mockResp, Err: nil, } data, err := getData(client, "http://example.com")
这是在 go 中进行测试的唯一方法吗?没有办法模拟未注入到函数中的 api 吗?
使用 go 进行 http 测试的惯用方法是使用 http/httptest ( 示例)
就您而言,您所需要做的就是使基本 url 可注入:
var apiendpoint = "http://endpoint/" func getprice(date string) (int, error) { url := (fmt.printf("%s/%s", apiendpoint, date)) resp, err := http.get(url) // unmarshall body and get price return price, nil }
然后在每个测试中:
srv := httptest.newserver(http.handlerfunc(func(w http.responsewriter, r *http.request) { // write the expected response to the writer })) defer srv.close() apiendpoint = srv.url price, err := getprice("1/1/2023") // handle error, check return
更好的设计是将您的 api 包装在客户端 struct
中,并使 getprice
成为接收器方法:
type priceclient struct { endpoint string } func (pc *priceclient) getprice(date string) (int, error) { url := (fmt.printf("%s/%s", pc.endpoint, date)) resp, err := http.get(url) // unmarshall body and get price return price, nil }
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { // Write the expected response to the writer })) defer srv.Close() c := &PriceClient{Endpoint: srv.URL} price, err := c.GetPrice("1/1/2023") // Handle error, check return
为了将来的参考,你还应该看看gomock,因为大多数其他模拟问题你会遇到语言没有内置解决方案的情况。
以上是Golang模拟强制改变函数定义的详细内容。更多信息请关注PHP中文网其他相关文章!