Change DAO to return the notification from an update, not Boolean

- tests reflect this
This commit is contained in:
Martyn Inglis
2016-09-13 12:29:40 +01:00
parent d2b936e631
commit 8541f6adcd
2 changed files with 29 additions and 7 deletions

View File

@@ -139,7 +139,7 @@ def _update_notification_status(notification, status):
status = _decide_permanent_temporary_failure(current_status=notification.status, status=status)
notification.status = status
dao_update_notification(notification)
return True
return notification
@statsd(namespace="dao")
@@ -152,7 +152,7 @@ def update_notification_status_by_id(notification_id, status):
Notification.status == 'pending')).first()
if not notification:
return False
return None
return _update_notification_status(
notification=notification,
@@ -169,7 +169,7 @@ def update_notification_status_by_reference(reference, status):
Notification.status == 'pending')).first()
if not notification:
return False
return None
return _update_notification_status(
notification=notification,