mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2024-09-20 07:47:31 -07:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Add test for mergedTombstoneReader

Browse source 
Signed-off-by: Goutham Veeramachaneni <cs14btech11014@iith.ac.in>
This commit is contained in:
Goutham Veeramachaneni 2017-05-22 10:31:50 +05:30
parent 0b70333ef6
commit 009dd2cde5
No known key found for this signature in database
GPG key ID: F1C217E8E9023CAD
2 changed files with 118 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tombstones.go
View file

@ -377,11 +377,12 @@ func (t *mergedTombstoneReader) Next() bool {
t.cur = bcur
t.bok = t.b.Next()
} else {
t.cur = acur
// Merge time ranges.
for _, r := range bcur.ranges {
acur.ranges = addNewInterval(acur.ranges, r)
}
t.cur = acur
t.aok = t.a.Next()
t.bok = t.b.Next()
}

116
tombstones_test.go
View file

@ -63,6 +63,11 @@ func TestAddingNewIntervals(t *testing.T) {
new: trange{21, 23},
exp: []trange{{1, 10}, {12, 20}, {21, 23}, {25, 30}},
},
{
exist: []trange{{1, 2}, {3, 5}, {6, 7}},
new: trange{6, 7},
exp: []trange{{1, 2}, {3, 5}, {6, 7}},
},
{
exist: []trange{{1, 10}, {12, 20}, {25, 30}},
new: trange{21, 25},
@ -199,3 +204,114 @@ func TestTombstoneReadersSeek(t *testing.T) {
return
})
}
func TestMergedTombstoneReader(t *testing.T) {
cases := []struct {
a, b TombstoneReader
exp TombstoneReader
}{
{
a: newMapTombstoneReader(
map[uint32][]trange{
2: []trange{{1, 2}},
3: []trange{{1, 4}, {5, 6}},
4: []trange{{10, 15}, {16, 20}},
5: []trange{{1, 4}, {5, 6}},
50: []trange{{10, 20}, {35, 50}},
600: []trange{{100, 2000}},
1000: []trange{},
1500: []trange{{10000, 500000}},
1600: []trange{{1, 2}, {3, 4}, {4, 5}, {6, 7}},
},
),
b: newMapTombstoneReader(
map[uint32][]trange{
2: []trange{{1, 2}},
3: []trange{{1, 4}, {5, 6}},
4: []trange{{10, 15}, {16, 20}},
5: []trange{{1, 4}, {5, 6}},
50: []trange{{10, 20}, {35, 50}},
600: []trange{{100, 2000}},
1000: []trange{},
1500: []trange{{10000, 500000}},
1600: []trange{{1, 2}, {3, 4}, {4, 5}, {6, 7}},
},
),
exp: newMapTombstoneReader(
map[uint32][]trange{
2: []trange{{1, 2}},
3: []trange{{1, 4}, {5, 6}},
4: []trange{{10, 15}, {16, 20}},
5: []trange{{1, 4}, {5, 6}},
50: []trange{{10, 20}, {35, 50}},
600: []trange{{100, 2000}},
1000: []trange{},
1500: []trange{{10000, 500000}},
1600: []trange{{1, 2}, {3, 5}, {6, 7}},
},
),
},
{
a: newMapTombstoneReader(
map[uint32][]trange{
2: []trange{{1, 2}},
3: []trange{{1, 4}, {5, 6}},
4: []trange{{10, 15}, {16, 20}},
5: []trange{{1, 4}, {5, 6}},
50: []trange{{10, 20}, {35, 50}},
600: []trange{{100, 2000}},
1000: []trange{},
1500: []trange{{10000, 500000}},
1600: []trange{{1, 2}, {3, 4}, {4, 5}, {6, 7}},
},
),
b: newMapTombstoneReader(
map[uint32][]trange{
20: []trange{{1, 2}},
30: []trange{{1, 4}, {5, 6}},
40: []trange{{10, 15}, {16, 20}},
60: []trange{{1, 4}, {5, 6}},
500: []trange{{10, 20}, {35, 50}},
6000: []trange{{100, 2000}},
10000: []trange{},
15000: []trange{{10000, 500000}},
1600: []trange{{1, 2}, {3, 4}, {4, 5}, {6, 7}},
},
),
exp: newMapTombstoneReader(
map[uint32][]trange{
2: []trange{{1, 2}},
3: []trange{{1, 4}, {5, 6}},
4: []trange{{10, 15}, {16, 20}},
5: []trange{{1, 4}, {5, 6}},
50: []trange{{10, 20}, {35, 50}},
600: []trange{{100, 2000}},
1000: []trange{},
1500: []trange{{10000, 500000}},
20: []trange{{1, 2}},
30: []trange{{1, 4}, {5, 6}},
40: []trange{{10, 15}, {16, 20}},
60: []trange{{1, 4}, {5, 6}},
500: []trange{{10, 20}, {35, 50}},
6000: []trange{{100, 2000}},
10000: []trange{},
15000: []trange{{10000, 500000}},
1600: []trange{{1, 2}, {3, 5}, {6, 7}},
},
),
},
}
for _, c := range cases {
res := newMergedTombstoneReader(c.a, c.b)
for c.exp.Next() {
require.True(t, res.Next())
require.Equal(t, c.exp.At(), res.At())
}
require.False(t, res.Next())
}
return
}