def test_create_distortion_pair_generates_distortion_set_id_if_none_provided(
self, cv_item_exists, cv_generic_save_view):
cv_item_exists.return_value = True
with current_app.test_request_context(
'/whatever',
headers={'Content-Type': 'application/json'},
data='{"start_x": "55"}'):
resp_object = cv.create_distortion_pair()
cv_item_exists.assert_called_once_with(ANY, 'distortion_set')
cv_generic_save_view.assert_called_once_with(
document_type='distortion_pair')
def test_create_distortion_pair_catches_exception(self, cv_item_exists,
cv_save_generic,
cv_generic_save_view):
cv_item_exists.side_effect = ThermalBaseError('zaboomafoo')
with current_app.test_request_context(
'/whatever',
headers={'Content-Type': 'application/json'},
data='{"distortion_set_id":"umma_gumma", "start_x": "55"}'):
resp_object = cv.create_distortion_pair()
cv_item_exists.assert_called_once_with('umma_gumma',
'distortion_set')
assert resp_object.data == '"zaboomafoo"'
assert resp_object.status_code == 400
def test_create_distortion_pair_uses_existing_distortion_set_id_if_none_provided_and_creates_one(
self, cv_item_exists, cv_save_generic, cv_generic_save_view):
cv_item_exists.return_value = False
with current_app.test_request_context(
'/whatever',
headers={'Content-Type': 'application/json'},
data='{"distortion_set_id":"umma_gumma", "start_x": "55"}'):
resp_object = cv.create_distortion_pair()
cv_item_exists.assert_called_once_with('umma_gumma',
'distortion_set')
cv_save_generic.assert_called_once_with(
{
'_id': 'umma_gumma',
'type': 'distortion_set'
}, 'distortion_set')
cv_generic_save_view.assert_called_once_with(
document_type='distortion_pair')
