fix: fixed multiple bugs in database handling

This commit is contained in:
2025-01-27 16:26:26 +01:00
parent 410ae12f8e
commit 31c4cbb055
2 changed files with 5 additions and 7 deletions

View File

@@ -1,15 +1,14 @@
from typing import AsyncGenerator from typing import AsyncGenerator
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
from sqlalchemy.orm import sessionmaker
from .common import SQLALCHEMY_DATABASE_URL, name from .common import SQLALCHEMY_DATABASE_URL, name, ssl_mode
async_engine = create_async_engine( 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, bind=async_engine,
class_=AsyncSession, class_=AsyncSession,
expire_on_commit=False, expire_on_commit=False,

View File

@@ -20,7 +20,6 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
transport=ASGITransport(app=app), base_url="http://testserver", follow_redirects=True transport=ASGITransport(app=app), base_url="http://testserver", follow_redirects=True
) )
cls.default_headers = headers cls.default_headers = headers
print("setting up abstract")
@classmethod @classmethod
def setup_database( def setup_database(
@@ -52,7 +51,7 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
async_db_url, async_db_url,
echo=False, echo=False,
pool_pre_ping=True, 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: async def get(self, url: str, r_code: int = 200, parse_json=True) -> dict | bytes: