diff --git a/airflow/utils/db_manager.py b/airflow/utils/db_manager.py index bc91f3c8003ca..46c4323dde539 100644 --- a/airflow/utils/db_manager.py +++ b/airflow/utils/db_manager.py @@ -86,7 +86,7 @@ def check_migration(self): return True return False - def _create_db_from_orm(self): + def create_db_from_orm(self): """Create database from ORM.""" self.log.info("Creating %s tables from the ORM", self.__class__.__name__) engine = self.session.get_bind().engine @@ -118,7 +118,7 @@ def initdb(self): if db_exists: self.upgradedb() else: - self._create_db_from_orm() + self.create_db_from_orm() def upgradedb(self, to_revision=None, from_revision=None, show_sql_only=False): """Upgrade the database.""" diff --git a/providers/fab/src/airflow/providers/fab/auth_manager/models/db.py b/providers/fab/src/airflow/providers/fab/auth_manager/models/db.py index 5e2c53977458c..8ffb6cfeab8b4 100644 --- a/providers/fab/src/airflow/providers/fab/auth_manager/models/db.py +++ b/providers/fab/src/airflow/providers/fab/auth_manager/models/db.py @@ -54,8 +54,8 @@ class FABDBManager(BaseDBManager): alembic_file = (PACKAGE_DIR / "alembic.ini").as_posix() supports_table_dropping = True - def _create_db_from_orm(self): - super()._create_db_from_orm() + def create_db_from_orm(self): + super().create_db_from_orm() _get_flask_db(settings.SQL_ALCHEMY_CONN).create_all() def upgradedb(self, to_revision=None, from_revision=None, show_sql_only=False): diff --git a/tests/utils/test_db_manager.py b/tests/utils/test_db_manager.py index b44262961f85c..34f849bf2e011 100644 --- a/tests/utils/test_db_manager.py +++ b/tests/utils/test_db_manager.py @@ -113,7 +113,7 @@ class MockDBManager(BaseDBManager): class TestBaseDBManager: @mock.patch.object(BaseDBManager, "get_alembic_config") @mock.patch.object(BaseDBManager, "get_current_revision") - @mock.patch.object(BaseDBManager, "_create_db_from_orm") + @mock.patch.object(BaseDBManager, "create_db_from_orm") def test_create_db_from_orm_called_from_init( self, mock_create_db_from_orm, mock_current_revision, mock_config, session ):