minor: Context refactor

elastic · Jul 12, 2023 · 1a953cf · 1a953cf
1 parent a7b9976
commit 1a953cf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/aggregators/aggregator_test.go b/aggregators/aggregator_test.go
@@ -1136,13 +1136,12 @@ func BenchmarkAggregateCombinedMetrics(b *testing.B) {
 	if err != nil {
 		b.Fatal(err)
 	}
+	ctx, cancel := context.WithCancel(context.Background())
+	b.Cleanup(func() { cancel() })
 	b.ResetTimer()
 	for i := 0; i < b.N; i++ {
 		for _, kv := range kvs {
-			if err := agg.AggregateCombinedMetrics(
-				context.Background(),
-				kv.Key, kv.Value,
-			); err != nil {
+			if err := agg.AggregateCombinedMetrics(ctx, kv.Key, kv.Value); err != nil {
 				b.Fatal(err)
 			}
 		}