Fixes query-tee tests.

Signed-off-by: Cyril Tovena <cyril.tovena@gmail.com>
grafana · Jan 12, 2021 · 1e6544e · 1e6544e
1 parent 7385735
commit 1e6544e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cmd/querytee/response_comparator.go b/cmd/querytee/response_comparator.go
@@ -6,18 +6,19 @@ import (
 
 	"github.com/cortexproject/cortex/pkg/util"
 	"github.com/go-kit/kit/log/level"
+	jsoniter "github.com/json-iterator/go"
 
 	"github.com/grafana/loki/pkg/loghttp"
 )
 
 func compareStreams(expectedRaw, actualRaw json.RawMessage, tolerance float64) error {
 	var expected, actual loghttp.Streams
 
-	err := json.Unmarshal(expectedRaw, &expected)
+	err := jsoniter.Unmarshal(expectedRaw, &expected)
 	if err != nil {
 		return err
 	}
-	err = json.Unmarshal(actualRaw, &actual)
+	err = jsoniter.Unmarshal(actualRaw, &actual)
 	if err != nil {
 		return err
 	}