Update mock

grafana · Aug 14, 2024 · 67ba83b · 67ba83b
1 parent b3a48a2
commit 67ba83b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tempodb/compactor_test.go b/tempodb/compactor_test.go
@@ -48,6 +48,7 @@ func (m *mockJobSharder) Owns(string) bool { return true }
 
 type mockOverrides struct {
 	blockRetention      time.Duration
+	disabled            bool
 	maxBytesPerTrace    int
 	maxCompactionWindow time.Duration
 }
@@ -56,6 +57,10 @@ func (m *mockOverrides) BlockRetentionForTenant(_ string) time.Duration {
 	return m.blockRetention
 }
 
+func (m *mockOverrides) DisabledForTenant(_ string) bool {
+	return m.disabled
+}
+
 func (m *mockOverrides) MaxBytesPerTraceForTenant(_ string) int {
 	return m.maxBytesPerTrace
 }