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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
|
try:
from unittest.mock import create_autospec
except ImportError:
from mock import create_autospec
try:
from unittest.mock import MagicMock
except ImportError:
from mock import MagicMock
import time
import pytest
from retry.api import retry_call
from retry.api import retry
def test_retry(monkeypatch):
mock_sleep_time = [0]
def mock_sleep(seconds):
mock_sleep_time[0] += seconds
monkeypatch.setattr(time, 'sleep', mock_sleep)
hit = [0]
tries = 5
delay = 1
backoff = 2
@retry(tries=tries, delay=delay, backoff=backoff)
def f():
hit[0] += 1
1 / 0
with pytest.raises(ZeroDivisionError):
f()
assert hit[0] == tries
assert mock_sleep_time[0] == sum(
delay * backoff ** i for i in range(tries - 1))
def test_tries_inf():
hit = [0]
target = 10
@retry(tries=float('inf'))
def f():
hit[0] += 1
if hit[0] == target:
return target
else:
raise ValueError
assert f() == target
def test_tries_minus1():
hit = [0]
target = 10
@retry(tries=-1)
def f():
hit[0] += 1
if hit[0] == target:
return target
else:
raise ValueError
assert f() == target
def test_max_delay(monkeypatch):
mock_sleep_time = [0]
def mock_sleep(seconds):
mock_sleep_time[0] += seconds
monkeypatch.setattr(time, 'sleep', mock_sleep)
hit = [0]
tries = 5
delay = 1
backoff = 2
max_delay = delay # Never increase delay
@retry(tries=tries, delay=delay, max_delay=max_delay, backoff=backoff)
def f():
hit[0] += 1
1 / 0
with pytest.raises(ZeroDivisionError):
f()
assert hit[0] == tries
assert mock_sleep_time[0] == delay * (tries - 1)
def test_fixed_jitter(monkeypatch):
mock_sleep_time = [0]
def mock_sleep(seconds):
mock_sleep_time[0] += seconds
monkeypatch.setattr(time, 'sleep', mock_sleep)
hit = [0]
tries = 10
jitter = 1
@retry(tries=tries, jitter=jitter)
def f():
hit[0] += 1
1 / 0
with pytest.raises(ZeroDivisionError):
f()
assert hit[0] == tries
assert mock_sleep_time[0] == sum(range(tries - 1))
def test_retry_call():
f_mock = MagicMock(side_effect=RuntimeError)
tries = 2
try:
retry_call(f_mock, exceptions=RuntimeError, tries=tries)
except RuntimeError:
pass
assert f_mock.call_count == tries
def test_retry_call_2():
side_effect = [RuntimeError, RuntimeError, 3]
f_mock = MagicMock(side_effect=side_effect)
tries = 5
result = None
try:
result = retry_call(f_mock, exceptions=RuntimeError, tries=tries)
except RuntimeError:
pass
assert result == 3
assert f_mock.call_count == len(side_effect)
def test_retry_call_with_args():
def f(value=0):
if value < 0:
return value
else:
raise RuntimeError
return_value = -1
result = None
f_mock = MagicMock(spec=f, return_value=return_value)
try:
result = retry_call(f_mock, fargs=[return_value])
except RuntimeError:
pass
assert result == return_value
assert f_mock.call_count == 1
def test_retry_call_with_kwargs():
def f(value=0):
if value < 0:
return value
else:
raise RuntimeError
kwargs = {'value': -1}
result = None
f_mock = MagicMock(spec=f, return_value=kwargs['value'])
try:
result = retry_call(f_mock, fkwargs=kwargs)
except RuntimeError:
pass
assert result == kwargs['value']
assert f_mock.call_count == 1
|
