mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 17:01:35 -05:00
92897e20a8
Our previous test ws returning a notification without a `sent_by` attribute, which meant that cost was always 0. Unfortunately, this meant that returning a real value for cost was untested and (whaddya know) it broke immediately. Old test scenario: - billable_units=1, sent_by=None, cost=0 New scenarios - billable_units=0, sent_by='mmg', cost=0 - billable_units=1, sent_by='mmg', cost=1
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
import json
|
|
import pytest
|
|
|
|
from app import DATETIME_FORMAT
|
|
from tests import create_authorization_header
|
|
from tests.app.conftest import sample_notification as create_sample_notification
|
|
|
|
|
|
@pytest.mark.parametrize('billable_units, provider', [
|
|
(1, 'mmg'),
|
|
(0, 'mmg'),
|
|
(1, None)
|
|
])
|
|
def test_get_notification_by_id_returns_200(
|
|
client, notify_db, notify_db_session, sample_provider_rate, billable_units, provider
|
|
):
|
|
sample_notification = create_sample_notification(
|
|
notify_db, notify_db_session, billable_units=billable_units, sent_by=provider
|
|
)
|
|
|
|
auth_header = create_authorization_header(service_id=sample_notification.service_id)
|
|
response = client.get(
|
|
path='/v2/notifications/{}'.format(sample_notification.id),
|
|
headers=[('Content-Type', 'application/json'), auth_header])
|
|
|
|
assert response.status_code == 200
|
|
assert response.headers['Content-type'] == 'application/json'
|
|
|
|
json_response = json.loads(response.get_data(as_text=True))
|
|
|
|
expected_template_response = {
|
|
'id': '{}'.format(sample_notification.serialize()['template']['id']),
|
|
'version': sample_notification.serialize()['template']['version'],
|
|
'uri': sample_notification.serialize()['template']['uri']
|
|
}
|
|
|
|
expected_response = {
|
|
'id': '{}'.format(sample_notification.id),
|
|
'reference': None,
|
|
'email_address': None,
|
|
'phone_number': '{}'.format(sample_notification.to),
|
|
'line_1': None,
|
|
'line_2': None,
|
|
'line_3': None,
|
|
'line_4': None,
|
|
'line_5': None,
|
|
'line_6': None,
|
|
'postcode': None,
|
|
'cost': sample_notification.cost(),
|
|
'type': '{}'.format(sample_notification.notification_type),
|
|
'status': '{}'.format(sample_notification.status),
|
|
'template': expected_template_response,
|
|
'created_at': sample_notification.created_at.strftime(DATETIME_FORMAT),
|
|
'sent_at': sample_notification.sent_at,
|
|
'completed_at': sample_notification.completed_at()
|
|
}
|
|
|
|
assert json_response == expected_response
|