This commit is contained in:
Kenneth Kehl
2024-11-15 07:48:49 -08:00
parent 8c7cb44399
commit b0bf042cee
8 changed files with 33 additions and 21 deletions

View File

@@ -3,8 +3,10 @@ from unittest.mock import ANY, call
import pytest
from freezegun import freeze_time
from sqlalchemy import select
from sqlalchemy.exc import SQLAlchemyError
from app import db
from app.celery import nightly_tasks
from app.celery.nightly_tasks import (
_delete_notifications_older_than_retention_by_type,
@@ -230,7 +232,7 @@ def test_save_daily_notification_processing_time(
save_daily_notification_processing_time(date_provided)
persisted_to_db = FactProcessingTime.query.all()
persisted_to_db = db.session.execute(select(FactProcessingTime)).scalars().all()
assert len(persisted_to_db) == 1
assert persisted_to_db[0].local_date == date(2021, 1, 17)
assert persisted_to_db[0].messages_total == 2
@@ -269,7 +271,7 @@ def test_save_daily_notification_processing_time_when_in_est(
save_daily_notification_processing_time(date_provided)
persisted_to_db = FactProcessingTime.query.all()
persisted_to_db = db.session.execute(select(FactProcessingTime)).scalars().all()
assert len(persisted_to_db) == 1
assert persisted_to_db[0].local_date == date(2021, 4, 17)
assert persisted_to_db[0].messages_total == 2

View File

@@ -464,7 +464,7 @@ def test_create_nightly_notification_status_for_service_and_day(notify_db_sessio
create_notification(template=first_template)
create_notification_history(template=second_template)
assert len(FactNotificationStatus.query.all()) == 0
assert len(db.session.execute(select(FactNotificationStatus)).scalars().all()) == 0
create_nightly_notification_status_for_service_and_day(
str(process_day),
@@ -540,7 +540,7 @@ def test_create_nightly_notification_status_for_service_and_day_overwrites_old_d
NotificationType.SMS,
)
new_fact_data = FactNotificationStatus.query.all()
new_fact_data = db.session.execute(select(FactNotificationStatus)).scalars().all()
assert len(new_fact_data) == 1
assert new_fact_data[0].notification_count == 1