fusenapi/server/upload/test/uploadfilebackendhandler_test.go
2023-07-07 18:44:34 +08:00

75 lines
1.6 KiB
Go
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

package test
import (
"fmt"
"fusenapi/utils/format"
fstests "fusenapi/utils/fstests"
"log"
"os"
"testing"
"github.com/474420502/requests"
"github.com/tidwall/gjson"
)
func TestCaseRenderMegre(t *testing.T) {
var err error
var resp *requests.Response
var result gjson.Result
// 获取 session并携带 JWT token
ses := fstests.GetSessionWithUserToken(t, userver, cnf.Host, cnf.Port)
tp := ses.Post(fmt.Sprintf("http://%s:%d/upload/upload-file-backend", cnf.Host, cnf.Port))
mp := tp.CreateBodyMultipart()
f, err := os.Open("./fusen.webp")
if err != nil {
panic(err)
}
err = mp.AddFieldFile("file", f.Name(), f)
if err != nil {
log.Println(err)
}
err = mp.AddField("file_type", "image")
if err != nil {
panic(err)
}
err = mp.AddField("category", string(format.TCategoryRenderMegre))
if err != nil {
panic(err)
}
tp.SetBodyFormData(mp)
// 向服务器发送 GET 请求,获取用户基本信息
resp, err = tp.TestExecute(gserver)
if err != nil {
t.Error(err)
}
result = resp.Json()
if result.Get("code").Int() != 200 {
t.Error("code != 200")
}
if !result.Get("data").Exists() {
t.Error("data not exists")
}
if !result.Get("data.upload_url").Exists() {
t.Error("upload_url not exists")
}
// 向服务器发送 GET 请求,访问图片链接
imgResp, err := requests.Get(result.Get("data.upload_url").String()).Execute()
if err != nil {
t.Errorf("failed to access image URL: %v", err)
}
// 检查图片响应状态码
if imgResp.GetStatusCode() != 200 {
t.Errorf("unexpected image response status: %d", imgResp.GetStatusCode())
}
}