darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-03 01:41:05 -05:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit for sqlalchemy2.0 migration

Browse Source
This commit is contained in:
Aditi Anand
2024-03-13 12:42:11 -04:00
parent 158316a77b
commit 385ec3583f
23 changed files with 1694 additions and 1590 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
tests/conftest.py
View File

@@ -94,23 +94,24 @@ def notify_db_session(_notify_db, sms_providers):
yield _notify_db.session
_notify_db.session.remove()
for tbl in reversed(_notify_db.metadata.sorted_tables):
if tbl.name not in [
"provider_details",
"key_types",
"branding_type",
"job_status",
"provider_details_history",
"template_process_type",
"notifications_all_time_view",
"notification_status_types",
"organization_types",
"service_permission_types",
"auth_type",
"invite_status_type",
"service_callback_type",
]:
_notify_db.engine.execute(tbl.delete())
with _notify_db.engine.begin() as connection:
for tbl in reversed(_notify_db.metadata.sorted_tables):
if tbl.name not in [
"provider_details",
"key_types",
"branding_type",
"job_status",
"provider_details_history",
"template_process_type",
"notifications_all_time_view",
"notification_status_types",
"organization_types",
"service_permission_types",
"auth_type",
"invite_status_type",
"service_callback_type",
]:
connection.execute(tbl.delete())
_notify_db.session.commit()