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

Merge pull request #155 from alphagov/notification_api_get_update

Browse Source 
Update schemas to return more details about the job and template for …
This commit is contained in:
NIcholas Staples
2016-03-16 10:30:09 +00:00
parent 041ac318d7 16882961ad
commit 1425290eef
2 changed files with 17 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/schemas.py
View File

@@ -171,6 +171,9 @@ class SmsAdminNotificationSchema(SmsNotificationSchema):
class NotificationStatusSchema(BaseSchema):
template = fields.Nested(TemplateSchema, only=["id", "name"], dump_only=True)
job = fields.Nested(JobSchema, only=["id", "file_name"], dump_only=True)
class Meta:
model = models.Notification

16
tests/app/notifications/test_rest.py
View File

@@ -25,7 +25,13 @@ def test_get_notification_by_id(notify_api, sample_notification):
notification = json.loads(response.get_data(as_text=True))['notification']
assert notification['status'] == 'sent'
assert notification['template'] == sample_notification.template.id
assert notification['template'] == {
'id': sample_notification.template.id,
'name': sample_notification.template.name}
assert notification['job'] == {
'id': str(sample_notification.job.id),
'file_name': sample_notification.job.file_name
}
assert notification['to'] == '+447700900855'
assert notification['service'] == str(sample_notification.service_id)
assert response.status_code == 200
@@ -64,7 +70,13 @@ def test_get_all_notifications(notify_api, sample_notification):
notifications = json.loads(response.get_data(as_text=True))
assert notifications['notifications'][0]['status'] == 'sent'
assert notifications['notifications'][0]['template'] == sample_notification.template.id
assert notifications['notifications'][0]['template'] == {
'id': sample_notification.template.id,
'name': sample_notification.template.name}
assert notifications['notifications'][0]['job'] == {
'id': str(sample_notification.job.id),
'file_name': sample_notification.job.file_name
}
assert notifications['notifications'][0]['to'] == '+447700900855'
assert notifications['notifications'][0]['service'] == str(sample_notification.service_id)
assert response.status_code == 200