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()