mudhorn/prometheus
1
0
Fork 0
mirror of https://github.com/prometheus/prometheus.git synced 2025-03-05 20:59:13 -08:00
Code Issues Actions 4 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

86
tsdb/index/postings_test.go
View file

@ -973,37 +973,69 @@ func TestMemPostingsStats(t *testing.T) {
} }
func TestMemPostings_Delete(t *testing.T) { func TestMemPostings_Delete(t *testing.T) {
p := NewMemPostings() t.Run("some postings", func(t *testing.T) {
p.Add(1, labels.FromStrings("lbl1", "a")) p := NewMemPostings()
p.Add(2, labels.FromStrings("lbl1", "b")) p.Add(1, labels.FromStrings("lbl1", "a"))
p.Add(3, labels.FromStrings("lbl2", "a")) p.Add(2, labels.FromStrings("lbl1", "b"))
p.Add(3, labels.FromStrings("lbl2", "a"))
before := p.Get(allPostingsKey.Name, allPostingsKey.Value) before := p.Get(allPostingsKey.Name, allPostingsKey.Value)
deletedRefs := map[storage.SeriesRef]struct{}{ deletedRefs := map[storage.SeriesRef]struct{}{
2: {}, 2: {},
} }
affectedLabels := map[labels.Label]struct{}{ affectedLabels := map[labels.Label]struct{}{
{Name: "lbl1", Value: "b"}: {}, {Name: "lbl1", Value: "b"}: {},
} }
p.Delete(deletedRefs, affectedLabels) p.Delete(deletedRefs, affectedLabels)
after := p.Get(allPostingsKey.Name, allPostingsKey.Value) after := p.Get(allPostingsKey.Name, allPostingsKey.Value)
// Make sure postings gotten before the delete have the old data when // Make sure postings gotten before the delete have the old data when
// iterated over. // iterated over.
expanded, err := ExpandPostings(before) expanded, err := ExpandPostings(before)
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, []storage.SeriesRef{1, 2, 3}, expanded) require.Equal(t, []storage.SeriesRef{1, 2, 3}, expanded)
// Make sure postings gotten after the delete have the new data when // Make sure postings gotten after the delete have the new data when
// iterated over. // iterated over.
expanded, err = ExpandPostings(after) expanded, err = ExpandPostings(after)
require.NoError(t, err) require.NoError(t, err)
require.Equal(t, []storage.SeriesRef{1, 3}, expanded) require.Equal(t, []storage.SeriesRef{1, 3}, expanded)
deleted := p.Get("lbl1", "b") deleted := p.Get("lbl1", "b")
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