From f244afc0b4ee5bda4e94af554989e9a50d7a8924 Mon Sep 17 00:00:00 2001 From: Ayush Shah Date: Wed, 12 Jun 2024 14:00:24 +0530 Subject: [PATCH] Add Test Suite SSL (#16619) --- .../interface/sqlalchemy/sqa_test_suite_interface.py | 4 ++-- 1 file changed, 2 insertions(+), 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 index a0cec41d7734..408cce66feb6 100644 --- 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