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)
+		}
 	}
 }