Add Test Suite SSL (#16619)

open-metadata · Jun 12, 2024 · f244afc · f244afc
1 parent 906b730
commit f244afc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ingestion/src/metadata/data_quality/interface/sqlalchemy/sqa_test_suite_interface.py b/ingestion/src/metadata/data_quality/interface/sqlalchemy/sqa_test_suite_interface.py
@@ -29,13 +29,13 @@
 from metadata.generated.schema.tests.testDefinition import TestDefinition
 from metadata.ingestion.connections.session import create_and_bind_session
 from metadata.ingestion.ometa.ometa_api import OpenMetadata
-from metadata.ingestion.source.connections import get_connection
 from metadata.mixins.sqalchemy.sqa_mixin import SQAInterfaceMixin
 from metadata.profiler.processor.runner import QueryRunner
 from metadata.profiler.processor.sampler.sqlalchemy.sampler import SQASampler
 from metadata.utils.constants import TEN_MIN
 from metadata.utils.importer import import_test_case_class
 from metadata.utils.logger import test_suite_logger
+from metadata.utils.ssl_manager import get_ssl_connection
 from metadata.utils.timeout import cls_timeout
 
 logger = test_suite_logger()
@@ -72,7 +72,7 @@ def __init__(
 
  def create_session(self):
  self.session = create_and_bind_session(
- get_connection(self.service_connection_config)
+ get_ssl_connection(self.service_connection_config)
  )
 
  @property