def test_handle_validate_and_initialize_error(env, validate_user_module):
data = [{"body": INPUT_DATA}]
request_processor = Mock()
context = Mock()
context.request_processor = [request_processor]
transform_fn = Mock()
model_fn = Mock()
transformer = Transformer()
transformer._model = MODEL
transformer._transform_fn = transform_fn
transformer._model_fn = model_fn
test_error_message = "Foo"
validate_user_module.side_effect = ValueError(test_error_message)
assert transformer._initialized is False
response = transformer.transform(data, context)
assert test_error_message in str(response)
context.set_response_status.assert_called_with(
code=http_client.INTERNAL_SERVER_ERROR, phrase=test_error_message)
def test_handle_validate_and_initialize_user_error(env, validate_user_module):
test_status_code = http_client.FORBIDDEN
test_error_message = "Foo"
class FooUserError(BaseInferenceToolkitError):
def __init__(self, status_code, message):
self.status_code = status_code
self.message = message
self.phrase = "Foo"
data = [{"body": INPUT_DATA}]
context = Mock()
transform_fn = Mock()
model_fn = Mock()
transformer = Transformer()
transformer._model = MODEL
transformer._transform_fn = transform_fn
transformer._model_fn = model_fn
validate_user_module.side_effect = FooUserError(test_status_code,
test_error_message)
assert transformer._initialized is False
response = transformer.transform(data, context)
assert test_error_message in str(response)
context.set_response_status.assert_called_with(code=http_client.FORBIDDEN,
phrase=test_error_message)
def test_validate_and_initialize(env, validate_user_module):
transformer = Transformer()
model_fn = Mock()
transformer._model_fn = model_fn
assert transformer._initialized is False
transformer.validate_and_initialize()
assert transformer._initialized is True
transformer.validate_and_initialize()
model_fn.assert_called_once_with(environment.model_dir)
env.assert_called_once_with()
validate_user_module.assert_called_once_with()
