set online alembic migrations

Author: sumaro2101

api_v1/tests/conftest.py

@@ -9,6 +9,8 @@
 from asgi_lifespan import LifespanManager
 from fastapi import FastAPI
 from sqlalchemy.pool import NullPool
+from alembic.config import Config
+from alembic import command
 
 from config import test_connection, settings, BaseModel
 from config import db_connection
@@ -21,6 +23,12 @@
 )
 
 
+cfg = Config(settings.alembic.CONFIG_PATH.as_posix())
+cfg.set_main_option('script_location',
+                    settings.alembic.MIGRATION_PATH.as_posix(),
+                    )
+
+
 @pytest.fixture(scope='session', autouse=True)
 def event_loop(request):
     loop = asyncio.get_event_loop_policy().new_event_loop()
@@ -39,7 +47,7 @@ async def app() -> AsyncGenerator[LifespanManager, Any]:
     async def lifespan(app: FastAPI):
         async with db_setup.engine.begin() as conn:
             await conn.run_sync(BaseModel.metadata.create_all)
-            sys.stdout.write('alembic upgrade head')
+            await conn.run_sync(alembic_do_upgrade)
             yield
             await conn.run_sync(BaseModel.metadata.drop_all)
 
@@ -63,3 +71,11 @@ async def client(app: FastAPI) -> AsyncGenerator[httpx.AsyncClient, Any]:
         base_url=current_home + current_api,
     ) as client:
         yield client
+
+
+def alembic_do_upgrade(connection):
+    """
+    Upgrade миграция алембик
+    """
+    cfg.attributes['connection'] = connection
+    command.upgrade(cfg, 'head')

async_alembic/env.py

@@ -3,7 +3,8 @@
 
 from sqlalchemy import pool
 from sqlalchemy.engine import Connection
-from sqlalchemy.ext.asyncio import async_engine_from_config
+from sqlalchemy.ext.asyncio import async_engine_from_config, AsyncEngine
+from sqlalchemy import engine_from_config
 
 from alembic import context
 
@@ -84,8 +85,22 @@ async def run_async_migrations() -> None:
 
 def run_migrations_online() -> None:
     """Run migrations in 'online' mode."""
-
-    asyncio.run(run_async_migrations())
+    connectable = context.config.attributes.get('connection', None)
+    if connectable is None:
+        connectable = AsyncEngine(
+            engine_from_config(
+                context.config.get_section(
+                    context.config.config_ini_section,
+                    ),
+                prefix='sqlalchemy.',
+                poolclass=pool.NullPool,
+                future=True,
+            )
+        ) 
+    if isinstance(connectable, AsyncEngine):
+        asyncio.run(run_async_migrations(connectable))
+    else:
+        do_run_migrations(connectable)
 
 
 if context.is_offline_mode():