waitgroup for test timing

Signed-off-by: Owen Diehl <[email protected]>
grafana · Oct 31, 2023 · c27710c · c27710c
1 parent 6e8c8f2
commit c27710c
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/pkg/storage/bloom/v1/fuse_test.go b/pkg/storage/bloom/v1/fuse_test.go
@@ -3,6 +3,7 @@ package v1
 import (
 	"bytes"
 	"context"
+	"sync"
 	"testing"
 
 	"github.com/grafana/dskit/concurrency"
@@ -61,6 +62,8 @@ func TestFusedQuerier(t *testing.T) {
 	}
 
 	resps := make([][]output, nReqs)
+	var g sync.WaitGroup
+	g.Add(1)
 	go func() {
 		require.Nil(t, concurrency.ForEachJob(
 			context.Background(),
@@ -73,10 +76,16 @@ func TestFusedQuerier(t *testing.T) {
 				return nil
 			},
 		))
+		g.Done()
 	}()
 
 	fused := querier.Fuse(itrs)
+
 	require.Nil(t, fused.Run())
+	for _, input := range inputs {
+		close(input[0].response)
+	}
+	g.Wait()
 
 	for i, input := range inputs {
 		for j, req := range input {