Function TestAsyncResponseFansIn

modules/frontend/pipeline/responses_test.go:157–194

(t *testing.T)

}

func TestAsyncResponseFansIn(t *testing.T) {
	leakOpts := goleak.IgnoreCurrent()

	// create a random hierarchy of async responses and add a bunch of responses.
	// count the added responses and confirm the number we pull is the same.
	wg := sync.WaitGroup{}
	rootResp := newAsyncResponse()

	expected := 0
	wg.Add(1)
	go func() {
		defer wg.Done()
		defer rootResp.SendComplete()

		expected = addResponses(rootResp)
	}()

	actual := 0
	wg.Add(1)
	go func() {
		defer wg.Done()

		for {
			resp, done, err := rootResp.Next(context.Background())
			if done {
				return
			}
			actual++
			require.NoError(t, err)
			require.NotNil(t, resp)
		}
	}()

	wg.Wait()
	require.Equal(t, expected, actual)

	goleak.VerifyNone(t, leakOpts)
}

func addResponses(r *asyncResponse) int {
	responsesToAdd := rand.Intn(5)