Function TestSimulateHTTP_SSEStreamMatchAndEarlyClose_316

pkg/util_test.go:229–320

(t *testing.T)

}

func TestSimulateHTTP_SSEStreamMatchAndEarlyClose_316(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", "text/event-stream; charset=utf-8")
		flusher, ok := w.(http.Flusher)
		require.True(t, ok, "server response writer must support flushing")

		_, _ = w.Write([]byte("id:100\nevent:ticker\ndata:{\"value\":1}\n\n"))
		flusher.Flush()
		_, _ = w.Write([]byte("id:101\nevent:ticker\ndata:{\"value\":2}\n\n"))
		flusher.Flush()
		_, _ = w.Write([]byte("id:102\nevent:ticker\ndata:{\"value\":3}\n\n"))
		flusher.Flush()

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

	expectedSSEBody := strings.Join([]string{
		"id:100",
		"event:ticker",
		"data:{\"value\":1}",
		"",
		"id:101",
		"event:ticker",
		"data:{\"value\":2}",
		"",
	}, "\n")

	tc := &models.TestCase{
		Name: "sse-match-and-close",
		HTTPReq: models.HTTPReq{
			Method: "GET",
			URL:    server.URL,
			Header: map[string]string{
				"Accept": "text/event-stream",
			},
		},
		HTTPResp: models.HTTPResp{
			Header: map[string]string{
				"Content-Type": "text/event-stream; charset=utf-8",
			},
			Body: expectedSSEBody,
			StreamBody: []models.HTTPStreamChunk{
				{
					Data: []models.HTTPStreamDataField{
						{Key: "id", Value: "100"},
						{Key: "event", Value: "ticker"},
						{Key: "data", Value: `{"value":1}`},
					},
				},
				{
					Data: []models.HTTPStreamDataField{
						{Key: "id", Value: "101"},