From c961aeaddb0e3193341e969b6fa8bd8765a255bc Mon Sep 17 00:00:00 2001
From: Judah Rand <17158624+judahrand@users.noreply.github.com>
Date: Tue, 18 Jan 2022 12:29:03 +0000
Subject: [PATCH] Wrap mocked function

---
 tests/test_full_table_interruption.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_full_table_interruption.py b/tests/test_full_table_interruption.py
index 60c8e409..61a8afcc 100644
--- 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