setup test

Author: darixsamani

tests/setup.py

@@ -1,4 +1,4 @@
-from sqlalchemy import create_engine
+from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
 from sqlalchemy.orm import sessionmaker
 from fastapi.testclient import TestClient
 from app import app
@@ -17,19 +17,17 @@ def get_url():
 
 SQLALCHEMY_DATABASE_URL = get_url()
 
-engine = create_engine(SQLALCHEMY_DATABASE_URL)
-TestingSessionLocal = sessionmaker(
-    autocommit=False, autoflush=False, bind=engine)
 
-# Base.metadata.create_all(bind=engine)
 
+engine =  create_async_engine(SQLALCHEMY_DATABASE_URL)
+TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine, class_=AsyncSession)
 
-def override_get_db():
-    try:
-        db = TestingSessionLocal()
-        yield db
-    finally:
-        db.close()
+Base.metadata.create_all(bind=engine)
+
+
+async def override_get_db():
+    async with TestingSessionLocal() as session:
+        yield session
 
 
 app.dependency_overrides[get_db] = override_get_db

tests/test_users.py

@@ -1 +1,3 @@
-from .setup import client
+from .setup import client
+
+