1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
import time
from unittest import mock
import pytest
from gitlab import utils
def test_handle_retry_on_status_ignores_unknown_status_code():
retry = utils.Retry(max_retries=1, retry_transient_errors=True)
assert retry.handle_retry_on_status(418) is False
def test_handle_retry_on_status_accepts_retry_after_header(
monkeypatch: pytest.MonkeyPatch,
):
mock_sleep = mock.Mock()
monkeypatch.setattr(time, "sleep", mock_sleep)
retry = utils.Retry(max_retries=1)
headers = {"Retry-After": "1"}
assert retry.handle_retry_on_status(429, headers=headers) is True
assert isinstance(mock_sleep.call_args[0][0], int)
def test_handle_retry_on_status_accepts_ratelimit_reset_header(
monkeypatch: pytest.MonkeyPatch,
):
mock_sleep = mock.Mock()
monkeypatch.setattr(time, "sleep", mock_sleep)
retry = utils.Retry(max_retries=1)
headers = {"RateLimit-Reset": str(int(time.time() + 1))}
assert retry.handle_retry_on_status(429, headers=headers) is True
assert isinstance(mock_sleep.call_args[0][0], float)
def test_handle_retry_on_status_returns_false_when_max_retries_reached():
retry = utils.Retry(max_retries=0)
assert retry.handle_retry_on_status(429) is False
|
