mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2024-11-14 17:44:06 -08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

Add some more tests for MemPostings_Delete

Browse source 
Signed-off-by: Oleg Zaytsev <mail@olegzaytsev.com>
This commit is contained in:
Oleg Zaytsev 2024-09-27 10:14:39 +02:00
parent 4fd2556baa
commit ada8a6ef10
No known key found for this signature in database
GPG key ID: 7E9FE9FD48F512EF
1 changed files with 59 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
tsdb/index/postings_test.go
View file

@ -973,6 +973,7 @@ func TestMemPostingsStats(t *testing.T) {
} }
func TestMemPostings_Delete(t *testing.T) { func TestMemPostings_Delete(t *testing.T) {
t.Run("some postings", func(t *testing.T) {
p := NewMemPostings() p := NewMemPostings()
p.Add(1, labels.FromStrings("lbl1", "a")) p.Add(1, labels.FromStrings("lbl1", "a"))
p.Add(2, labels.FromStrings("lbl1", "b")) p.Add(2, labels.FromStrings("lbl1", "b"))
@ -1004,6 +1005,37 @@ func TestMemPostings_Delete(t *testing.T) {
expanded, err = ExpandPostings(deleted) expanded, err = ExpandPostings(deleted)
require.NoError(t, err) require.NoError(t, err)
require.Empty(t, expanded, "expected empty postings, got %v", expanded) require.Empty(t, expanded, "expected empty postings, got %v", expanded)
})
t.Run("all postings", func(t *testing.T) {
p := NewMemPostings()
p.Add(1, labels.FromStrings("lbl1", "a"))
p.Add(2, labels.FromStrings("lbl1", "b"))
p.Add(3, labels.FromStrings("lbl2", "a"))
deletedRefs := map[storage.SeriesRef]struct{}{1: {}, 2: {}, 3: {}}
affectedLabels := map[labels.Label]struct{}{
{Name: "lbl1", Value: "a"}: {},
{Name: "lbl1", Value: "b"}: {},
{Name: "lbl1", Value: "c"}: {},
}
p.Delete(deletedRefs, affectedLabels)
after := p.Get(allPostingsKey.Name, allPostingsKey.Value)
expanded, err := ExpandPostings(after)
require.NoError(t, err)
require.Empty(t, expanded)
})
t.Run("nothing on empty mempostings", func(t *testing.T) {
p := NewMemPostings()
deletedRefs := map[storage.SeriesRef]struct{}{}
affectedLabels := map[labels.Label]struct{}{}
p.Delete(deletedRefs, affectedLabels)
after := p.Get(allPostingsKey.Name, allPostingsKey.Value)
expanded, err := ExpandPostings(after)
require.NoError(t, err)
require.Empty(t, expanded)
})
} }
// BenchmarkMemPostings_Delete is quite heavy, so consider running it with // BenchmarkMemPostings_Delete is quite heavy, so consider running it with