Function TestDoneChanCloses

modules/frontend/v1/request_batch_test.go:76–103

(_ *testing.T)

}

func TestDoneChanCloses(_ *testing.T) {
	rb := &requestBatch{}

	const totalRequests = 3

	ctx := context.Background()
	cancelCtx, cancel := context.WithCancel(ctx)

	req := httptest.NewRequest("GET", "http://example.com", nil)
	prequest := pipeline.NewHTTPRequest(req)
	prequest.SetContext(cancelCtx)

	for i := 0; i < totalRequests-1; i++ {
		_ = rb.add(&request{request: prequest})
	}

	wg := &sync.WaitGroup{}
	wg.Add(1)
	go func() {
		done := rb.doneChan(make(<-chan struct{}))
		<-done
		wg.Done()
	}()

	cancel()
	wg.Wait()
	// this test won't return unless doneChan closes
}

func TestDoneChanClosesOnStop(_ *testing.T) {
	rb := &requestBatch{}