@@ -144,7 +144,7 @@ async fn add_action_smoke_test() -> Result<(), Error> {
144144 ..Default :: default ( )
145145 } ;
146146 let store = RedisStore :: new_standard ( spec) . await . expect ( "Working spec" ) ;
147- fake_redis_backend. set_subscription_manager ( store. subscription_manager ( ) . unwrap ( ) ) ;
147+ fake_redis_backend. set_subscription_manager ( store. subscription_manager ( ) . await . unwrap ( ) ) ;
148148
149149 let notifier = Arc :: new ( Notify :: new ( ) ) ;
150150 let awaited_action_db = StoreAwaitedActionDb :: new (
@@ -153,6 +153,7 @@ async fn add_action_smoke_test() -> Result<(), Error> {
153153 MockInstantWrapped :: default,
154154 move || WORKER_OPERATION_ID . into ( ) ,
155155 )
156+ . await
156157 . unwrap ( ) ;
157158
158159 let mut subscription = awaited_action_db
@@ -249,7 +250,7 @@ async fn test_multiple_clients_subscribe_to_same_action() -> Result<(), Error> {
249250 ..Default :: default ( )
250251 } ;
251252 let store = RedisStore :: new_standard ( spec) . await . expect ( "Working spec" ) ;
252- fake_redis_backend. set_subscription_manager ( store. subscription_manager ( ) . unwrap ( ) ) ;
253+ fake_redis_backend. set_subscription_manager ( store. subscription_manager ( ) . await . unwrap ( ) ) ;
253254
254255 let notifier = Arc :: new ( Notify :: new ( ) ) ;
255256 let worker_operation_id = Arc :: new ( Mutex :: new ( WORKER_OPERATION_ID_1 ) ) ;
@@ -260,6 +261,7 @@ async fn test_multiple_clients_subscribe_to_same_action() -> Result<(), Error> {
260261 MockInstantWrapped :: default,
261262 move || worker_operation_id_clone. lock ( ) . clone ( ) . into ( ) ,
262263 )
264+ . await
263265 . unwrap ( ) ;
264266
265267 let task_change_notify = Arc :: new ( Notify :: new ( ) ) ;
@@ -415,6 +417,7 @@ async fn test_outdated_version() -> Result<(), Error> {
415417 MockInstantWrapped :: default,
416418 move || worker_operation_id_clone. lock ( ) . clone ( ) . into ( ) ,
417419 )
420+ . await
418421 . unwrap ( ) ;
419422
420423 let worker_awaited_action = make_awaited_action ( "WORKER_OPERATION_ID" ) ;
@@ -462,7 +465,7 @@ async fn test_orphaned_client_operation_id_returns_none() -> Result<(), Error> {
462465 ..Default :: default ( )
463466 } ;
464467 let store = RedisStore :: new_standard ( spec) . await . expect ( "Working spec" ) ;
465- fake_redis_backend. set_subscription_manager ( store. subscription_manager ( ) . unwrap ( ) ) ;
468+ fake_redis_backend. set_subscription_manager ( store. subscription_manager ( ) . await . unwrap ( ) ) ;
466469
467470 // Manually set up the orphaned state in the fake backend:
468471 // 1. Add client_id → operation_id mapping (cid_* key)
@@ -488,6 +491,7 @@ async fn test_orphaned_client_operation_id_returns_none() -> Result<(), Error> {
488491 MockInstantWrapped :: default,
489492 move || worker_operation_id_clone. lock ( ) . clone ( ) . into ( ) ,
490493 )
494+ . await
491495 . unwrap ( ) ;
492496
493497 // Try to get the awaited action by the client operation ID
0 commit comments