mirrors/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-08-06 14:17:12 +02:00
Code Issues Projects Releases Wiki Activity

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

Browse Source 
- 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>
This commit is contained in:
AxcelXander 2025-06-11 13:06:45 -07:00 committed by GitHub
parent 94d5e0f41c
commit 472f0de661
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 21 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
storage/remote/queue_manager_test.go
View File

@ -1931,27 +1931,31 @@ func BenchmarkBuildV2WriteRequest(b *testing.B) {
} }
func TestDropOldTimeSeries(t *testing.T) { 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) { func TestIsSampleOld(t *testing.T) {