FilesExpand file tree

 main

/

test_retry.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

41 lines (28 loc) · 1.2 KB

 main

/

test_retry.py

Top

File metadata and controls

• Code
• Blame

41 lines (28 loc) · 1.2 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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