浅谈pytest中重试机制的多种方法

pip install pytest-rerunfailures

pytest --reruns 3  # 对所有失败测试重试3次
pytest --reruns 3 --reruns-delay 2  # 重试3次，每次间隔2秒

@pytest.mark.flaky(reruns=3)
def test_example():
    assert 1 + 1 == 2

@pytest.mark.flaky(reruns=3, reruns_delay=1)
def test_example_with_delay():
    assert 2 * 2 == 4

pytest --reruns 1 --reruns-delay 1 -m flaky

pip install pytest-retry

@pytest.mark.retry(tries=3, delay=1)
def test_retry_specific():
    import random
    assert random.choice([True, False])

def pytest_runtest_makereport(item, call):
    if call.excinfo is not None:
        # 获取重试次数配置
        reruns = getattr(item, "execution_count", 1)
        if reruns > 1:
            # 实现重试逻辑
            pass

def retry(times=3, delay=1):
    def decorator(func):
        def wrapper(*args, ​**kwargs):
            for i in range(times):
                try:
                    return func(*args, ​**kwargs)
                except AssertionError as e:
                    if i == times - 1:
                        raise
                    time.sleep(delay)
        return wrapper
    return decorator

@retry(times=3, delay=0.5)
def test_custom_retry():
    assert False

@pytest.mark.flaky(reruns=3, condition=os.getenv("CI") == "true")
def test_conditional_retry():
    assert some_flaky_operation()

# 示例CI配置
pytest --reruns 3 --reruns-delay 1 --junitxml=report.xml