当前位置：首页 > 文章列表 > Golang > Go教程 > GOST-DOM如何避免进行HTTP调用

GOST-DOM如何避免进行HTTP调用

2025-02-18 22:39:48 0浏览收藏

珍惜时间，勤奋学习！今天给大家带来《GOST-DOM如何避免进行HTTP调用》，正文内容主要涉及到等等，如果你正在学习Golang，或者是对Golang有疑问，欢迎大家关注我！后面我会持续更新相关内容的，希望都能帮到正在学习的大家！

本文介绍了GOST-DOM的实现，这是一个用Go语言编写的无头浏览器。

Go语言编写的Web应用程序非常易于测试。Web应用程序为单个功能ServeHTTP提供HTTP请求。测试代码只需调用此函数即可测试Web应用程序的行为，但仍然可以通过HTTP请求、响应、头信息、主体和状态码来表示测试，而不是控制器方法调用。

GOST-DOM充分利用了这一点，原因如下：

性能: 测试只需调用Go代码，避免了TCP堆栈的开销。
隔离: 消除了管理TCP端口的需求，极大地简化了测试隔离。这是最重要的部分，虽然不阻止测试进行隔离，但管理TCP端口增加了测试环境的复杂性。

HTTP客户端

我希望浏览器核心无需处理HTTP请求。幸运的是，Go的标准库可以直接使用。

传出的HTTP请求由http.Client实例处理；它通过RoundTripper接口抽象了传输层。

Go的一个重要特性是许多基本操作都是由单方法接口抽象的。RoundTripper只有一个方法：

type RoundTripper interface {
    RoundTrip(*Request) (*Response, error)
}

这意味着，只需创建一个实现RoundTripper接口的类型，该类型调用ServeHTTP即可。

接口实现

http.HandlerFunc接收http.ResponseWriter。这并非具体类型，而是一个接口。

除了促进生产代码的可测试性设计外，Go还支持net/http/httptest包中优秀的测试工具。ResponseRecorder是一个有效的ResponseWriter，测试代码可以传递给实现，并简化了处理响应体流的过程。ResponseRecorder还可以生成合适的*http.Response，通过调用recorder.Result()获得。

在RoundTripper和Handler中表示请求的类型相同，因此我创建了一个副本，以避免测试服务器代码中的突变影响浏览器中生成的请求对象。TestRoundTripper的第一个版本如下：

// TestRoundTripper implements the http.RoundTripper interface
// that communicates directly with an http.Handler instance.
type TestRoundTripper struct {
    http.Handler
}

func (h TestRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
    rec := httptest.NewRecorder()
    // make a copy, so the http handler doesn't mutate the outgoing request
    reqCopy := new(http.Request)
    *reqCopy = *req
    h.ServeHTTP(rec, reqCopy)
    return rec.Result(), nil
}

起初效果很好，但有一些问题。

一个略微奇怪的设计决策

我认为这是一个奇怪的设计决策，在一个设计良好的API中不应该出现。

传递给RoundTripper的*http.Request和传递给http.Handler的*http.Request是相同的类型，但它们实际上是不同的东西。一个是传出的请求，另一个是传入的请求。

虽然它们有相似之处，但并不完全相同，并且该类型具有仅与处理传入请求相关的属性（例如解码表单数据）。它们也有不同的有效性规则。

首先，即将发出的请求可以具有空请求体，而传入请求始终具有主体。其次，传入请求是context.Context的来源。

修复请求很容易：

func (h TestRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
    rec := httptest.NewRecorder()
    body := req.Body
    if body == nil {
        body = nullReader{} // nullReader is a custom type for handling nil bodies
    }
    clientReq, err := http.NewRequest(req.Method, req.URL.String(), body)
    if err != nil {
        return nil, err
    }
    clientReq.Header = req.Header
    clientReq.Trailer = req.Trailer
    h.ServeHTTP(rec, clientReq)
    return rec.Result(), nil
}

现在，浏览器本身将使用提供的http.Client实例创建HTTP请求。测试代码可以控制客户端，用调用http.Handler替换默认的HTTP请求行为，而浏览器的传出请求现在只是一个简单的函数调用。

添加Cookie支持

Go还提供了一个Cookie Jar。因此，添加Cookie支持非常简单。构建http.Client与简单处理程序通信的函数很简单：

import (
    "net/http"
    "net/http/cookiejar"
)

func NewHttpClientFromHandler(handler http.Handler) http.Client {
    cookieJar, err := cookiejar.New(nil)
    if err != nil {
        panic(err)
    }
    return http.Client{
        Transport: TestRoundTripper{Handler: handler},
        Jar:       cookieJar,
    }
}

测试身份提供商集成 (未来功能)

虽然这还不是一个功能，但它旨在扩展此功能以支持多个HTTP处理程序，模拟不同的主机名。这对于测试OAuth身份验证流程或使用外部身份提供商登录可能很有价值。您可以创建一个测试HTTP处理程序来模拟身份提供商的行为，并独立于外部提供商进行测试。

在之前的经验中，这通常是通过使用某些测试用户配置的真实身份提供商来完成的。但是这种方法有一些缺点：