@@ -376,14 +376,26 @@ def mns_event_will_be_triggered_with_correct_data_for_created_event(context):
376376
377377@then ("MNS event will not be triggered for the event" )
378378def mns_event_will_not_be_triggered_for_the_event (context ):
379- message_body = read_message (context , queue_type = "notification" , action = "CREATE" )
379+ message_body = read_message (
380+ context ,
381+ queue_type = "notification" ,
382+ action = "CREATE" ,
383+ wait_time_seconds = 5 ,
384+ max_empty_polls = 1 ,
385+ )
380386 print ("No MNS create event is created" )
381387 assert message_body is None , "Not expected a message but queue returned a message"
382388
383389
384390@then ("MNS event will not be triggered for the update event" )
385391def validate_mns_event_not_triggered_for_updated_event (context ):
386- message_body = read_message (context , queue_type = "notification" , action = "UPDATE" )
392+ message_body = read_message (
393+ context ,
394+ queue_type = "notification" ,
395+ action = "UPDATE" ,
396+ wait_time_seconds = 5 ,
397+ max_empty_polls = 3 ,
398+ )
387399 print ("no MNS update event is created" )
388400 assert message_body is None , "Not expected a message but queue returned a message"
389401
@@ -483,7 +495,13 @@ def validate_sqs_message(context, message_body, action):
483495
484496def mns_event_will_be_triggered_with_correct_data_for_deleted_event (context ):
485497 if context .patient .identifier [0 ].value is None :
486- message_body = read_message (context , queue_type = "notification" , action = "DELETE" )
498+ message_body = read_message (
499+ context ,
500+ queue_type = "notification" ,
501+ action = "DELETE" ,
502+ wait_time_seconds = 5 ,
503+ max_empty_polls = 3 ,
504+ )
487505 print (
488506 "No MNS delete event is created as expected since NHS number is not present in the original immunization event"
489507 )
0 commit comments