Enhance TestDropOldTimeSeries to test both v1 and v2 protocols (#16709)

- Wrapped existing test logic in a loop to run with both protocol versions
- Ensures consistent behavior across protocol versions for dropping old time series

Signed-off-by: AxcelXander <tyz666@bu.edu>
Co-authored-by: AxcelXander <tyz666@bu.edu>

storage/remote/queue_manager_test.go

@@ -1931,27 +1931,31 @@ func BenchmarkBuildV2WriteRequest(b *testing.B) {
 }
 
 func TestDropOldTimeSeries(t *testing.T) {
-	size := 10
+	// Test both v1 and v2 remote write protocols
-	nSeries := 6
+	for _, protoMsg := range []config.RemoteWriteProtoMsg{config.RemoteWriteProtoMsgV1, config.RemoteWriteProtoMsgV2} {
-	nSamples := config.DefaultQueueConfig.Capacity * size
+		t.Run(fmt.Sprint(protoMsg), func(t *testing.T) {
-	samples, newSamples, series := createTimeseriesWithOldSamples(nSamples, nSeries)
+			size := 10
+			nSeries := 6
+			nSamples := config.DefaultQueueConfig.Capacity * size
+			samples, newSamples, series := createTimeseriesWithOldSamples(nSamples, nSeries)
 
-	// TODO(alexg): test with new version
+			c := NewTestWriteClient(protoMsg)
-	c := NewTestWriteClient(config.RemoteWriteProtoMsgV1)
+			c.expectSamples(newSamples, series)
-	c.expectSamples(newSamples, series)
 
-	cfg := config.DefaultQueueConfig
+			cfg := config.DefaultQueueConfig
-	mcfg := config.DefaultMetadataConfig
+			mcfg := config.DefaultMetadataConfig
-	cfg.MaxShards = 1
+			cfg.MaxShards = 1
-	cfg.SampleAgeLimit = model.Duration(60 * time.Second)
+			cfg.SampleAgeLimit = model.Duration(60 * time.Second)
-	m := newTestQueueManager(t, cfg, mcfg, defaultFlushDeadline, c, config.RemoteWriteProtoMsgV1)
+			m := newTestQueueManager(t, cfg, mcfg, defaultFlushDeadline, c, protoMsg)
-	m.StoreSeries(series, 0)
+			m.StoreSeries(series, 0)
 
-	m.Start()
+			m.Start()
-	defer m.Stop()
+			defer m.Stop()
 
-	m.Append(samples)
+			m.Append(samples)
-	c.waitForExpectedData(t, 30*time.Second)
+			c.waitForExpectedData(t, 30*time.Second)
+		})
+	}
 }
 
 func TestIsSampleOld(t *testing.T) {