Update tests/test_salesforce_switch_rep_method_incrmntl_ft.py

Co-authored-by: bhtowles <[email protected]>
singer-io · Nov 14, 2023 · f2d7ac6 · f2d7ac6
1 parent eeed6f3
commit f2d7ac6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_salesforce_switch_rep_method_incrmntl_ft.py b/tests/test_salesforce_switch_rep_method_incrmntl_ft.py
@@ -73,7 +73,7 @@ def test_run(self):
                                        if record.get('action') == 'upsert']
                 incrmntl_primary_keys = {message[primary_key] for message in incrmntl_sync_messages}
                 fulltbl_sync_messages = [record['data'] for record in
-                                        fulltbl_sync_records.get(stream).get('messages')
+                                        fulltbl_sync_records.get(stream, {}).get('messages')
                                         if record.get('action') == 'upsert']
                 filtered_fulltbl_sync_messages = [message for message in fulltbl_sync_messages
                                                   if message[replication_key] >= self.start_date]