mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-03-13 23:02:43 -04:00
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from unittest.mock import Mock
|
|
from notifications_utils.template import LetterPreviewTemplate
|
|
|
|
from app.template_previews import TemplatePreview
|
|
|
|
|
|
def test_from_utils_template_calls_through(mocker, mock_get_service_letter_template):
|
|
mock_from_db = mocker.patch('app.template_previews.TemplatePreview.from_database_object')
|
|
template = LetterPreviewTemplate(mock_get_service_letter_template(None, None)['data'])
|
|
|
|
ret = TemplatePreview.from_utils_template(template, 'foo')
|
|
|
|
assert ret == mock_from_db.return_value
|
|
mock_from_db.assert_called_once_with(template._template, 'foo', template.values)
|
|
|
|
|
|
def test_from_database_object_makes_request(mocker, client):
|
|
resp = Mock(content='a', status_code='b', headers={'c': 'd'})
|
|
request_mock = mocker.patch('app.template_previews.requests.post', return_value=resp)
|
|
mocker.patch('app.template_previews.current_service', __getitem__=Mock(return_value='123'))
|
|
|
|
ret = TemplatePreview.from_database_object(template='foo', filetype='bar')
|
|
|
|
assert ret[0] == 'a'
|
|
assert ret[1] == 'b'
|
|
assert list(ret[2]) == [('c', 'd')]
|
|
url = 'http://localhost:6013/preview.bar'
|
|
data = {
|
|
'letter_contact_block': '123',
|
|
'template': 'foo',
|
|
'values': None
|
|
}
|
|
headers = {'Authorization': 'Token my-secret-key'}
|
|
|
|
request_mock.assert_called_once_with(url, json=data, headers=headers)
|