mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-15 15:28:02 -04:00
b53fdf1f3f
This version of the client removed the request method, path and body from the encode and decode methods. The biggest changes here is to the unit tests.
78 lines
3.2 KiB
Python
78 lines
3.2 KiB
Python
import json
|
|
from datetime import (date, timedelta)
|
|
from flask import url_for
|
|
from tests import create_authorization_header
|
|
|
|
|
|
def test_fragment_count(notify_api, sample_provider_statistics):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
endpoint = url_for(
|
|
'service.get_service_provider_aggregate_statistics',
|
|
service_id=str(sample_provider_statistics.service.id))
|
|
auth_header = create_authorization_header()
|
|
resp = client.get(
|
|
endpoint,
|
|
headers=[auth_header]
|
|
)
|
|
assert resp.status_code == 200
|
|
json_resp = json.loads(resp.get_data(as_text=True))
|
|
assert json_resp['data']['sms_count'] == 1
|
|
|
|
|
|
def test_fragment_count_from_to(notify_api, sample_provider_statistics):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
today_str = date.today().strftime('%Y-%m-%d')
|
|
endpoint = url_for(
|
|
'service.get_service_provider_aggregate_statistics',
|
|
service_id=str(sample_provider_statistics.service.id),
|
|
date_from=today_str,
|
|
date_to=today_str)
|
|
auth_header = create_authorization_header()
|
|
resp = client.get(
|
|
endpoint,
|
|
headers=[auth_header]
|
|
)
|
|
assert resp.status_code == 200
|
|
json_resp = json.loads(resp.get_data(as_text=True))
|
|
assert json_resp['data']['sms_count'] == 1
|
|
|
|
|
|
def test_fragment_count_from_greater_than_to(notify_api, sample_provider_statistics):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
today_str = date.today().strftime('%Y-%m-%d')
|
|
yesterday_str = date.today() - timedelta(days=1)
|
|
endpoint = url_for(
|
|
'service.get_service_provider_aggregate_statistics',
|
|
service_id=str(sample_provider_statistics.service.id),
|
|
date_from=today_str,
|
|
date_to=yesterday_str)
|
|
auth_header = create_authorization_header()
|
|
resp = client.get(
|
|
endpoint,
|
|
headers=[auth_header]
|
|
)
|
|
assert resp.status_code == 400
|
|
json_resp = json.loads(resp.get_data(as_text=True))
|
|
assert 'date_from needs to be greater than date_to' in json_resp['message']['_schema']
|
|
|
|
|
|
def test_fragment_count_in_future(notify_api, sample_provider_statistics):
|
|
with notify_api.test_request_context():
|
|
with notify_api.test_client() as client:
|
|
tomorrow_str = (date.today() + timedelta(days=1)).strftime('%Y-%m-%d')
|
|
endpoint = url_for(
|
|
'service.get_service_provider_aggregate_statistics',
|
|
service_id=str(sample_provider_statistics.service.id),
|
|
date_from=tomorrow_str)
|
|
auth_header = create_authorization_header()
|
|
resp = client.get(
|
|
endpoint,
|
|
headers=[auth_header]
|
|
)
|
|
assert resp.status_code == 400
|
|
json_resp = json.loads(resp.get_data(as_text=True))
|
|
assert 'Date cannot be in the future' in json_resp['message']['date_from']
|