Wrap mocked function

transferwise · Jan 18, 2022 · c961aea · c961aea
1 parent 45a9a9d
commit c961aea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_full_table_interruption.py b/tests/test_full_table_interruption.py
@@ -63,7 +63,7 @@ def setUp(self):
         global CAUGHT_MESSAGES
         CAUGHT_MESSAGES.clear()
 
-    @mock.patch('tap_postgres.sync_logical_streams')
+    @mock.patch('tap_postgres.sync_logical_streams', wraps=tap_postgres.sync_logical_streams)
     def test_catalog(self, mock_sync_logical_streams):
         singer.write_message = singer_write_message_no_cow
         pg_common.write_schema_message = singer_write_message_ok