fix unit test

Signed-off-by: Ben Ye <[email protected]>
thanos-io · Oct 5, 2023 · bcdd429 · bcdd429
1 parent ebea84f
commit bcdd429
Showing 1 changed file with 31 additions and 30 deletions.
diff --git a/pkg/store/cache/filter_cache_test.go b/pkg/store/cache/filter_cache_test.go
@@ -5,6 +5,7 @@ package storecache
 
 import (
 	"context"
+	"github.com/thanos-io/thanos/pkg/tenancy"
 	"testing"
 
 	"github.com/efficientgo/core/testutil"
@@ -46,19 +47,19 @@ func TestFilterCache(t *testing.T) {
 		{
 			name: "empty enabled items",
 			verifyFunc: func(t *testing.T, c IndexCache) {
-				c.StorePostings(blockID, postingKeys[0], testPostingData)
-				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData)
-				c.StoreSeries(blockID, 1, testSeriesData)
+				c.StorePostings(blockID, postingKeys[0], testPostingData, tenancy.DefaultTenant)
+				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData, tenancy.DefaultTenant)
+				c.StoreSeries(blockID, 1, testSeriesData, tenancy.DefaultTenant)
 
-				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys)
+				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys, tenancy.DefaultTenant)
 				testutil.Equals(t, 0, len(missed))
 				testutil.Equals(t, testPostingData, hits[postingKeys[0]])
 
-				ep, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers)
+				ep, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers, tenancy.DefaultTenant)
 				testutil.Equals(t, true, hit)
 				testutil.Equals(t, testExpandedPostingsData, ep)
 
-				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1})
+				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1}, tenancy.DefaultTenant)
 				testutil.Equals(t, 0, len(misses))
 				testutil.Equals(t, testSeriesData, seriesHit[1])
 			},
@@ -67,19 +68,19 @@ func TestFilterCache(t *testing.T) {
 			name:         "all enabled items",
 			enabledItems: []string{cacheTypeSeries, cacheTypePostings, cacheTypeExpandedPostings},
 			verifyFunc: func(t *testing.T, c IndexCache) {
-				c.StorePostings(blockID, postingKeys[0], testPostingData)
-				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData)
-				c.StoreSeries(blockID, 1, testSeriesData)
+				c.StorePostings(blockID, postingKeys[0], testPostingData, tenancy.DefaultTenant)
+				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData, tenancy.DefaultTenant)
+				c.StoreSeries(blockID, 1, testSeriesData, tenancy.DefaultTenant)
 
-				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys)
+				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys, tenancy.DefaultTenant)
 				testutil.Equals(t, 0, len(missed))
 				testutil.Equals(t, testPostingData, hits[postingKeys[0]])
 
-				ep, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers)
+				ep, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers, tenancy.DefaultTenant)
 				testutil.Assert(t, true, hit)
 				testutil.Equals(t, testExpandedPostingsData, ep)
 
-				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1})
+				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1}, tenancy.DefaultTenant)
 				testutil.Equals(t, 0, len(misses))
 				testutil.Equals(t, testSeriesData, seriesHit[1])
 			},
@@ -88,18 +89,18 @@ func TestFilterCache(t *testing.T) {
 			name:         "only enable postings",
 			enabledItems: []string{cacheTypePostings},
 			verifyFunc: func(t *testing.T, c IndexCache) {
-				c.StorePostings(blockID, postingKeys[0], testPostingData)
-				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData)
-				c.StoreSeries(blockID, 1, testSeriesData)
+				c.StorePostings(blockID, postingKeys[0], testPostingData, tenancy.DefaultTenant)
+				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData, tenancy.DefaultTenant)
+				c.StoreSeries(blockID, 1, testSeriesData, tenancy.DefaultTenant)
 
-				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys)
+				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys, tenancy.DefaultTenant)
 				testutil.Equals(t, 0, len(missed))
 				testutil.Equals(t, testPostingData, hits[postingKeys[0]])
 
-				_, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers)
+				_, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers, tenancy.DefaultTenant)
 				testutil.Equals(t, false, hit)
 
-				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1})
+				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1}, tenancy.DefaultTenant)
 				testutil.Equals(t, 1, len(misses))
 				testutil.Equals(t, 0, len(seriesHit))
 			},
@@ -108,19 +109,19 @@ func TestFilterCache(t *testing.T) {
 			name:         "only enable expanded postings",
 			enabledItems: []string{cacheTypeExpandedPostings},
 			verifyFunc: func(t *testing.T, c IndexCache) {
-				c.StorePostings(blockID, postingKeys[0], testPostingData)
-				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData)
-				c.StoreSeries(blockID, 1, testSeriesData)
+				c.StorePostings(blockID, postingKeys[0], testPostingData, tenancy.DefaultTenant)
+				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData, tenancy.DefaultTenant)
+				c.StoreSeries(blockID, 1, testSeriesData, tenancy.DefaultTenant)
 
-				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys)
+				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys, tenancy.DefaultTenant)
 				testutil.Equals(t, 1, len(missed))
 				testutil.Equals(t, 0, len(hits))
 
-				ep, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers)
+				ep, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers, tenancy.DefaultTenant)
 				testutil.Equals(t, true, hit)
 				testutil.Equals(t, testExpandedPostingsData, ep)
 
-				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1})
+				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1}, tenancy.DefaultTenant)
 				testutil.Equals(t, 1, len(misses))
 				testutil.Equals(t, 0, len(seriesHit))
 			},
@@ -129,18 +130,18 @@ func TestFilterCache(t *testing.T) {
 			name:         "only enable series",
 			enabledItems: []string{cacheTypeSeries},
 			verifyFunc: func(t *testing.T, c IndexCache) {
-				c.StorePostings(blockID, postingKeys[0], testPostingData)
-				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData)
-				c.StoreSeries(blockID, 1, testSeriesData)
+				c.StorePostings(blockID, postingKeys[0], testPostingData, tenancy.DefaultTenant)
+				c.StoreExpandedPostings(blockID, expandedPostingsMatchers, testExpandedPostingsData, tenancy.DefaultTenant)
+				c.StoreSeries(blockID, 1, testSeriesData, tenancy.DefaultTenant)
 
-				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys)
+				hits, missed := c.FetchMultiPostings(ctx, blockID, postingKeys, tenancy.DefaultTenant)
 				testutil.Equals(t, 1, len(missed))
 				testutil.Equals(t, 0, len(hits))
 
-				_, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers)
+				_, hit := c.FetchExpandedPostings(ctx, blockID, expandedPostingsMatchers, tenancy.DefaultTenant)
 				testutil.Equals(t, false, hit)
 
-				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1})
+				seriesHit, misses := c.FetchMultiSeries(ctx, blockID, []storage.SeriesRef{1}, tenancy.DefaultTenant)
 				testutil.Equals(t, 0, len(misses))
 				testutil.Equals(t, testSeriesData, seriesHit[1])
 			},