From d36f01abf635ea5d94b7f5fd79596cfeae5cf0db Mon Sep 17 00:00:00 2001 From: IceS2 Date: Wed, 25 Sep 2024 17:37:56 +0200 Subject: [PATCH] Fix tearDown by using the proper file loader (#17994) --- ingestion/tests/cli_e2e/common/test_cli_db.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ingestion/tests/cli_e2e/common/test_cli_db.py b/ingestion/tests/cli_e2e/common/test_cli_db.py index cb59aad0156..981e092ef40 100644 --- a/ingestion/tests/cli_e2e/common/test_cli_db.py +++ b/ingestion/tests/cli_e2e/common/test_cli_db.py @@ -18,9 +18,9 @@ from abc import ABC, abstractmethod from pathlib import Path from typing import Optional -import yaml from sqlalchemy.engine import Engine +from metadata.config.common import load_config_file from metadata.generated.schema.entity.services.databaseService import DatabaseService from metadata.generated.schema.metadataIngestion.workflow import ( OpenMetadataWorkflowConfig, @@ -54,7 +54,7 @@ class CliCommonDB: @classmethod def tearDownClass(cls): workflow = OpenMetadataWorkflowConfig.model_validate( - yaml.safe_load(open(cls.config_file_path)) + load_config_file(Path(cls.config_file_path)) ) db_service: DatabaseService = cls.openmetadata.get_by_name( DatabaseService, workflow.source.serviceName