Return notification postage in response for .post_precompiled_letter_notification

This commit is contained in:
Pea Tyczynska
2019-01-16 16:57:57 +00:00
parent 276a9a3828
commit 9ab97d3481
2 changed files with 3 additions and 2 deletions

View File

@@ -94,7 +94,8 @@ def post_precompiled_letter_notification():
resp = { resp = {
'id': notification.id, 'id': notification.id,
'reference': notification.client_reference 'reference': notification.client_reference,
'postage': notification.postage
} }
return jsonify(resp), 201 return jsonify(resp), 201

View File

@@ -510,7 +510,7 @@ def test_post_precompiled_letter_notification_returns_201(
assert notification_history.postage == expected_postage assert notification_history.postage == expected_postage
resp_json = json.loads(response.get_data(as_text=True)) resp_json = json.loads(response.get_data(as_text=True))
assert resp_json == {'id': str(notification.id), 'reference': 'letter-reference'} assert resp_json == {'id': str(notification.id), 'reference': 'letter-reference', 'postage': expected_postage}
def test_post_letter_notification_throws_error_for_invalid_postage(client, notify_user, mocker): def test_post_letter_notification_throws_error_for_invalid_postage(client, notify_user, mocker):