1 回答
TA贡献1815条经验 获得超12个赞
没有一种方法可以替代 Do 方法,但是有一种方法可以实现您的目标。
创建一个返回任意响应主体的往返类型:
type respondWithReader struct{ body io.Reader }
func (rr respondWithReader) RoundTrip(req *http.Request) (*http.Response, error) {
return &http.Response{
Proto: "HTTP/1.0",
ProtoMajor: 1,
Header: make(http.Header),
Close: true,
Body: ioutil.NopCloser(rr.body),
}, nil
}
创建一个失败的 io.Reader:
var errReadFail = errors.New("blah!")
type failReader int
func (failReader) Read([]byte) (int, error) {
return 0, errReadFail
}
将 stock 客户端与上面的传输和阅读器一起使用:
c := http.Client{Transport: respondWithReader{body: failReader(0)}}
resp, err := c.Get("http://whatever.com")
if err != nil {
t.Error(err)
}
defer resp.Body.Close()
// ReadAll returns errReadFail
_, err = ioutil.ReadAll(resp.Body)
if err != errReadFail {
t.Errorf("got err %v, expect %v", err, errReadFail)
}
- 1 回答
- 0 关注
- 65 浏览
添加回答
举报