creyD/creyPY
1
0
Fork 0
mirror of https://github.com/creyD/creyPY.git synced 2026-04-12 19:30:30 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: added ssl option to test database

Browse Source
This commit is contained in:
Conrad
2025-01-27 13:16:55 +01:00
parent 1f224c44bc
commit 410ae12f8e
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
creyPY/fastapi/testing_unit.py
View File

@@ -24,13 +24,18 @@ class AbstractTestAPI(unittest.IsolatedAsyncioTestCase):
@classmethod @classmethod
def setup_database( def setup_database(
cls, sync_db_url: str, async_db_url: str, base: Type[Base], btree_gist: bool = False cls,
sync_db_url: str,
async_db_url: str,
base: Type[Base],
btree_gist: bool = False,
ssl_mode: str = "require",
): ):
cls.engine_s = create_engine( cls.engine_s = create_engine(
sync_db_url, sync_db_url,
echo=False, echo=False,
pool_pre_ping=True, pool_pre_ping=True,
connect_args={"sslmode": "require"}, connect_args={"sslmode": ssl_mode},
) )
if database_exists(cls.engine_s.url): if database_exists(cls.engine_s.url):
drop_database(cls.engine_s.url) drop_database(cls.engine_s.url)