Function TestResponsesPropagateUpstream

modules/frontend/v1/request_batch_test.go:158–196

(t *testing.T)

}

func TestResponsesPropagateUpstream(t *testing.T) {
	rb := &requestBatch{}

	const totalRequests = 3
	wg := &sync.WaitGroup{}

	for i := int32(0); i < totalRequests; i++ {
		responseChan := make(chan *http.Response)

		wg.Add(1)
		go func(expectedCode int32) {
			resp := <-responseChan
			assert.Equal(t, expectedCode, int32(resp.StatusCode))
			wg.Done()
		}(i)

		req := httptest.NewRequest("GET", "http://example.com", nil)
		_ = rb.add(&request{
			request:  pipeline.NewHTTPRequest(req),
			response: responseChan,
		})
	}

	// if the reported results mismatches the actual length we should error
	err := rb.reportResultsToPipeline(make([]*httpgrpc.HTTPResponse, totalRequests+1))
	require.Error(t, err)

	responses := make([]*httpgrpc.HTTPResponse, totalRequests)
	for i := int32(0); i < totalRequests; i++ {
		responses[i] = &httpgrpc.HTTPResponse{
			Code: i,
		}
	}
	err = rb.reportResultsToPipeline(responses)
	require.NoError(t, err)

	wg.Wait()
	// this test won't return unless all responseChan's receive a response
}