mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-16 20:48:37 -04:00
more
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user