Golang 从httpmock 了解http.RoundTripper

最近组内推进Golang的单元测试，代码有很多第三方请求的逻辑，如果冒昧的写单测可能影响第三方数据，或者拿不到期望数据。这个时候我们了解到可以通过github.com/jarcoal/httpmock（后面统称httpmock）来解决这个问题。

我们先来看一段代码：

func TestHttpReq(t *testing.T) {
	httpmock.Activate()
	defer httpmock.DeactivateAndReset()

	//mock期望的请求方式、url与返回结果
	httpmock.RegisterResponder("GET", "https://foo.bar.com/articles",
		httpmock.NewStringResponder(200, `[{"id": 1, "name": "My Great Article"}]`))

	client := &http.Client{}
	req, err := http.NewRequest("GET", "https://foo.bar.com/articles", nil)

	if err != nil {
		fmt.Println(err)
		return
	}
	//req.Header.Add("Cookie", "IPISP=ISP; IPLOC=ZZ")

	res, err := client.Do(req)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer res.Body.Close()

	body, err := ioutil.ReadAll(res.Body)
	if err != nil {
		fmt.Println(err)
		return
	}
	fmt.Println(string(body))
}

然后我们来执行上面的代码

[@gd_54_224 gopkgs]$ go test -v -test.run=TestHttpReq ./httputil/
=== RUN   TestHttpReq
[{"id": 1, "name": "My Great Article"}]
--- PASS: TestHttpReq (0.00s)
PASS
ok      github.com/goith/gopkgs/httputil        0.011s
[@gd_54_224 gopkgs]$ 

如上所示，发现结果确实是mock的期望返回结果。

那么这个原理是什么呢？httpmock在中间做了哪些事情？带着这个好奇心我们来看httpmock是怎么做的。

首先，我们先看下httpmock.Activate的实现：

func Activate() {
	...
	// make sure that if Activate is called multiple times it doesn't
	// overwrite the InitialTransport with a mock transport.
	if http.DefaultTransport != DefaultTransport {
		InitialTransport = http.DefaultTransport
	}

	http.DefaultTransport = DefaultTransport
}