mirror of
https://github.com/GSA/notifications-api.git
synced 2026-07-17 21:20:59 -04:00
Optimise the query for getting the platform statistics for all services. The page should render for all time after this change.
This is one step closer to eliminating the need to read from NotificationHistory.
This commit is contained in:
@@ -289,6 +289,7 @@ def set_up_data():
|
||||
create_notification(sms_template, created_at=datetime(2018, 10, 31, 11, 0, 0))
|
||||
create_notification(sms_template, created_at=datetime(2018, 10, 31, 12, 0, 0), status='delivered')
|
||||
create_notification(email_template, created_at=datetime(2018, 10, 31, 13, 0, 0), status='delivered')
|
||||
return service_1, service_2
|
||||
|
||||
|
||||
def test_fetch_notification_statuses_for_job(sample_template):
|
||||
@@ -308,8 +309,32 @@ def test_fetch_notification_statuses_for_job(sample_template):
|
||||
|
||||
@freeze_time('2018-10-31 14:00')
|
||||
def test_fetch_stats_for_all_services_by_date_range(notify_db_session):
|
||||
set_up_data()
|
||||
results = fetch_stats_for_all_services_by_date_range( start_date=date(2018, 10, 29),
|
||||
end_date=date(2018, 10, 31))
|
||||
print(results)
|
||||
assert len(results) == 2
|
||||
service_1, service_2 = set_up_data()
|
||||
results = fetch_stats_for_all_services_by_date_range(start_date=date(2018, 10, 29),
|
||||
end_date=date(2018, 10, 31))
|
||||
assert len(results) == 5
|
||||
|
||||
assert results[0].service_id == service_1.id
|
||||
assert results[0].notification_type == 'email'
|
||||
assert results[0].status == 'delivered'
|
||||
assert results[0].count == 4
|
||||
|
||||
assert results[1].service_id == service_1.id
|
||||
assert results[1].notification_type == 'sms'
|
||||
assert results[1].status == 'created'
|
||||
assert results[1].count == 2
|
||||
|
||||
assert results[2].service_id == service_1.id
|
||||
assert results[2].notification_type == 'sms'
|
||||
assert results[2].status == 'delivered'
|
||||
assert results[2].count == 11
|
||||
|
||||
assert results[3].service_id == service_2.id
|
||||
assert results[3].notification_type == 'letter'
|
||||
assert results[3].status == 'delivered'
|
||||
assert results[3].count == 10
|
||||
|
||||
assert results[4].service_id == service_2.id
|
||||
assert not results[4].notification_type
|
||||
assert not results[4].status
|
||||
assert not results[4].count
|
||||
|
||||
@@ -1642,31 +1642,37 @@ def test_get_detailed_services_only_includes_todays_notifications(notify_db, not
|
||||
}
|
||||
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
'set_time',
|
||||
['2017-03-28T12:00:00', '2017-01-28T12:00:00', '2017-01-02T12:00:00', '2017-10-31T12:00:00']
|
||||
)
|
||||
def test_get_detailed_services_for_date_range(notify_db, notify_db_session, set_time):
|
||||
@pytest.mark.parametrize("start_date_delta, end_date_delta",
|
||||
[(2, 1),
|
||||
(3, 2),
|
||||
(1, 0)
|
||||
])
|
||||
@freeze_time('2017-03-28T12:00:00')
|
||||
def test_get_detailed_services_for_date_range(sample_template, start_date_delta, end_date_delta):
|
||||
from app.service.rest import get_detailed_services
|
||||
|
||||
with freeze_time(set_time):
|
||||
create_sample_notification(notify_db, notify_db_session, created_at=datetime.utcnow() - timedelta(days=3))
|
||||
create_sample_notification(notify_db, notify_db_session, created_at=datetime.utcnow() - timedelta(days=2))
|
||||
create_sample_notification(notify_db, notify_db_session, created_at=datetime.utcnow() - timedelta(days=1))
|
||||
create_sample_notification(notify_db, notify_db_session, created_at=datetime.utcnow())
|
||||
create_ft_notification_status(bst_date=(datetime.utcnow() - timedelta(days=3)).date(),
|
||||
service=sample_template.service,
|
||||
notification_type='sms')
|
||||
create_ft_notification_status(bst_date=(datetime.utcnow() - timedelta(days=2)).date(),
|
||||
service=sample_template.service,
|
||||
notification_type='sms')
|
||||
create_ft_notification_status(bst_date=(datetime.utcnow() - timedelta(days=1)).date(),
|
||||
service=sample_template.service,
|
||||
notification_type='sms')
|
||||
|
||||
start_date = (datetime.utcnow() - timedelta(days=2)).date()
|
||||
end_date = (datetime.utcnow() - timedelta(days=1)).date()
|
||||
create_notification(template=sample_template, created_at=datetime.utcnow(), status='delivered')
|
||||
|
||||
start_date = (datetime.utcnow() - timedelta(days=start_date_delta)).date()
|
||||
end_date = (datetime.utcnow() - timedelta(days=end_date_delta)).date()
|
||||
|
||||
data = get_detailed_services(only_active=False, include_from_test_key=True,
|
||||
start_date=start_date, end_date=end_date)
|
||||
|
||||
assert len(data) == 1
|
||||
assert data[0]['statistics'] == {
|
||||
EMAIL_TYPE: {'delivered': 0, 'failed': 0, 'requested': 0},
|
||||
SMS_TYPE: {'delivered': 0, 'failed': 0, 'requested': 2},
|
||||
LETTER_TYPE: {'delivered': 0, 'failed': 0, 'requested': 0}
|
||||
}
|
||||
assert data[0]['statistics'][EMAIL_TYPE] == {'delivered': 0, 'failed': 0, 'requested': 0}
|
||||
assert data[0]['statistics'][SMS_TYPE] == {'delivered': 2, 'failed': 0, 'requested': 2}
|
||||
assert data[0]['statistics'][LETTER_TYPE] == {'delivered': 0, 'failed': 0, 'requested': 0}
|
||||
|
||||
|
||||
def test_search_for_notification_by_to_field(client, sample_template, sample_email_template):
|
||||
|
||||
Reference in New Issue
Block a user