RepoMirrors
/
prometheus
mirror of https://github.com/prometheus/prometheus
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

scrape: extend TestDroppedTargetsList to check counts 

Signed-off-by: Bryan Boreham <bjboreham@gmail.com>
This commit is contained in:
Bryan Boreham 2023-08-20 14:25:32 +01:00
parent 4d8e380269
commit 627c99424b
1 changed files with 11 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
scrape/scrape_test.go
View File

@ -88,27 +88,30 @@ func TestDroppedTargetsList(t *testing.T) {
SourceLabels: model.LabelNames{"job"}, SourceLabels: model.LabelNames{"job"},
}, },
}, },
KeepDroppedTargets: 1,
} }
tgs = []*targetgroup.Group{ tgs = []*targetgroup.Group{
{ {
Targets: []model.LabelSet{ Targets: []model.LabelSet{
{model.AddressLabel: "127.0.0.1:9090"}, {model.AddressLabel: "127.0.0.1:9090"},
{model.AddressLabel: "127.0.0.1:9091"},
}, },
}, },
} }
sp, _ = newScrapePool(cfg, app, 0, nil, &Options{}) sp, _ = newScrapePool(cfg, app, 0, nil, &Options{})
expectedLabelSetString = "{__address__=\"127.0.0.1:9090\", __scrape_interval__=\"0s\", __scrape_timeout__=\"0s\", job=\"dropMe\"}" expectedLabelSetString = "{__address__=\"127.0.0.1:9090\", __scrape_interval__=\"0s\", __scrape_timeout__=\"0s\", job=\"dropMe\"}"
expectedLength = 1 expectedLength = 2
) )
sp.Sync(tgs) sp.Sync(tgs)
sp.Sync(tgs) sp.Sync(tgs)
if len(sp.droppedTargets) != expectedLength { require.Equal(t, expectedLength, len(sp.droppedTargets))
t.Fatalf("Length of dropped targets exceeded expected length, expected %v, got %v", expectedLength, len(sp.droppedTargets)) require.Equal(t, expectedLength, sp.droppedTargetsCount)
} require.Equal(t, expectedLabelSetString, sp.droppedTargets[0].DiscoveredLabels().String())
if sp.droppedTargets[0].DiscoveredLabels().String() != expectedLabelSetString {
t.Fatalf("Got %v, expected %v", sp.droppedTargets[0].DiscoveredLabels().String(), expectedLabelSetString) // Check that count is still correct when we don't retain all dropped targets.
} sp.config.KeepDroppedTargets = 1
sp.Sync(tgs)
require.Equal(t, 1, len(sp.droppedTargets))
require.Equal(t, expectedLength, sp.droppedTargetsCount)
} }
// TestDiscoveredLabelsUpdate checks that DiscoveredLabels are updated // TestDiscoveredLabelsUpdate checks that DiscoveredLabels are updated