Go中怎么模拟 HTTPS 响应

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《Go中怎么模拟 HTTPS 响应》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

问题内容

我正在尝试为向 Web 服务发出请求的包编写测试。我遇到问题可能是由于我对 TLS 缺乏了解。

目前我的测试看起来像这样：

func TestSimple() {
    server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Content-Type", "application/json")
        w.WriteHeader(200)
        fmt.Fprintf(w, `{ "fake" : "json data here" }`)
    }))
    transport := &http.Transport{
        Proxy: func(req *http.Request) (*url.URL, error) {
            return url.Parse(server.URL)
        },
    }
    // Client is the type in my package that makes requests
    client := Client{
        c: http.Client{Transport: transport},
    }

    client.DoRequest() // ...
}

我的包有一个包变量（我希望它是一个常量..），用于查询 Web 服务的基地址。它是一个 https 网址。我在上面创建的测试服务器是纯 HTTP，没有 TLS。

默认情况下，我的测试失败并出现错误“tls：第一条记录看起来不像 TLS 握手”。

为了让它工作，我的测试在进行查询之前将包变量更改为纯 http URL 而不是 https。

有没有办法解决？我可以将包变量设为常量（https），然后设置一个http.Transport“降级”为未加密的 HTTP，或者httptest.NewTLSServer()改用它吗？

（当我尝试使用时，NewTLSServer()我得到“http：来自 127.0.0.1:45678 的 TLS 握手错误：tls：收到长度为 20037 的超大记录”）

正确答案

中的大多数行为都net/http可以被嘲笑、扩展或改变。虽然http.Client是实现 HTTP 客户端语义的具体类型，但它的所有字段都是导出的并且可以自定义。

特别是，该Client.Transport字段可以被替换以使客户端执行任何操作，从使用自定义协议（例如 ftp:// 或 file://）到直接连接到本地处理程序（不生成 HTTP 协议字节或通过网络发送任何内容） ）。

客户端函数，例如http.Get，都使用导出的http.DefaultClient包变量（您可以修改），因此使用这些便利函数的代码不必更改 为调用自定义客户端变量上的方法。请注意，虽然在公开可用的库中修改全局行为是不合理的，但在应用程序和测试（包括库测试）中这样做非常有用。

http://play.golang.org/p/afljO086iB包含一个http.RoundTripper重写请求 URL 以便将其路由到本地托管的自定义，以及httptest.Server将请求直接传递到的另一个示例http.Handler，以及自定义http.ResponseWriter实现，为了创建一个http.Response. 第二种方法不像第一种那样勤奋（它不会在 Response 值中填写尽可能多的字段），但效率更高，并且应该足够兼容以与大多数处理程序和客户端调用程序一起工作。

上面链接的代码也包含在下面：

package main

import (
    "fmt"
    "io"
    "log"
    "net/http"
    "net/http/httptest"
    "net/url"
    "os"
    "path"
    "strings"
)

func Handler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "hello %sn", path.Base(r.URL.Path))
}

func main() {
    s := httptest.NewServer(http.HandlerFunc(Handler))
    u, err := url.Parse(s.URL)
    if err != nil {
        log.Fatalln("failed to parse httptest.Server URL:", err)
    }
    http.DefaultClient.Transport = RewriteTransport{URL: u}
    resp, err := http.Get("https://google.com/path-one")
    if err != nil {
        log.Fatalln("failed to send first request:", err)
    }
    fmt.Println("[First Response]")
    resp.Write(os.Stdout)

    fmt.Print("n", strings.Repeat("-", 80), "nn")

    http.DefaultClient.Transport = HandlerTransport{http.HandlerFunc(Handler)}
    resp, err = http.Get("https://google.com/path-two")
    if err != nil {
        log.Fatalln("failed to send second request:", err)
    }
    fmt.Println("[Second Response]")
    resp.Write(os.Stdout)
}

// RewriteTransport is an http.RoundTripper that rewrites requests
// using the provided URL's Scheme and Host, and its Path as a prefix.
// The Opaque field is untouched.
// If Transport is nil, http.DefaultTransport is used
type RewriteTransport struct {
    Transport http.RoundTripper
    URL       *url.URL
}

func (t RewriteTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    // note that url.URL.ResolveReference doesn't work here
    // since t.u is an absolute url
    req.URL.Scheme = t.URL.Scheme
    req.URL.Host = t.URL.Host
    req.URL.Path = path.Join(t.URL.Path, req.URL.Path)
    rt := t.Transport
    if rt == nil {
        rt = http.DefaultTransport
    }
    return rt.RoundTrip(req)
}

type HandlerTransport struct{ h http.Handler }

func (t HandlerTransport) RoundTrip(req *http.Request) (*http.Response, error) {
    r, w := io.Pipe()
    resp := &http.Response{
        Proto:      "HTTP/1.1",
        ProtoMajor: 1,
        ProtoMinor: 1,
        Header:     make(http.Header),
        Body:       r,
        Request:    req,
    }
    ready := make(chan struct{})
    prw := &pipeResponseWriter{r, w, resp, ready}
    go func() {
        defer w.Close()
        t.h.ServeHTTP(prw, req)
    }()
    
本篇关于《Go中怎么模拟 HTTPS 响应》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！