当前位置：首页 > 文章列表 > Golang > Go问答 > Golang 单元测试常见的错误问题

Golang 单元测试常见的错误问题

来源：stackoverflow 2024-02-13 11:18:24 0浏览收藏

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《Golang 单元测试常见的错误问题》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

问题内容

我在尝试测试此函数时遇到问题，但内部有错误。以下是我的 responsejson 函数，它不返回错误，但发送响应 json。

func responsejson(w http.responsewriter, code int, message string) {
    jsonstatus := struct {
        code int `json:"code"`
        message string `json:"message"`
    }{
        message: message,
        code: code,
    }

    bs, err := json.marshal(jsonstatus);

    if err != nil {
        log.println("error in marshal json in responsejson: ", err)
        str := "internal server error. please contact the system administrator."
        io.writestring(w, str);
        return 
    } else {
        io.writestring(w, string(bs));
        return
    }
    
}

以下是我的单元测试代码，它创建了一个模拟 responsewriter，它能够成功测试编写器响应 json 的情况，没有错误。由于我没有在 responsejson() 函数中返回错误类型，因此如何在 test_responsejson 函数中测试它，如下所示？

func test_responsejson(t *testing.t) {
    responsejsontests := []struct {
        testname string
        code int
        message string
        expectedjsonresponse string
    } {
        {"successful login", http.statusok, "successfully logged in!", `{"code":200,"message":"successfully logged in!"}`},
        {"existing username", http.statusbadrequest, "username already exists. please try again.", `{"code":400,"message":"username already exists. please try again."}`},
    }

    for _, e := range responsejsontests {
        // creating a mock responsewriter
        w := httptest.newrecorder()

        responsejson(w, e.code, e.message)

        // read the response body as a string
        body, _ := io.readall(w.result().body)
        actual := string(body)

        expected := e.expectedjsonresponse
        if actual != expected {
            t.errorf("%s: expected %s but got %s", e.testname, e.expectedjsonresponse, actual)
        }
    }
}

此外，我还创建了一个函数，它为 log.println() 内置函数生成实际的日志输出。我知道 log.println() 函数是一个内置函数，它不太可能失败。但是，我希望在单元测试中实现 100% 的覆盖率。请帮忙！谢谢:)

func GenerateLogOutput(message string, errorMessage string) string {
    // Create a new bytes.Buffer to capture the log output
    var buf bytes.Buffer

    // Redirect log output to a different destination set as a buffer
    // By default, log message are written to the standard error stream os.Stderr
    log.SetOutput(&buf)

    // Generate an error
    err := errors.New(errorMessage)

    w := httptest.NewRecorder()

    // Calling the function
    InternalServerError(w, message, err)
    actualOutput := buf.String()

    return actualOutput
}

正确答案

简单地说，我们可以为 responsejson 函数编写一个测试用例，如下所示。

func Test_ResponseJson(t *testing.T) {
    tests := []struct {
        Code        int
        Message     string
        ExpectedStr string
    }{
        {
            Code:        1,
            Message:     "sample message",
            ExpectedStr: "{\"code\":1,\"message\":\"sample message\"}",
        },
    }
    for _, test := range tests {
        w := httptest.NewRecorder()

        ResponseJson(w, test.Code, test.Message)

        res := w.Result()
        data, err := ioutil.ReadAll(res.Body)
        res.Body.Close()

        actualStr := string(data)

        assert.Nil(t, err, "Invalid test data")
        assert.Equal(t, actualStr, test.ExpectedStr)
    }
}

我们无法从 bs 获取错误，err := json.marshal(jsonstatus)。 json.marshal 函数可以返回两种类型的错误。