@@ -80,7 +80,8 @@ def test_audit_table_updated_correctly(self):
8080 update_ack_file .upload_ack_file (
8181 temp_ack_file_key = f"TempAck/{ file_key } " ,
8282 message_id = message_id ,
83- supplier_queue = "queue-audit-table" ,
83+ supplier = "queue-audit-table-supplier" ,
84+ vaccine_type = "vaccine-type" ,
8485 accumulated_csv_content = accumulated_csv_content ,
8586 ack_data_rows = ack_data_rows ,
8687 archive_ack_file_key = f"forwardedFile/{ file_key } " ,
@@ -122,7 +123,9 @@ def test_next_queued_file_triggers_lambda(self):
122123 self .mock_get_row_count .side_effect = [3 , 3 ]
123124 next_file = "next_for_lambda.csv"
124125 next_message_id = "msg-next-lambda"
125- queue_name = "queue-lambda-trigger"
126+ supplier = "lambda-trigger-supplier"
127+ vaccine_type = "vaccine-type"
128+ queue_name = f"{ supplier } _{ vaccine_type } "
126129 self .mock_get_next_queued_file_details .return_value = {"filename" : next_file , "message_id" : next_message_id }
127130 accumulated_csv_content = StringIO ("header1|header2\n " )
128131 ack_data_rows = [
@@ -140,7 +143,8 @@ def test_next_queued_file_triggers_lambda(self):
140143 update_ack_file .upload_ack_file (
141144 temp_ack_file_key = f"TempAck/{ next_key } " ,
142145 message_id = "msg-lambda-trigger" ,
143- supplier_queue = queue_name ,
146+ supplier = supplier ,
147+ vaccine_type = vaccine_type ,
144148 accumulated_csv_content = accumulated_csv_content ,
145149 ack_data_rows = ack_data_rows ,
146150 archive_ack_file_key = f"forwardedFile/{ next_key } " ,
0 commit comments