diff --git a/tests/integration_tests/conftest.py b/tests/integration_tests/conftest.py index fee13c8950a..67d73ff147b 100644 --- a/tests/integration_tests/conftest.py +++ b/tests/integration_tests/conftest.py @@ -75,6 +75,11 @@ def setup_sample_data() -> Any: db.session.commit() +@pytest.fixture +def login_as_admin(login_as: Callable[..., None]): + yield login_as("admin") + + def drop_from_schema(engine: Engine, schema_name: str): schemas = engine.execute(f"SHOW SCHEMAS").fetchall() if schema_name not in [s[0] for s in schemas]: