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

fix more tests

Browse Source
This commit is contained in:
Kenneth Kehl
2024-10-30 11:15:08 -07:00
parent 79ceddfee4
commit 7ee741b91c
6 changed files with 75 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/app/dao/notification_dao/test_notification_dao_delete_notifications.py
View File

@@ -2,7 +2,9 @@ import uuid
from datetime import datetime, timedelta
from freezegun import freeze_time
from sqlalchemy import func, select
from app import db
from app.dao.notifications_dao import (
insert_notification_history_delete_notifications,
move_notifications_to_notification_history,
@@ -172,12 +174,13 @@ def test_move_notifications_just_deletes_test_key_notifications(sample_template)
assert Notification.query.count() == 0
assert NotificationHistory.query.count() == 2
assert (
NotificationHistory.query.filter(
NotificationHistory.key_type == KeyType.TEST
).count()
== 0
stmt = (
select(func.count())
.select_from(NotificationHistory)
.where(NotificationHistory.key_type == KeyType.TEST)
)
count = db.session.execute(stmt).scalar() or 0
assert count == 0
@freeze_time("2020-03-20 14:00")