Function TestSimulateHTTP_MultipartStreamMatchAndEarlyClose_320

pkg/util_test.go:616–704

(t *testing.T)

}

func TestSimulateHTTP_MultipartStreamMatchAndEarlyClose_320(t *testing.T) {
	ctx := context.Background()
	logger := zap.NewNop()
	const boundary = "myCustomBoundary"

	writePart := func(w http.ResponseWriter, contentType string, body string) {
		_, _ = w.Write([]byte("--" + boundary + "\r\n"))
		_, _ = w.Write([]byte("Content-Type: " + contentType + "\r\n\r\n"))
		_, _ = w.Write([]byte(body))
		_, _ = w.Write([]byte("\r\n"))
	}

	serverClosed := make(chan struct{})
	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		w.Header().Set("Content-Type", "multipart/x-mixed-replace; boundary="+boundary)
		flusher, ok := w.(http.Flusher)
		require.True(t, ok, "server response writer must support flushing")

		writePart(w, "application/json", `{"frame":1}`)
		flusher.Flush()
		writePart(w, "text/plain", "frame-2")
		flusher.Flush()
		writePart(w, "text/plain", "frame-3")
		flusher.Flush()

		<-r.Context().Done()
		close(serverClosed)
	}))
	defer server.Close()

	expectedBody := strings.Join([]string{
		"--" + boundary,
		"Content-Type: application/json",
		"",
		`{"frame":1}`,
		"--" + boundary,
		"Content-Type: text/plain",
		"",
		"frame-2",
		"",
	}, "\r\n")

	tc := &models.TestCase{
		Name: "multipart-match-and-close",
		HTTPReq: models.HTTPReq{
			Method: "GET",
			URL:    server.URL,
		},
		HTTPResp: models.HTTPResp{
			Header: map[string]string{
				"Content-Type": "multipart/x-mixed-replace; boundary=" + boundary,
			},
			Body: expectedBody,
			StreamBody: []models.HTTPStreamChunk{
				{
					Data: []models.HTTPStreamDataField{
						{Key: "raw", Value: expectedBody},
					},