-
Notifications
You must be signed in to change notification settings - Fork 524
/
metrics_test.go
126 lines (111 loc) · 5.78 KB
/
metrics_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
// SPDX-License-Identifier: AGPL-3.0-only
package test
import (
"testing"
dto "github.com/prometheus/client_model/go"
"github.com/stretchr/testify/require"
"google.golang.org/protobuf/proto"
)
func TestAssertGatherAndCompare(t *testing.T) {
g := fakeGatherer{
metrics: []*dto.MetricFamily{
{
Name: proto.String("cortex_distributor_deduped_samples_total"),
Help: proto.String("The total number of deduplicated samples."),
Metric: []*dto.Metric{
{
Label: []*dto.LabelPair{
{
Name: proto.String("environment"),
Value: proto.String("test"),
},
},
TimestampMs: proto.Int64(1000),
Counter: &dto.Counter{
Value: proto.Float64(1),
},
},
},
},
},
}
t.Run("don't specify any metrics", func(t *testing.T) {
// When not specifying any metrics, an error should be returned due to missing metric
// cortex_distributor_latest_seen_sample_timestamp_seconds.
err := gatherAndCompare(g, `
# HELP cortex_distributor_deduped_samples_total The total number of deduplicated samples.
# TYPE cortex_distributor_deduped_samples_total counter
cortex_distributor_deduped_samples_total{environment="test"} 1 1000
# HELP cortex_distributor_latest_seen_sample_timestamp_seconds Unix timestamp of latest received sample per user.
# TYPE cortex_distributor_latest_seen_sample_timestamp_seconds gauge
cortex_distributor_latest_seen_sample_timestamp_seconds{user="userA"} 1111
`)
require.EqualError(t, err, ` # HELP cortex_distributor_deduped_samples_total The total number of deduplicated samples.
# TYPE cortex_distributor_deduped_samples_total counter
cortex_distributor_deduped_samples_total{environment="test"} 1 1000
+# HELP cortex_distributor_latest_seen_sample_timestamp_seconds Unix timestamp of latest received sample per user.
+# TYPE cortex_distributor_latest_seen_sample_timestamp_seconds gauge
+cortex_distributor_latest_seen_sample_timestamp_seconds{user="userA"} 1111
`)
})
t.Run("specify required metric", func(t *testing.T) {
// When specifying that cortex_distributor_deduped_samples_total as the one required metric,
// cortex_distributor_latest_seen_sample_timestamp_seconds should be ignored even if it's missing.
AssertGatherAndCompare(t, g, `
# HELP cortex_distributor_deduped_samples_total The total number of deduplicated samples.
# TYPE cortex_distributor_deduped_samples_total counter
cortex_distributor_deduped_samples_total{environment="test"} 1 1000
# HELP cortex_distributor_latest_seen_sample_timestamp_seconds Unix timestamp of latest received sample per user.
# TYPE cortex_distributor_latest_seen_sample_timestamp_seconds gauge
cortex_distributor_latest_seen_sample_timestamp_seconds{user="userA"} 1111
`, "cortex_distributor_deduped_samples_total")
})
t.Run("specify required metric which isn't there", func(t *testing.T) {
// When specifying that cortex_distributor_deduped_samples_total as the one required metric,
// cortex_distributor_latest_seen_sample_timestamp_seconds should be ignored even if it's missing.
err := gatherAndCompare(g, `
# HELP cortex_distributor_deduped_samples_total The total number of deduplicated samples.
# TYPE cortex_distributor_deduped_samples_total counter
cortex_distributor_deduped_samples_total{environment="test"} 1 1000
# HELP cortex_distributor_latest_seen_sample_timestamp_seconds Unix timestamp of latest received sample per user.
# TYPE cortex_distributor_latest_seen_sample_timestamp_seconds gauge
cortex_distributor_latest_seen_sample_timestamp_seconds{user="userA"} 1111
`, "cortex_distributor_latest_seen_sample_timestamp_seconds")
require.EqualError(t, err, "expected metric name(s) not found: [cortex_distributor_latest_seen_sample_timestamp_seconds]")
})
t.Run("specify required metric and absent metric", func(t *testing.T) {
// Verify that cortex_distributor_deduped_samples_total is found among metrics returned by g,
// and that conversely, cortex_distributor_non_ha_samples_received_total is not found among
// metrics returned by g.
// cortex_distributor_latest_seen_sample_timestamp_seconds is ignored, since it's not among
// the specified metrics.
AssertGatherAndCompare(t, g, `
# HELP cortex_distributor_deduped_samples_total The total number of deduplicated samples.
# TYPE cortex_distributor_deduped_samples_total counter
cortex_distributor_deduped_samples_total{environment="test"} 1 1000
# HELP cortex_distributor_latest_seen_sample_timestamp_seconds Unix timestamp of latest received sample per user.
# TYPE cortex_distributor_latest_seen_sample_timestamp_seconds gauge
cortex_distributor_latest_seen_sample_timestamp_seconds{user="userA"} 1111
`, "cortex_distributor_deduped_samples_total", "cortex_distributor_non_ha_samples_received_total")
})
t.Run("specify absent metric which is actually there", func(t *testing.T) {
// Verify that cortex_distributor_deduped_samples_total is found among metrics returned by g,
// and that conversely, cortex_distributor_non_ha_samples_received_total is not found among
// metrics returned by g.
// cortex_distributor_latest_seen_sample_timestamp_seconds is ignored, since it's not among
// the specified metrics.
err := gatherAndCompare(g, `
# HELP cortex_distributor_latest_seen_sample_timestamp_seconds Unix timestamp of latest received sample per user.
# TYPE cortex_distributor_latest_seen_sample_timestamp_seconds gauge
cortex_distributor_latest_seen_sample_timestamp_seconds{user="userA"} 1111
`, "cortex_distributor_deduped_samples_total", "cortex_distributor_non_ha_samples_received_total")
require.EqualError(t, err, "should be absent: metrics=cortex_distributor_deduped_samples_total")
})
}
type fakeGatherer struct {
metrics []*dto.MetricFamily
err error
}
func (g fakeGatherer) Gather() ([]*dto.MetricFamily, error) {
return g.metrics, g.err
}