Skip to content

Commit d2a7d02

Browse files
committed
Integrate alembic check in test
1 parent 3936683 commit d2a7d02

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

airflow-core/tests/unit/utils/test_db.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828

2929
import pytest
3030
from alembic.autogenerate import compare_metadata
31+
from alembic.command import check
3132
from alembic.config import Config
3233
from alembic.migration import MigrationContext
3334
from alembic.runtime.environment import EnvironmentContext
@@ -302,3 +303,13 @@ def test_upgradedb_raises_if_lower_than_v3_0_0(self, mock_inspect, caplog):
302303
"Downgrade to revision less than 3.0.0 requires that `ab_user` table is present. "
303304
"Please add FabDBManager to [core] external_db_managers and run fab migrations before proceeding"
304305
) in caplog.text
306+
307+
def test_has_pending_upgrade_ops(self):
308+
with mock.patch.dict(
309+
os.environ, {"AIRFLOW__DATABASE__ALEMBIC_INI_FILE_PATH": "/tmp/alembic.ini"}, clear=True
310+
):
311+
config = _get_alembic_config()
312+
assert config.config_file_name == "/tmp/alembic.ini"
313+
314+
config = _get_alembic_config()
315+
check(config)

0 commit comments

Comments
 (0)