From c2373a81099194239230e5d1861521ab8660fa12 Mon Sep 17 00:00:00 2001
From: Aliaksandr Valialkin <valyala@victoriametrics.com>
Date: Tue, 30 Jan 2024 16:05:36 +0200
Subject: [PATCH] lib/promscrape: fix BenchmarkScrapeWorkScrapeInternal, which
 has been broken by the commit 65bc46032360b784181f5d3311e3e5f0535d0d48

---
 lib/promscrape/scrapework_timing_test.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/promscrape/scrapework_timing_test.go b/lib/promscrape/scrapework_timing_test.go
index bd83eb6ae1..b488a11de9 100644
--- a/lib/promscrape/scrapework_timing_test.go
+++ b/lib/promscrape/scrapework_timing_test.go
@@ -83,6 +83,7 @@ vm_tcplistener_write_calls_total{name="https", addr=":443"} 132356
 		sw.Config = &ScrapeWork{}
 		sw.ReadData = readDataFunc
 		sw.PushData = func(at *auth.Token, wr *prompbmarshal.WriteRequest) {}
+		tsmGlobal.Register(&sw)
 		timestamp := int64(0)
 		for pb.Next() {
 			if err := sw.scrapeInternal(timestamp, timestamp); err != nil {
@@ -90,5 +91,6 @@ vm_tcplistener_write_calls_total{name="https", addr=":443"} 132356
 			}
 			timestamp++
 		}
+		tsmGlobal.Unregister(&sw)
 	})
 }