Function TestCacheFor

tempodb/backend/cache/cache_test.go:19–105

(t *testing.T)

)

func TestCacheFor(t *testing.T) {
	reader, _, err := NewCache(&BloomConfig{
		CacheMaxBlockAge:        time.Hour,
		CacheMinCompactionLevel: 1,
	}, nil, nil, test.NewMockProvider(), log.NewNopLogger())
	require.NoError(t, err)

	rw := reader.(*readerWriter)

	// test.MockProvider will return the same cache for all requests. we need
	// to override individual caches so that the test can validate the cache returned
	rw.footerCache = test.NewMockClient()
	rw.columnIdxCache = test.NewMockClient()
	rw.offsetIdxCache = test.NewMockClient()
	rw.traceIDIdxCache = test.NewMockClient()
	rw.bloomCache = test.NewMockClient()

	testCases := []struct {
		name          string
		cacheInfo     *backend.CacheInfo
		expectedCache cache.Cache
	}{
		// first three caches are unconditionally returned
		{
			name:          "footer is always returned",
			cacheInfo:     &backend.CacheInfo{Role: cache.RoleParquetFooter},
			expectedCache: rw.footerCache,
		},
		{
			name:          "col idx is always returned",
			cacheInfo:     &backend.CacheInfo{Role: cache.RoleParquetColumnIdx},
			expectedCache: rw.columnIdxCache,
		},
		{
			name:          "offset idx is always returned",
			cacheInfo:     &backend.CacheInfo{Role: cache.RoleParquetOffsetIdx},
			expectedCache: rw.offsetIdxCache,
		},
		{
			name:          "trace id idx is always returned",
			cacheInfo:     &backend.CacheInfo{Role: cache.RoleTraceIDIdx},
			expectedCache: rw.traceIDIdxCache,
		},
		// bloom cache is returned if the meta is valid given the bloom config
		{
			name: "bloom - no meta means no cache",
			cacheInfo: &backend.CacheInfo{
				Role: cache.RoleBloom,
			},
		},
		{
			name: "bloom - compaction lvl and start time valid",
			cacheInfo: &backend.CacheInfo{
				Role: cache.RoleBloom,
				Meta: &backend.BlockMeta{CompactionLevel: 1, StartTime: time.Now()},
			},
			expectedCache: rw.bloomCache,
		},