diff --git a/creyPY/fastapi/db/async_session.py b/creyPY/fastapi/db/async_session.py index f0186ca..98dbc77 100644 --- a/creyPY/fastapi/db/async_session.py +++ b/creyPY/fastapi/db/async_session.py @@ -1,15 +1,14 @@ from typing import AsyncGenerator -from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine -from sqlalchemy.orm import sessionmaker +from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine -from .common import SQLALCHEMY_DATABASE_URL, name +from .common import SQLALCHEMY_DATABASE_URL, name, ssl_mode async_engine = create_async_engine( - SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True, connect_args={"sslmode": "require"} + SQLALCHEMY_DATABASE_URL + name, pool_pre_ping=True, connect_args={"sslmode": ssl_mode} ) -AsyncSessionLocal = sessionmaker( +AsyncSessionLocal = async_sessionmaker( bind=async_engine, class_=AsyncSession, expire_on_commit=False, diff --git a/creyPY/fastapi/testing_unit.py b/creyPY/fastapi/testing_unit.py index 0ef59be..c8607e0 100644 --- a/creyPY/fastapi/testing_unit.py +++ b/creyPY/fastapi/testing_unit.py @@ -20,7 +20,6 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase): transport=ASGITransport(app=app), base_url="http://testserver", follow_redirects=True ) cls.default_headers = headers - print("setting up abstract") @classmethod def setup_database( @@ -52,7 +51,7 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase): async_db_url, echo=False, pool_pre_ping=True, - connect_args={"sslmode": "require"}, + connect_args={"sslmode": ssl_mode}, ) async def get(self, url: str, r_code: int = 200, parse_json=True) -> dict | bytes: