improve MultipartForm test

labstack · Oct 6, 2024 · ab87b63 · ab87b63
1 parent d20a625
commit ab87b63
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/context_test.go b/context_test.go
@@ -774,6 +774,11 @@ func TestContextMultipartForm(t *testing.T) {
  buf := new(bytes.Buffer)
  mw := multipart.NewWriter(buf)
  mw.WriteField("name", "Jon Snow")
+ fileContent := "This is a test file"
+ w, err := mw.CreateFormFile("file", "test.txt")
+ if assert.NoError(t, err) {
+ w.Write([]byte(fileContent))
+ }
  mw.Close()
  req := httptest.NewRequest(http.MethodPost, "/", buf)
  req.Header.Set(HeaderContentType, mw.FormDataContentType())
@@ -782,6 +787,13 @@ func TestContextMultipartForm(t *testing.T) {
  f, err := c.MultipartForm()
  if assert.NoError(t, err) {
  assert.NotNil(t, f)
+
+ files := f.File["file"]
+ if assert.Len(t, files, 1) {
+ file := files[0]
+ assert.Equal(t, "test.txt", file.Filename)
+ assert.Equal(t, int64(len(fileContent)), file.Size)
+ }
  }
 }