def test_circuit_decorator_with_args(circuitbreaker_mock):
circuitbreaker_mock.return_value = None
circuit(10, 20, KeyError, 'foobar')
circuitbreaker_mock.assert_called_once_with(expected_exception=KeyError,
failure_threshold=10,
recovery_timeout=20,
name='foobar')
def test_circuit_decorator_with_args(circuitbreaker_mock):
circuitbreaker_mock.return_value = None
circuit(10, 20, KeyError, 'foobar')
circuitbreaker_mock.assert_called_once_with(
expected_exception=KeyError,
failure_threshold=10,
recovery_timeout=20,
name='foobar'
)
def test_circuitbreaker_call_fallback_function_with_parameters():
fallback = Mock(return_value=True)
cb = circuit(name='with_fallback', fallback_function=fallback)
# mock opened prop to see if fallback is called with correct parameters.
cb.opened = lambda self: True
func_decorated = cb.decorate(mocked_function)
func_decorated('test2', test='test')
# check args and kwargs are getting correctly to fallback function
fallback.assert_called_once_with('test2', test='test')
def test_circuit_decorator_without_args(circuitbreaker_mock):
function = lambda: True
circuit(function)
circuitbreaker_mock.assert_called_once_with(function)
def test_circuit_decorator_without_args(circuitbreaker_mock):
function = lambda: True
circuit(function)
circuitbreaker_mock.assert_called_once_with(function)