如何在Go测试中模拟文件上传？

在 Go 中测试文件上传

测试处理文件上传的端点时，需要设置 Request.FormFile 字段。不幸的是，简单地模拟完整的 FormFile 结构是一种过于复杂的方法。相反，可以利用 mime/multipart 包来创建必要的 FormFile 实例。

使用 CreateFormFile

CreateFormFile 函数是 Writer 类型的成员，提供了一种生成具有特定字段名称和文件名的表单数据标头的便捷方法。然后可以将生成的 io.Writer 传递给 httptest.NewRequest 函数。

使用管道的示例

一种方法是将 FormFile 写入 io.ReaderWriter 缓冲区或使用 io.Pipe。以下示例演示了后一种方法：

<code class="go">// Create a pipe to prevent buffering.
pr, pw := io.Pipe()

// Transform data to multipart form data and write it to the pipe.
writer := multipart.NewWriter(pw)
defer writer.Close()

go func() {
    // Create the "fileupload" form data field.
    part, err := writer.CreateFormFile("fileupload", "someimg.png")
    if err != nil {
        t.Error(err)
    }

    // Generate the image bytes.
    img := createImage()

    // Encode the image to the form data field writer.
    err = png.Encode(part, img)
    if err != nil {
        t.Error(err)
    }
}

// Read from the pipe into a new httptest.Request.
request := httptest.NewRequest("POST", "/", pr)
request.Header.Add("Content-Type", writer.FormDataContentType())</code>

处理请求

使用请求中的 FormFile 数据，您可以像平常一样在测试的端点中处理它。示例函数演示了在上传目录中创建文件。

附加说明

此方法允许您动态创建表单数据，并将其传递给测试框架，无需使用临时文件。您可以类似地使用encoding/csv 生成CSV 文件，而无需从文件系统读取。

以上是如何在Go测试中模拟文件上传？的详细内容。更多信息请关注PHP中文网其他相关文章！