cmd/promtool: in tests use labels.FromStrings

Replacing code which assumes the internal structure of `Labels`.

Signed-off-by: Bryan Boreham <bjboreham@gmail.com>
This commit is contained in:
Bryan Boreham 2022-07-21 16:50:48 +00:00 committed by Julien Pivotto
parent 8297f5cb6b
commit c438b50133
2 changed files with 16 additions and 22 deletions

View file

@ -134,10 +134,7 @@ func TestBackfillRuleIntegration(t *testing.T) {
series := selectedSeries.At() series := selectedSeries.At()
if len(series.Labels()) != 3 { if len(series.Labels()) != 3 {
require.Equal(t, 2, len(series.Labels())) require.Equal(t, 2, len(series.Labels()))
x := labels.Labels{ x := labels.FromStrings("__name__", "grp2_rule1", "name1", "val1")
labels.Label{Name: "__name__", Value: "grp2_rule1"},
labels.Label{Name: "name1", Value: "val1"},
}
require.Equal(t, x, series.Labels()) require.Equal(t, x, series.Labels())
} else { } else {
require.Equal(t, 3, len(series.Labels())) require.Equal(t, 3, len(series.Labels()))
@ -259,10 +256,7 @@ func TestBackfillLabels(t *testing.T) {
selectedSeries := q.Select(false, nil, labels.MustNewMatcher(labels.MatchRegexp, "", ".*")) selectedSeries := q.Select(false, nil, labels.MustNewMatcher(labels.MatchRegexp, "", ".*"))
for selectedSeries.Next() { for selectedSeries.Next() {
series := selectedSeries.At() series := selectedSeries.At()
expectedLabels := labels.Labels{ expectedLabels := labels.FromStrings("__name__", "rulename", "name1", "value-from-rule")
labels.Label{Name: "__name__", Value: "rulename"},
labels.Label{Name: "name1", Value: "value-from-rule"},
}
require.Equal(t, expectedLabels, series.Labels()) require.Equal(t, expectedLabels, series.Labels())
} }
require.NoError(t, selectedSeries.Err()) require.NoError(t, selectedSeries.Err())

View file

@ -51,20 +51,20 @@ func TestSDCheckResult(t *testing.T) {
expectedSDCheckResult := []sdCheckResult{ expectedSDCheckResult := []sdCheckResult{
{ {
DiscoveredLabels: labels.Labels{ DiscoveredLabels: labels.FromStrings(
labels.Label{Name: "__address__", Value: "localhost:8080"}, "__address__", "localhost:8080",
labels.Label{Name: "__scrape_interval__", Value: "1m"}, "__scrape_interval__", "1m",
labels.Label{Name: "__scrape_timeout__", Value: "10s"}, "__scrape_timeout__", "10s",
labels.Label{Name: "foo", Value: "bar"}, "foo", "bar",
}, ),
Labels: labels.Labels{ Labels: labels.FromStrings(
labels.Label{Name: "__address__", Value: "localhost:8080"}, "__address__", "localhost:8080",
labels.Label{Name: "__scrape_interval__", Value: "1m"}, "__scrape_interval__", "1m",
labels.Label{Name: "__scrape_timeout__", Value: "10s"}, "__scrape_timeout__", "10s",
labels.Label{Name: "foo", Value: "bar"}, "foo", "bar",
labels.Label{Name: "instance", Value: "localhost:8080"}, "instance", "localhost:8080",
labels.Label{Name: "newfoo", Value: "bar"}, "newfoo", "bar",
}, ),
Error: nil, Error: nil,
}, },
} }