zalupka

test/test.py

@@ -3,29 +3,12 @@ import asyncio
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from backend.session import session_maker
+from migrations.env import run_async_migrations
 from models import Product, ProductBarcode
 
 
 async def main(session: AsyncSession):
-    client_ids = [8, 18]
-    for client_id in client_ids:
-        for i in range(1, 500 + 1):
-            product = Product(
-                name=f"Товар №{i}",
-                article=f"Ариткул товара №{i}",
-                client_id=client_id
-            )
-            session.add(product)
-            await session.flush()
-            for j in range(1, 5 + 1):
-                barcode = ProductBarcode(
-                    barcode=f"Штрихкод №{j} для товара №{i}",
-                    product_id=product.id
-                )
-                session.add(barcode)
-            await session.flush()
-    await session.commit()
-
+    await run_async_migrations()
 
 async def preload():
     async with session_maker() as session: