Fix failing unit tests

aws · Sep 27, 2023 · 525d28b · 525d28b
1 parent 2234322
commit 525d28b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/core/tests/test_unload_sys_table.py b/core/tests/test_unload_sys_table.py
@@ -38,7 +38,7 @@
 conn.cursor.return_value = cursor
 
 
-def mock_get_connection_cred(self, val, max_attempts):
+def mock_get_connection_cred(self, val):
     return {
         "host": "somehost",
         "port": 5437,
@@ -82,9 +82,7 @@ def test_unload_system_table_with_unload_query(self, mock_boto, mock_debug):
 
         unload_object.unload_system_table()
 
-        mock_debug.assert_called_once_with(
-            "Executed unload query: stl_unload"
-        )
+        mock_debug.assert_called_once_with("Executed unload query: stl_unload")
 
     @patch.object(ReplayPrep, "get_connection_credentials", mock_get_connection_cred)
     @patch("core.replay.unload_sys_table.db_connect", mock_db_connect_error)