fix test

Signed-off-by: alanprot <[email protected]>

Author: alanprot

pkg/compactor/shuffle_sharding_grouper_test.go

@@ -747,6 +747,10 @@ type RingMock struct {
 	mock.Mock
 }
 
+func (r *RingMock) GetInstanceIdByAddr(addr string) (string, error) {
+	return "", nil
+}
+
 func (r *RingMock) Collect(ch chan<- prometheus.Metric) {}
 
 func (r *RingMock) Describe(ch chan<- *prometheus.Desc) {}

pkg/distributor/query.go

@@ -2,11 +2,11 @@ package distributor
 
 import (
 	"context"
-	"github.com/go-kit/log/level"
 	"io"
 	"sort"
 	"time"
 
+	"github.com/go-kit/log/level"
 	"github.com/opentracing/opentracing-go"
 	"github.com/prometheus/common/model"
 	"github.com/prometheus/prometheus/model/labels"
@@ -166,12 +166,13 @@ func (d *Distributor) queryIngestersExemplars(ctx context.Context, replicationSe
 			return nil, err
 		}
 
-		resp, err := client.(ingester_client.IngesterClient).QueryExemplars(ctx, req)
 		ingesterId, err := d.ingestersRing.GetInstanceIdByAddr(ing.Addr)
 		if err != nil {
 			level.Warn(d.log).Log("msg", "instance not found in the ring", "addr", ing.Addr, "err", err)
 		}
 
+		resp, err := client.(ingester_client.IngesterClient).QueryExemplars(ctx, req)
+
 		d.ingesterQueries.WithLabelValues(ingesterId).Inc()
 		if err != nil {
 			d.ingesterQueryFailures.WithLabelValues(ingesterId).Inc()

pkg/ring/util_test.go

@@ -17,6 +17,10 @@ type RingMock struct {
 	mock.Mock
 }
 
+func (r *RingMock) GetInstanceIdByAddr(addr string) (string, error) {
+	return "", nil
+}
+
 func (r *RingMock) Collect(ch chan<- prometheus.Metric) {}
 
 func (r *RingMock) Describe(ch chan<- *prometheus.Desc) {}