mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 12:05:07 +00:00
imSp4rky
89f433963a
A per-call max_retries=0 logged "Retrying... (1/0)" before raising, and the last of the default retries logged "(6/5)".
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
"""RnetSession.request logs "Retrying" only when a retry follows."""
|
|
|
|
import logging
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from unshackle.core import session as session_module
|
|
from unshackle.core.session import MaxRetriesError, RnetSession
|
|
|
|
|
|
@pytest.fixture
|
|
def rate_limited(monkeypatch):
|
|
s = RnetSession()
|
|
monkeypatch.setattr(s, "ensure_client", lambda: SimpleNamespace(request=lambda *a, **k: None))
|
|
monkeypatch.setattr(s, "get_sleep_time", lambda *a: 0)
|
|
monkeypatch.setattr(
|
|
session_module, "RnetResponse", lambda raw: SimpleNamespace(status_code=429, reason="Too Many Requests")
|
|
)
|
|
return s
|
|
|
|
|
|
@pytest.mark.parametrize("max_retries", [0, 2])
|
|
def test_retry_log_counts_only_real_retries(rate_limited, caplog, max_retries):
|
|
with caplog.at_level(logging.WARNING), pytest.raises(MaxRetriesError):
|
|
rate_limited.get("https://example.invalid/x", max_retries=max_retries)
|
|
|
|
retries = [r.getMessage() for r in caplog.records if "Retrying" in r.getMessage()]
|
|
assert retries == [f"429 Too Many Requests(/x). Retrying... ({n}/{max_retries})" for n in range(1, max_retries + 1)]
|