lock fix for flaky test (#2268)

grafana · Jun 26, 2020 · 48b8504 · 48b8504
1 parent 8161b92
commit 48b8504
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/querier/queryrange/querysharding_test.go b/pkg/querier/queryrange/querysharding_test.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"fmt"
 	"sort"
+	"sync"
 	"testing"
 	"time"
 
@@ -125,9 +126,12 @@ func Test_shardSplitter(t *testing.T) {
 }
 
 func Test_astMapper(t *testing.T) {
+	var lock sync.Mutex
 	called := 0
 
 	handler := queryrange.HandlerFunc(func(ctx context.Context, req queryrange.Request) (queryrange.Response, error) {
+		lock.Lock()
+		defer lock.Unlock()
 		resp := lokiResps[called]
 		called++
 		return resp, nil