登录
首页 >  Golang >  Go问答

如何UT上传文件

来源:stackoverflow

时间:2024-04-04 23:21:33 495浏览 收藏

珍惜时间,勤奋学习!今天给大家带来《如何UT上传文件》,正文内容主要涉及到等等,如果你正在学习Golang,或者是对Golang有疑问,欢迎大家关注我!后面我会持续更新相关内容的,希望都能帮到正在学习的大家!

问题内容

我正在努力为我的 graphql api 制作 ut。我需要在上传文件的地方测试突变。 我在这个项目中使用 gqlgen。

...
localfile, err := os.open("./file.xlsx")
if err != nil {
    fmt.errorf(err.error())
}

c.mustpost(queries.upload_csv, &resp, client.var("id", id), client.var("file", localfile), client.addheader("authorization", "bearer "+hub.accesstoken))

c.mustpost 恐慌并发送错误:

--- FAIL: TestUploadCSV (0.00s)
panic: [{"message":"map[string]interface {} is not an Upload","path":["uploadCSV","file"]}] [recovered]
panic: [{"message":"map[string]interface {} is not an Upload","path":["uploadCSV","file"]}]

如何将 localfile 发送到我的 api?我考虑过通过curl 来实现它,但我不确定这是否是一种干净的方法。


解决方案


你不能像这样传递 os.file 。您需要实际读取该文件,构建 mime 多部分请求正文(请参阅 spec)并在 post 请求中发送它。

buf := &bytes.Buffer{}
w := multipart.NewWriter(...)

// add other required fields (operations, map) here

// load file (you can do these directly I am emphasizing them 
// as variables so code below is more understandable
fileKey := "0" // file key in 'map'
fileName := "file.xslx" // file name
fileContentType := "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
fileContents, err := ioutil.ReadFile("./file.xlsx")
// ...

// make multipart body
h := make(textproto.MIMEHeader)

h.Set("Content-Disposition", fmt.Sprintf(`form-data; name="%s"; filename="%s"`, fileKey, fileName))
h.Set("Content-Type", fileContentType)
ff, err := bodyWriter.CreatePart(h)
// ...
_, err = ff.Write(fileContents)
// ...
err = bodyWriter.Close()
// ...

req, err := http.NewRequest("POST", fmt.Sprintf("https://endpoint"), buf)
//...

gqlgen 存储库本身中有一个很好的工作示例:example/fileupload/fileupload_test.go

在该示例中,每个文件都加载到我链接的行上定义的 file 结构类型(并由其表示),乍一看可能会有点混乱。

好了,本文到此结束,带大家了解了《如何UT上传文件》,希望本文对你有所帮助!关注golang学习网公众号,给大家分享更多Golang知识!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>