fix unit tests

yuchen-db · yuchen-db · commit 06e7126a93a6 · 2024-05-17T15:43:38.000-07:00
diff --git a/internal/cortex/querier/queryrange/query_bytes_fetched.go b/internal/cortex/querier/queryrange/query_bytes_fetched.go
@@ -32,9 +32,13 @@ func sumQueryBytesFetched(responses ...Response) uint64 {
 }
 
 func QueryBytesFetchedPrometheusResponseHeaders(responses ...Response) []*PrometheusResponseHeader {
+	n := sumQueryBytesFetched(responses...)
+	if n == 0 {
+		return nil
+	}
 	return []*PrometheusResponseHeader{{
 		Name:   QueryBytesFetchedHeaderName,
-		Values: []string{strconv.FormatUint(sumQueryBytesFetched(responses...), 10)},
+		Values: []string{strconv.FormatUint(n, 10)},
 	}}
 }
 

Original file line number	Diff line number	Diff line change
`@@ -32,9 +32,13 @@ func sumQueryBytesFetched(responses ...Response) uint64 {`
`32`	`32`	`}`
`33`	`33`
`34`	`34`	`func QueryBytesFetchedPrometheusResponseHeaders(responses ...Response) []*PrometheusResponseHeader {`
	`35`	`+ n := sumQueryBytesFetched(responses...)`
	`36`	`+ if n == 0 {`
	`37`	`+ return nil`
	`38`	`+ }`
`35`	`39`	`return []*PrometheusResponseHeader{{`
`36`	`40`	`Name: QueryBytesFetchedHeaderName,`
`37`		`- Values: []string{strconv.FormatUint(sumQueryBytesFetched(responses...), 10)},`
	`41`	`+ Values: []string{strconv.FormatUint(n, 10)},`
`38`	`42`	`}}`
`39`	`43`	`}`
`40`	`44`