init testu

tests/conftest.py

@@ -0,0 +1,44 @@
+import os
+import sys
+from pathlib import Path
+import pytest
+from sqlalchemy.ext.asyncio import async_sessionmaker, AsyncSession
+
+ROOT_DIR = Path(__file__).resolve().parents[1]
+sys.path.insert(0, str(ROOT_DIR / "src"))
+
+from logger import log
+
+
+def pytest_sessionstart():
+    os.environ['MYSQL_HOST'] = 'localhost'
+    os.environ['MYSQL_USER'] = 'root'
+    os.environ['MYSQL_PASSWORD'] = 'root'
+
+
+@pytest.fixture(scope="function")
+async def db():
+    # from database.transaction import get_session
+    #
+    # async with get_session() as db:
+    #     yield db
+    #     await db.rollback()
+
+
+    import database
+    from database.config import create_db_engine
+
+    test_engine = create_db_engine()
+    test_async_session = async_sessionmaker(test_engine, expire_on_commit=True, class_=AsyncSession)
+
+    database.config.engine = test_engine
+    database.async_session = test_async_session
+
+    async with test_async_session() as session:
+        async with session.begin():
+            try:
+                yield session
+                await session.rollback()
+            except Exception as e:
+                log.exception("Error", exc_info=e)
+                await session.rollback()