模拟多个URL以返回虚拟内容的方法

从现在开始，努力学习吧！本文《模拟多个URL以返回虚拟内容的方法》主要讲解了等等相关知识点，我会在golang学习网中持续更新相关的系列文章，欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧，希望能帮到你！

我正在编写某种递归解析器。最简单的形式是：

1. 从第一个链接的页面正文中获取all链接
2. 对每个链接重复第一步

所以现在我想测试一下。问题是我无法找出模拟所有这些页面的最佳方法。我使用 http 包，并且我已经使用 httptest 包编写了一些测试（通过 httptest.NewServer）。但现在看来对我的任务来说没有什么用处。我想最好的方法是将 http.Client 与自定义 Transport 结构一起使用，但它有很多样板文件和额外的臭代码。有没有更优雅的方法来做到这一点？

解决方案

我曾多次使用自定义 transport 来测试客户端。通常我会创建一些辅助类型和函数来减少样板代码。

这样的事情可能是一个开始。

package main

import (
    "bytes"
    "fmt"
    "io"
    "io/ioutil"
    "net/http"
    "os"
)

type roundTripFunc func(*http.Request) (*http.Response, error)

func (r roundTripFunc) RoundTrip(req *http.Request) (resp *http.Response, err error) {
    return r(req)
}

func main() {
    c := &http.Client{
        Transport: roundTripFunc(func(req *http.Request) (resp *http.Response, err error) {
            return &http.Response{
                StatusCode: 200,
                Body:       ioutil.NopCloser(bytes.NewBufferString("test")),
            }, nil
        }),
    }

    r, _ := c.Get("/")
    fmt.Printf("%#v\n", r)
    io.Copy(os.Stdout, r.Body)
}

例如，如果您测试 json api 客户端，您可以创建一个往返辅助函数来负责解码、编码、标头等。在您的情况下，也许您可​​以使往返函数将主机标头和 url 路径映射到文件装置中路径？

今天关于《模拟多个URL以返回虚拟内容的方法》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！