fix: used setup_ssl.side_effect

open-metadata · Jan 7, 2025 · 4a89183 · 4a89183
1 parent 2b0b488
commit 4a89183
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/ingestion/tests/unit/test_ssl_manager.py b/ingestion/tests/unit/test_ssl_manager.py
@@ -128,13 +128,16 @@ def test_init(self, mock_ssl_manager, test_connection):
 
 
 class CassandraSourceSSLTest(TestCase):
-    @patch("metadata.ingestion.source.database.cassandra.connection.get_connection")
+    @patch("metadata.utils.ssl_manager.SSLManager.setup_ssl")
     @patch(
         "metadata.ingestion.source.database.cassandra.metadata.CassandraSource.test_connection"
     )
-    def test_init(self, get_connection, test_connection):
+    @patch("metadata.ingestion.source.database.cassandra.connection.get_connection")
+    def test_init(self, get_connection, test_connection, setup_ssl):
         get_connection.return_value = True
         test_connection.return_value = True
+        setup_ssl.side_effect = lambda x: x
+
         config = WorkflowSource(
             **{
                 "type": "cassandra",