diff --git a/tests/app/public_contracts/test_GET_notification.py b/tests/app/public_contracts/test_GET_notification.py index 738aea56a..0a3adc74c 100644 --- a/tests/app/public_contracts/test_GET_notification.py +++ b/tests/app/public_contracts/test_GET_notification.py @@ -16,13 +16,20 @@ def _get_notification(client, notification, url): return client.get(url, headers=[auth_header]) -def test_get_v2_notification(client, sample_notification): +def test_get_v2_sms_contract(client, sample_notification): response_json = return_json_from_response(_get_notification( client, sample_notification, '/v2/notifications/{}'.format(sample_notification.id) )) validate(response_json, get_notification_response) +def test_get_v2_email_contract(client, sample_email_notification): + response_json = return_json_from_response(_get_notification( + client, sample_email_notification, '/v2/notifications/{}'.format(sample_email_notification.id) + )) + validate(response_json, get_notification_response) + + def test_get_api_sms_contract(client, sample_notification): response_json = return_json_from_response(_get_notification( client, sample_notification, '/notifications/{}'.format(sample_notification.id)