예제 #1
0
def test_backoff_basic(failure, strategy, max_tries, max_delay, retry_execute):
    """Test the :ref:`backoff_utils._backoff.backoff` function."""
    global _attempts  # pylint: disable=W0603,C0103
    if not failure:
        with pytest.raises(ZeroDivisionError) as excinfo:
            start_time = datetime.utcnow()
            backoff(to_execute=divide_by_zero_function,
                    args=[False],
                    kwargs=None,
                    strategy=strategy,
                    retry_execute=retry_execute,
                    retry_args=[True],
                    retry_kwargs=None,
                    max_tries=max_tries,
                    max_delay=max_delay,
                    catch_exceptions=[type(ZeroDivisionError())],
                    on_failure=None,
                    on_success=None)
        end_time = datetime.utcnow()
        elapsed_time = start_time - end_time
        elapsed_time = elapsed_time.total_seconds()
        if max_delay is not None:
            assert elapsed_time <= max_delay
            assert _attempts <= max_tries
        else:
            assert _attempts == max_tries
        assert 'Subsequent Attempt' in str(excinfo.value)
    else:
        with pytest.raises(failure):
            start_time = datetime.utcnow()
            backoff(to_execute=divide_by_zero_function,
                    args=[False],
                    kwargs=None,
                    strategy=strategy,
                    retry_execute=retry_execute,
                    retry_args=[True],
                    retry_kwargs=None,
                    max_tries=max_tries,
                    max_delay=max_delay,
                    catch_exceptions=[type(ZeroDivisionError())],
                    on_failure=None,
                    on_success=None)
        end_time = datetime.utcnow()
        elapsed_time = start_time - end_time
        elapsed_time = elapsed_time.total_seconds()

    _attempts = 0
예제 #2
0
 def wrapper(*args, **kwargs):
     return backoff(to_execute=func,
                    args=args,
                    kwargs=kwargs,
                    strategy=strategy,
                    max_tries=max_tries,
                    max_delay=max_delay,
                    catch_exceptions=catch_exceptions,
                    on_failure=on_failure,
                    on_success=on_success)
예제 #3
0
def test_backoff_on_failure_function(strategy, max_tries, on_failure):
    """Test the :ref:`backoff_utils._backoff.backoff` function."""
    global _attempts  # pylint: disable=W0603,C0103
    global _was_successful  # pylint: disable=W0603,C0103

    with pytest.raises(AttributeError) as excinfo:
        backoff(to_execute=divide_by_zero_function,
                args=[False],
                kwargs=None,
                strategy=strategy,
                retry_execute=None,
                retry_args=[True],
                retry_kwargs=None,
                max_tries=max_tries,
                catch_exceptions=[type(ZeroDivisionError())],
                on_failure=on_failure,
                on_success=None)
    assert _attempts == max_tries
    assert _was_successful is False
    _attempts = 0
    _was_successful = False
예제 #4
0
def test_backoff_on_success(strategy, max_tries, on_success):
    """Test the :ref:`backoff_utils._backoff.backoff` function."""
    global _attempts  # pylint: disable=W0603,C0103
    global _was_successful  # pylint: disable=W0603,C0103

    return_value = backoff(to_execute=successful_function,
                           args=[False, max_tries],
                           kwargs=None,
                           strategy=strategy,
                           retry_execute=None,
                           retry_args=[True, max_tries],
                           retry_kwargs=None,
                           max_tries=max_tries,
                           catch_exceptions=[type(ZeroDivisionError())],
                           on_failure=None,
                           on_success=on_success)
    assert _attempts == max_tries
    assert _was_successful is True
    assert return_value == successful_function(True, max_tries)
    _attempts = 0
    _was_successful = False