@@ -210,11 +210,8 @@ func Test_TenantFederationRegexResolver_WhenSingleTenantMatched(t *testing.T) {
210210 require .NoError (t , err )
211211 require .Equal (t , 200 , res .StatusCode )
212212
213- // wait to upload blocks
214- require .NoError (t , ingester .WaitSumMetricsWithOptions (e2e .Greater (0 ), []string {"cortex_ingester_shipper_uploads_total" }, e2e .WaitMissingMetrics ))
215-
216213 // wait to update knownUsers
217- require .NoError (t , querier .WaitSumMetricsWithOptions (e2e .Greater ( 0 ), [] string { "cortex_regex_resolver_last_update_run_timestamp_seconds" }, e2e . WaitMissingMetrics ))
214+ require .NoError (t , querier .WaitSumMetrics (e2e .Equals ( 1 ), "cortex_regex_resolver_discovered_users" ))
218215
219216 clientForMatchOneTenant , err = e2ecortex .NewClient (distributor .HTTPEndpoint (), queryFrontend .HTTPEndpoint (), "" , "" , "user-.+" )
220217 require .NoError (t , err )
@@ -377,13 +374,10 @@ func runQuerierTenantFederationTest_UseRegexResolver(t *testing.T, cfg querierTe
377374 require .NoError (t , querier2 .WaitSumMetrics (e2e .Equals (512 * 2 ), "cortex_ring_tokens_total" ))
378375 }
379376
380- // wait to upload blocks
381- require .NoError (t , ingester .WaitSumMetricsWithOptions (e2e .Greater (0 ), []string {"cortex_ingester_shipper_uploads_total" }, e2e .WaitMissingMetrics ))
382-
383377 // wait to update knownUsers
384- require .NoError (t , querier .WaitSumMetricsWithOptions (e2e .Greater ( 0 ), [] string { "cortex_regex_resolver_last_update_run_timestamp_seconds" }, e2e . WaitMissingMetrics ))
378+ require .NoError (t , querier .WaitSumMetrics (e2e .Equals ( float64 ( numUsers )), "cortex_regex_resolver_discovered_users" ))
385379 if cfg .shuffleShardingEnabled {
386- require .NoError (t , querier2 .WaitSumMetricsWithOptions (e2e .Greater ( 0 ), [] string { "cortex_regex_resolver_last_update_run_timestamp_seconds" }, e2e . WaitMissingMetrics ))
380+ require .NoError (t , querier2 .WaitSumMetrics (e2e .Equals ( float64 ( numUsers )), "cortex_regex_resolver_discovered_users" ))
387381 }
388382
389383 // query all tenants
0 commit comments