Fix tests

Matatika · Nov 12, 2024 · b5ff4b6 · b5ff4b6
1 parent 5c56103
commit b5ff4b6
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 9 deletions.
diff --git a/tap_google_sheets/tests/test_child_sheet_name_setting.py b/tap_google_sheets/tests/test_child_sheet_name_setting.py
@@ -4,7 +4,6 @@
 
 import responses
 import singer_sdk._singerlib as singer
-import singer_sdk.io_base as io
 
 import tap_google_sheets.tests.utils as test_utils
 from tap_google_sheets.tap import TapGoogleSheets
@@ -27,7 +26,7 @@ def setUp(self):
         responses.reset()
         del test_utils.SINGER_MESSAGES[:]
 
-        io.singer_write_message = test_utils.accumulate_singer_messages
+        TapGoogleSheets.write_message = test_utils.accumulate_singer_messages
 
     @responses.activate()
     def test_discovered_stream_name(self):

diff --git a/tap_google_sheets/tests/test_discovered_stream_name.py b/tap_google_sheets/tests/test_discovered_stream_name.py
@@ -18,7 +18,7 @@ def setUp(self):
         responses.reset()
         del test_utils.SINGER_MESSAGES[:]
 
-        io.singer_write_message = test_utils.accumulate_singer_messages
+        TapGoogleSheets.write_message = test_utils.accumulate_singer_messages
 
     @responses.activate()
     def test_discovered_stream_name(self):

diff --git a/tap_google_sheets/tests/test_ignoring_unnamed_columns.py b/tap_google_sheets/tests/test_ignoring_unnamed_columns.py
@@ -4,7 +4,6 @@
 
 import responses
 import singer_sdk._singerlib as singer
-import singer_sdk.io_base as io
 
 import tap_google_sheets.tests.utils as test_utils
 from tap_google_sheets.tap import TapGoogleSheets
@@ -19,7 +18,7 @@ def setUp(self):
         responses.reset()
         del test_utils.SINGER_MESSAGES[:]
 
-        io.singer_write_message = test_utils.accumulate_singer_messages
+        TapGoogleSheets.write_message = test_utils.accumulate_singer_messages
 
     @responses.activate()
     def test_ignoring_unnamed_columns(self):

diff --git a/tap_google_sheets/tests/test_table_name_setting.py b/tap_google_sheets/tests/test_table_name_setting.py
@@ -19,7 +19,7 @@ def setUp(self):
         responses.reset()
         del test_utils.SINGER_MESSAGES[:]
 
-        io.singer_write_message = test_utils.accumulate_singer_messages
+        TapGoogleSheets.write_message = test_utils.accumulate_singer_messages
 
     @responses.activate()
     def test_output_name(self):

diff --git a/tap_google_sheets/tests/test_underscoring_column_names.py b/tap_google_sheets/tests/test_underscoring_column_names.py
@@ -4,7 +4,6 @@
 
 import responses
 import singer_sdk._singerlib as singer
-import singer_sdk.io_base as io
 
 import tap_google_sheets.tests.utils as test_utils
 from tap_google_sheets.tap import TapGoogleSheets
@@ -19,7 +18,7 @@ def setUp(self):
         responses.reset()
         del test_utils.SINGER_MESSAGES[:]
 
-        io.singer_write_message = test_utils.accumulate_singer_messages
+        TapGoogleSheets.write_message = test_utils.accumulate_singer_messages
 
     @responses.activate()
     def test_underscoring_column_names(self):

diff --git a/tap_google_sheets/tests/utils.py b/tap_google_sheets/tests/utils.py
@@ -17,7 +17,7 @@
 }
 
 
-def accumulate_singer_messages(message):
+def accumulate_singer_messages(_, message):
     """Collect singer library write_message in tests."""
     SINGER_MESSAGES.append(message)