diff --git a/ingestion/tests/unit/test_build_connection_url.py b/ingestion/tests/unit/test_build_connection_url.py index ae35d745554..3c5516c3b2b 100644 --- a/ingestion/tests/unit/test_build_connection_url.py +++ b/ingestion/tests/unit/test_build_connection_url.py @@ -70,7 +70,7 @@ class TestGetConnectionURL(unittest.TestCase): hostPort="localhost:3306", databaseSchema="openmetadata_db", ) - engine_connection = MySQLConnection(connection).get_client() + engine_connection = MySQLConnection(connection).client self.assertEqual( str(engine_connection.url), "mysql+pymysql://openmetadata_user:openmetadata_password@localhost:3306/openmetadata_db", @@ -93,7 +93,7 @@ class TestGetConnectionURL(unittest.TestCase): "get_token", return_value=AccessToken(token="mocked_token", expires_on=100), ): - engine_connection = MySQLConnection(connection).get_client() + engine_connection = MySQLConnection(connection).client self.assertEqual( str(engine_connection.url), "mysql+pymysql://openmetadata_user:mocked_token@localhost:3306/openmetadata_db",