Function TestSimulateHTTP_NDJSONStreamMatchAndEarlyClose_319

pkg/util_test.go:539–614

(t *testing.T)

}

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

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

		_, _ = w.Write([]byte("{\"id\":1,\"ok\":true}\n"))
		flusher.Flush()
		_, _ = w.Write([]byte("{\"id\":2,\"ok\":false}\n"))
		flusher.Flush()
		_, _ = w.Write([]byte("{\"id\":3,\"ok\":true}\n"))
		flusher.Flush()

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

	expectedBody := "{\"id\":1,\"ok\":true}\n{\"id\":2,\"ok\":false}\n"

	tc := &models.TestCase{
		Name: "ndjson-match-and-close",
		HTTPReq: models.HTTPReq{
			Method: "GET",
			URL:    server.URL,
		},
		HTTPResp: models.HTTPResp{
			Header: map[string]string{
				"Content-Type": "application/x-ndjson",
			},
			Body: expectedBody,
			StreamBody: []models.HTTPStreamChunk{
				{
					Data: []models.HTTPStreamDataField{
						{Key: "raw", Value: `{"id":1,"ok":true}`},
					},
				},
				{
					Data: []models.HTTPStreamDataField{
						{Key: "raw", Value: `{"id":2,"ok":false}`},
					},
				},
			},
		},
	}

	// Use SimulateHTTPStreaming for streaming test cases
	streamResp, err := SimulateHTTPStreaming(ctx, tc, "test-set", logger, SimulationConfig{APITimeout: 3})
	require.NoError(t, err)
	require.NotNil(t, streamResp)

	// Compare the stream
	noiseKeys := map[string]struct{}{}
	matched, _, _, compareErr := CompareHTTPStream(tc.HTTPResp, streamResp.Reader, streamResp.StreamConfig, noiseKeys, logger)