Function TestSimulateHTTP_MultipartRebuildWithFileNames_315

pkg/util_test.go:144–227

(t *testing.T)

}

func TestSimulateHTTP_MultipartRebuildWithFileNames_315(t *testing.T) {
	ctx := context.Background()
	logger := zap.NewNop()

	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		contentType := r.Header.Get("Content-Type")
		mediaType, params, err := mime.ParseMediaType(contentType)
		if err != nil {
			t.Errorf("failed to parse media type: %v", err)
			w.WriteHeader(http.StatusBadRequest)
			return
		}
		assert.Equal(t, "multipart/form-data", mediaType)
		if params["boundary"] == "" {
			t.Errorf("missing multipart boundary")
			w.WriteHeader(http.StatusBadRequest)
			return
		}
		assert.NotEqual(t, "legacy-boundary", params["boundary"])

		reader := multipart.NewReader(r.Body, params["boundary"])
		fields := map[string]string{}
		files := map[string]string{}
		fileNames := map[string]string{}
		for {
			part, err := reader.NextPart()
			if err == io.EOF {
				break
			}
			if err != nil {
				t.Errorf("failed to read multipart part: %v", err)
				w.WriteHeader(http.StatusBadRequest)
				return
			}
			name := part.FormName()
			data, err := io.ReadAll(part)
			if err != nil {
				t.Errorf("failed to read multipart part data: %v", err)
				w.WriteHeader(http.StatusBadRequest)
				return
			}
			if part.FileName() != "" {
				files[name] = string(data)
				fileNames[name] = part.FileName()
				continue
			}
			fields[name] = string(data)
		}

		assert.Equal(t, "text-value", fields["text"])
		assert.Equal(t, "binary-content", files["payload"])
		assert.Equal(t, "blob.bin", fileNames["payload"])

		w.WriteHeader(http.StatusOK)
	}))
	defer server.Close()

	tc := &models.TestCase{