def test_call_distort_image_handles_exception(self, av_get_document_with_exception): av_get_document_with_exception.side_effect = ThermalBaseError('bill_dauterive') resp_object = av.call_distort_image(image_id='starting_image_id') assert resp_object.data == '"bill_dauterive"' assert resp_object.status_code == 400
def test_call_distort_image_handles_exception( self, av_get_document_with_exception): av_get_document_with_exception.side_effect = ThermalBaseError( 'bill_dauterive') resp_object = av.call_distort_image(image_id='starting_image_id') assert resp_object.data == '"bill_dauterive"' assert resp_object.status_code == 400
def test_call_distort_image_calls_expected_methods(self, av_get_document_with_exception, av_gather_and_enforce_request_args, as_distort_image_shepards): av_get_document_with_exception.return_value = {'buffalo': 'loin'} av_gather_and_enforce_request_args.return_value = {'distortion_set_id': 'some_ds_id'} response_object = av.call_distort_image(image_id='starting_image_id') resp_object_data = json.loads(response_object.data) assert 'distorted_image_id' in resp_object_data assert len(resp_object_data) == 1 assert response_object.status_code == 202 av_get_document_with_exception.assert_has_calls([call('starting_image_id', document_type='picture'), call('some_ds_id', document_type='distortion_set')]) as_distort_image_shepards.assert_called_once_with(image_id_in='starting_image_id', image_id_out=ANY, distortion_set_id='some_ds_id')
def test_call_distort_image_calls_expected_methods( self, av_get_document_with_exception, av_gather_and_enforce_request_args, as_distort_image_shepards): av_get_document_with_exception.return_value = {'buffalo': 'loin'} av_gather_and_enforce_request_args.return_value = { 'distortion_set_id': 'some_ds_id' } response_object = av.call_distort_image(image_id='starting_image_id') resp_object_data = json.loads(response_object.data) assert 'distorted_image_id' in resp_object_data assert len(resp_object_data) == 1 assert response_object.status_code == 202 av_get_document_with_exception.assert_has_calls([ call('starting_image_id', document_type='picture'), call('some_ds_id', document_type='distortion_set') ]) as_distort_image_shepards.assert_called_once_with( image_id_in='starting_image_id', image_id_out=ANY, distortion_set_id='some_ds_id')