22
22
nest_asyncio .apply ()
23
23
24
24
25
- async def test_backoff_nb_req (setup_catalog , event_loop , rmock , mocker , fake_check , produce_mock ):
25
+ async def test_backoff_nb_req (setup_catalog , rmock , mocker , fake_check , produce_mock ):
26
26
await fake_check (resource = 2 , resource_id = "c5187912-24a5-49ea-a725-5e1e3d472efe" )
27
27
mocker .patch ("udata_hydra.config.BACKOFF_NB_REQ" , 1 )
28
28
mocker .patch ("udata_hydra.config.BACKOFF_PERIOD" , 0.25 )
29
29
rurl = RESOURCE_URL
30
30
rmock .head (rurl , status = 200 )
31
- event_loop . run_until_complete ( start_checks (iterations = 1 ) )
31
+ await start_checks (iterations = 1 )
32
32
# verify that we actually backed-off
33
33
assert ("HEAD" , URL (rurl )) not in rmock .requests
34
34
@@ -44,9 +44,7 @@ async def test_backoff_nb_req(setup_catalog, event_loop, rmock, mocker, fake_che
44
44
(0 , - 1 , False ),
45
45
],
46
46
)
47
- async def test_backoff_rate_limiting (
48
- setup_catalog , event_loop , rmock , fake_check , produce_mock , ratelimit
49
- ):
47
+ async def test_backoff_rate_limiting (setup_catalog , rmock , fake_check , produce_mock , ratelimit ):
50
48
remain , limit , should_backoff = ratelimit
51
49
await fake_check (
52
50
resource = 2 ,
@@ -59,7 +57,7 @@ async def test_backoff_rate_limiting(
59
57
rurl = RESOURCE_URL
60
58
rmock .head (rurl , status = 200 )
61
59
rmock .get (rurl , status = 200 )
62
- event_loop . run_until_complete ( start_checks (iterations = 1 ) )
60
+ await start_checks (iterations = 1 )
63
61
# verify that we actually backed-off
64
62
if should_backoff :
65
63
assert ("HEAD" , URL (rurl )) not in rmock .requests
@@ -68,7 +66,7 @@ async def test_backoff_rate_limiting(
68
66
69
67
70
68
async def test_backoff_rate_limiting_lifted (
71
- setup_catalog , event_loop , rmock , mocker , fake_check , produce_mock , db
69
+ setup_catalog , rmock , mocker , fake_check , produce_mock , db
72
70
):
73
71
await fake_check (
74
72
resource = 2 ,
@@ -97,7 +95,7 @@ async def test_backoff_rate_limiting_lifted(
97
95
98
96
99
97
async def test_backoff_rate_limiting_cooled_off (
100
- setup_catalog , event_loop , rmock , mocker , fake_check , produce_mock , db
98
+ setup_catalog , rmock , mocker , fake_check , produce_mock , db
101
99
):
102
100
await fake_check (
103
101
resource = 2 ,
@@ -133,9 +131,7 @@ async def test_backoff_rate_limiting_cooled_off(
133
131
assert ("HEAD" , URL (rurl )) in rmock .requests
134
132
135
133
136
- async def test_backoff_nb_req_lifted (
137
- setup_catalog , event_loop , rmock , mocker , fake_check , produce_mock , db
138
- ):
134
+ async def test_backoff_nb_req_lifted (setup_catalog , rmock , mocker , fake_check , produce_mock , db ):
139
135
await fake_check (resource = 2 , resource_id = "c5187912-24a5-49ea-a725-5e1e3d472efe" )
140
136
mocker .patch ("udata_hydra.config.BACKOFF_NB_REQ" , 1 )
141
137
mocker .patch ("udata_hydra.config.BACKOFF_PERIOD" , 0.25 )
@@ -155,7 +151,7 @@ async def test_backoff_nb_req_lifted(
155
151
156
152
157
153
async def test_backoff_on_429_status_code (
158
- setup_catalog , event_loop , rmock , mocker , fake_check , produce_mock , db
154
+ setup_catalog , rmock , mocker , fake_check , produce_mock , db
159
155
):
160
156
resource_id = "c5187912-24a5-49ea-a725-5e1e3d472efe"
161
157
await fake_check (resource = 2 , resource_id = resource_id )
@@ -186,16 +182,14 @@ async def test_backoff_on_429_status_code(
186
182
assert ("HEAD" , URL (rurl )) in rmock .requests
187
183
188
184
189
- async def test_no_backoff_domains (
190
- setup_catalog , event_loop , rmock , mocker , fake_check , produce_mock
191
- ):
185
+ async def test_no_backoff_domains (setup_catalog , rmock , mocker , fake_check , produce_mock ):
192
186
await fake_check (resource = 2 )
193
187
mocker .patch ("udata_hydra.config.BACKOFF_NB_REQ" , 1 )
194
188
mocker .patch ("udata_hydra.config.NO_BACKOFF_DOMAINS" , ["example.com" ])
195
189
magic = MagicMock ()
196
190
mocker .patch ("udata_hydra.context.monitor" ).return_value = magic
197
191
rurl = RESOURCE_URL
198
192
rmock .get (rurl , status = 200 )
199
- event_loop . run_until_complete ( start_checks (iterations = 1 ) )
193
+ await start_checks (iterations = 1 )
200
194
# verify that we actually did not back-off
201
195
assert not magic .add_backoff .called
0 commit comments