@@ -7,60 +7,10 @@ import (
77
88 "github.com/prometheus/prometheus/model/labels"
99 "github.com/prometheus/prometheus/promql"
10- "github.com/prometheus/prometheus/storage"
11- "github.com/prometheus/prometheus/tsdb/chunkenc"
1210 "github.com/stretchr/testify/assert"
1311 "github.com/stretchr/testify/require"
1412)
1513
16- type mockSeries struct {
17- lset labels.Labels
18- }
19-
20- func (s * mockSeries ) Labels () labels.Labels { return s .lset }
21- func (s * mockSeries ) Iterator (chunkenc.Iterator ) chunkenc.Iterator { return chunkenc .NewNopIterator () }
22-
23- func TestFilterSeriesSet (t * testing.T ) {
24- series := []storage.Series {
25- & mockSeries {lset : labels .FromStrings ("__name__" , "http_requests" , "job" , "api" , "env" , "prod" )},
26- & mockSeries {lset : labels .FromStrings ("__name__" , "http_requests" , "job" , "web" , "env" , "prod" )},
27- & mockSeries {lset : labels .FromStrings ("__name__" , "http_requests" , "job" , "api" , "env" , "dev" )},
28- }
29-
30- // Filter to only job="api"
31- filters := []* labels.Matcher {labels .MustNewMatcher (labels .MatchEqual , "job" , "api" )}
32- ss := newFilterSeriesSet (newSliceSeriesSet (series ), filters )
33-
34- var got []labels.Labels
35- for ss .Next () {
36- got = append (got , ss .At ().Labels ())
37- }
38- require .NoError (t , ss .Err ())
39- assert .Len (t , got , 2 )
40- assert .Equal (t , "api" , got [0 ].Get ("job" ))
41- assert .Equal (t , "api" , got [1 ].Get ("job" ))
42- }
43-
44- func TestFilterSeriesSet_RegexFilter (t * testing.T ) {
45- series := []storage.Series {
46- & mockSeries {lset : labels .FromStrings ("__name__" , "cpu" , "host" , "server-1" )},
47- & mockSeries {lset : labels .FromStrings ("__name__" , "cpu" , "host" , "server-2" )},
48- & mockSeries {lset : labels .FromStrings ("__name__" , "cpu" , "host" , "db-1" )},
49- }
50-
51- filters := []* labels.Matcher {labels .MustNewMatcher (labels .MatchRegexp , "host" , "server-.*" )}
52- ss := newFilterSeriesSet (newSliceSeriesSet (series ), filters )
53-
54- var got []labels.Labels
55- for ss .Next () {
56- got = append (got , ss .At ().Labels ())
57- }
58- require .NoError (t , ss .Err ())
59- assert .Len (t , got , 2 )
60- assert .Equal (t , "server-1" , got [0 ].Get ("host" ))
61- assert .Equal (t , "server-2" , got [1 ].Get ("host" ))
62- }
63-
6414func TestPrefetchCache_FindSuperset (t * testing.T ) {
6515 // Cache has a broad entry: __name__="http", job=~".*"
6616 cache := & prefetchCache {
@@ -202,12 +152,3 @@ func TestExecutePrefetch(t *testing.T) {
202152 require .Len (t , cache .entries , 1 )
203153 assert .Len (t , cache .entries [0 ].vector , 2 )
204154}
205-
206- func TestBuildSelectorExpr (t * testing.T ) {
207- ms := []* labels.Matcher {
208- labels .MustNewMatcher (labels .MatchEqual , "__name__" , "http" ),
209- labels .MustNewMatcher (labels .MatchRegexp , "job" , ".*" ),
210- }
211- got := buildSelectorExpr (ms )
212- assert .Equal (t , `{__name__="http",job=~".*"}` , got )
213- }
0 commit comments