def test_handle_expected_exceptions_throw():
mock_method = MagicMock(side_effect=ValueError('HALP'))
mock_method.__name__ = 'mock_meth'
decorated = handle_expected_exceptions(mock_method)
assert_equals(decorated.__name__, mock_method.__name__)
result = decorated(self, 1, kwarg='foo')
def test_handle_expected_exceptions():
mock_method = MagicMock()
mock_method.__name__ = 'MockMethod'
decorated = handle_expected_exceptions(mock_method)
assert_equals(decorated.__name__, mock_method.__name__)
result = decorated(self, 1, 2, 3)
assert_equals(result, mock_method.return_value)
assert_equals(ipython_display.send_error.call_count, 0)
mock_method.assert_called_once_with(self, 1, 2, 3)
def test_handle_expected_exceptions_handle():
mock_method = MagicMock(
side_effect=LivyUnexpectedStatusException('ridiculous'))
mock_method.__name__ = 'MockMethod2'
decorated = handle_expected_exceptions(mock_method)
assert_equals(decorated.__name__, mock_method.__name__)
result = decorated(self, 1, kwarg='foo')
assert_is(result, None)
assert_equals(ipython_display.send_error.call_count, 1)
mock_method.assert_called_once_with(self, 1, kwarg='foo')