From 525d28b737ed860e27431e56c733c446059a6dd2 Mon Sep 17 00:00:00 2001 From: Sathiish Kumar Date: Wed, 27 Sep 2023 11:44:07 -0700 Subject: [PATCH] Fix failing unit tests --- core/tests/test_unload_sys_table.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/core/tests/test_unload_sys_table.py b/core/tests/test_unload_sys_table.py index 5ec18a361..257948540 100644 --- 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)