darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 07:35:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Fix None type error in last-used endpoint.

Browse Source
This commit is contained in:
Rebecca Law
2020-02-06 11:23:50 +00:00
parent 32082521ee
commit 09f798ea14
2 changed files with 12 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/template_statistics/rest.py
View File

@@ -64,4 +64,4 @@ def get_last_used_datetime_for_template(service_id, template_id):
last_date_used = dao_get_last_date_template_was_used(template_id=template_id, last_date_used = dao_get_last_date_template_was_used(template_id=template_id,
service_id=service_id) service_id=service_id)
return jsonify(last_date_used=last_date_used.strftime(DATETIME_FORMAT)) return jsonify(last_date_used=last_date_used.strftime(DATETIME_FORMAT) if last_date_used else last_date_used)

11
tests/app/template_statistics/test_rest.py
View File

@@ -210,6 +210,17 @@ def test_get_last_used_datetime_for_template(
assert json_resp['last_date_used'] == date_from_notification.strftime(DATETIME_FORMAT) assert json_resp['last_date_used'] == date_from_notification.strftime(DATETIME_FORMAT)
def test_get_last_used_datetime_for_template_returns_none_if_no_usage_of_template(
admin_request, sample_template
):
json_resp = admin_request.get(
'template_statistics.get_last_used_datetime_for_template',
service_id=str(sample_template.service_id),
template_id=sample_template.id
)
assert json_resp['last_date_used'] is None
def test_get_last_used_datetime_for_template_returns_400_if_service_does_not_exist( def test_get_last_used_datetime_for_template_returns_400_if_service_does_not_exist(
admin_request, sample_template admin_request, sample_template
): ):