mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 08:51:30 -05:00
69 lines
2.6 KiB
Python
69 lines
2.6 KiB
Python
|
from datetime import date, timedelta
|
||
|
|
|
||
|
|
from flask import json
|
||
|
|
from freezegun import freeze_time
|
||
|
|
|
||
|
|
from tests import create_authorization_header
|
||
|
|
from tests.app.conftest import sample_notification_statistics as create_sample_notification_statistics
|
||
|
|
|
||
|
|
|
||
|
|
def test_get_notification_statistics(notify_api, sample_notification_statistics):
|
||
|
|
with notify_api.test_request_context():
|
||
|
|
with notify_api.test_client() as client:
|
||
|
|
auth_header = create_authorization_header(
|
||
|
|
service_id=sample_notification_statistics.service_id
|
||
|
|
)
|
||
|
|
|
||
|
|
response = client.get(
|
||
|
|
'/notifications/statistics',
|
||
|
|
headers=[auth_header]
|
||
|
|
)
|
||
|
|
|
||
|
|
notifications = json.loads(response.get_data(as_text=True))
|
||
|
|
stats = notifications['data'][0]
|
||
|
|
assert stats['emails_requested'] == 2
|
||
|
|
assert stats['emails_delivered'] == 1
|
||
|
|
assert stats['emails_failed'] == 1
|
||
|
|
assert stats['sms_requested'] == 2
|
||
|
|
assert stats['sms_delivered'] == 1
|
||
|
|
assert stats['service'] == str(sample_notification_statistics.service_id)
|
||
|
|
assert response.status_code == 200
|
||
|
|
|
||
|
|
|
||
|
|
@freeze_time('1955-11-05T12:00:00')
|
||
|
|
def test_get_notification_statistics_only_returns_today(notify_api, notify_db, notify_db_session, sample_service):
|
||
|
|
with notify_api.test_request_context():
|
||
|
|
with notify_api.test_client() as client:
|
||
|
|
yesterdays_notification_statistics = create_sample_notification_statistics(
|
||
|
|
notify_db,
|
||
|
|
notify_db_session,
|
||
|
|
service=sample_service,
|
||
|
|
day=date.today() - timedelta(days=1)
|
||
|
|
)
|
||
|
|
todays_notification_statistics = create_sample_notification_statistics(
|
||
|
|
notify_db,
|
||
|
|
notify_db_session,
|
||
|
|
service=sample_service,
|
||
|
|
day=date.today()
|
||
|
|
)
|
||
|
|
tomorrows_notification_statistics = create_sample_notification_statistics(
|
||
|
|
notify_db,
|
||
|
|
notify_db_session,
|
||
|
|
service=sample_service,
|
||
|
|
day=date.today() + timedelta(days=1)
|
||
|
|
)
|
||
|
|
|
||
|
|
auth_header = create_authorization_header(
|
||
|
|
service_id=sample_service.id
|
||
|
|
)
|
||
|
|
|
||
|
|
response = client.get(
|
||
|
|
'/notifications/statistics',
|
||
|
|
headers=[auth_header]
|
||
|
|
)
|
||
|
|
|
||
|
|
notifications = json.loads(response.get_data(as_text=True))
|
||
|
|
assert len(notifications['data']) == 1
|
||
|
|
assert notifications['data'][0]['day'] == date.today().isoformat()
|
||
|
|
assert response.status_code == 200
|