refactor: rename get_sqla_engine_with_context (#28012)

This commit is contained in:
Beto Dealmeida
2024-04-12 13:31:05 -04:00
committed by GitHub
parent 06077d42a8
commit 99a1601aea
45 changed files with 99 additions and 100 deletions

View File

@@ -38,7 +38,7 @@ class TestPrestoValidator(SupersetTestCase):
self.validator = PrestoDBSQLValidator
self.database = MagicMock()
self.database_engine = (
self.database.get_sqla_engine_with_context.return_value.__enter__.return_value
self.database.get_sqla_engine.return_value.__enter__.return_value
)
self.database_conn = self.database_engine.raw_connection.return_value
self.database_cursor = self.database_conn.cursor.return_value