如何清除响应写入器中的数据？

本篇文章给大家分享《如何清除响应写入器中的数据？》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

我正在为elasticsearch中间件编写测试，其中我使用一个函数来构建测试服务器，在其中我为每个测试传递一段配置结构，并在处理函数中迭代它们，并将预期响应写入回复作者。这是我的职责。

func newmockclient(url string) (*elasticsearch, error) {
    client, err := elastic.newsimpleclient(elastic.seturl(url))
    if err != nil {
        return nil, fmt.errorf("error while initializing elastic client: %v", err)
    }
    es := &elasticsearch{
        url:    url,
        client: client,
    }
    return es, nil
}

type serversetup struct {
    method, path, body, response string
    httpstatus                   int
}

func buildtestserver(t *testing.t, setups []*serversetup) *httptest.server {
    handlerfunc := http.handlerfunc(func(w http.responsewriter, r *http.request) {
        requestbytes, _ := ioutil.readall(r.body)
        requestbody := string(requestbytes)

        matched := false
        for _, setup := range setups {
            if r.method == setup.method && r.url.escapedpath() == setup.path {
                matched = true
                if setup.httpstatus == 0 {
                    w.writeheader(http.statusok)
                } else {
                    w.writeheader(setup.httpstatus)
                }
                _, err := w.write([]byte(setup.response))
                if err != nil {
                    t.fatalf("unable to write test server response: %v", err)
                }
            }
        }

        if !matched {
            t.fatalf("no requests matched setup. got method %s, path %s, body %s", r.method, r.url.escapedpath(), requestbody)
        }
    })

    return httptest.newserver(handlerfunc)
}

它是从 github.com/github/vulcanizer 复制的。当我使用它运行单个测试时，它工作得很好。例如本次测试

func testcreateindex(t *testing.t) {
    setup := &serversetup{
        method:   "put",
        path:     "/test",
        response: `{"acknowledged": true, "shards_acknowledged": true, "index": "test"}`,
    }

    ts := buildtestserver(t, []*serversetup{setup})

    es, _ := newmockclient(ts.url)

    err := es.createindex(context.background(), "test", nil)
    if err != nil {
        t.fatalf("index creation failed with error: %v\n", err)
    }

}

但是当我尝试在像这样的单个测试中检查不同的行为时，我收到错误 http: multiple response.writeheader requests

func testdeleteindex(t *testing.t) {
    setup := &serversetup{
        method:   "delete",
        path:     "/test",
        response: `{"acknowledged": true}`,
    }

    errsetup := &serversetup{
        method:   "delete",
        path:     "/test",
        response: `{"acknowledged": false}`,
    }

    ctx := context.background()

    ts := buildtestserver(t, []*serversetup{setup, errsetup})
    defer ts.close()

    es, _ := newmockclient(ts.url)

    err := es.deleteindex(ctx, "test")
    if err != nil {
        t.fatalf("index creation failed with error: %v\n", err)
    }

    err = es.deleteindex(ctx, "test")
    if err == nil {
        t.fatal("expected error but not found")
    }
}

我猜测这是因为当我第二次运行deleteindex时，它再次对服务器执行ping操作，但响应编写器已被写入，因此它无法向其中写入任何其他内容。

无论如何，我可以在处理程序函数的开头进行检查，例如

handlerFunc := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    if w != nil{
        // clear data in response writer
    }

// .........



}

解决方案

我认为您所做的不是测试功能的正确方法。您需要将测试与测试用例分开以检查不同的行为，例如：

func Test_DeleteIndex(t *testing.T) {
    t.Run("Should be ok with correct setup", func(t *testing.T) {
        setup := &ServerSetup{
            Method:   "DELETE",
            Path:     "/test",
            Response: `{"acknowledged": true}`,
        }
        ctx := context.Background()
        ts := buildTestServer(t, []*ServerSetup{setup})
        defer ts.Close()
        es, _ := newMockClient(ts.URL)
        err := es.deleteIndex(ctx, "test")
        require.NoError(t, err)
    })

    t.Run("Shouldn't be ok with wrong setup", func(t *testing.T) {
        setup := &ServerSetup{
            Method:   "DELETE",
            Path:     "/test",
            Response: `{"acknowledged": false}`,
        }
        ctx := context.Background()
        ts := buildTestServer(t, []*ServerSetup{setup})
        defer ts.Close()
        es, _ := newMockClient(ts.URL)
        err := es.deleteIndex(ctx, "test")
        require.Error(t, err)
    })
}

到这里，我们也就讲完了《如何清除响应写入器中的数据？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！