2222nest_asyncio .apply ()
2323
2424
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 ):
2626 await fake_check (resource = 2 , resource_id = "c5187912-24a5-49ea-a725-5e1e3d472efe" )
2727 mocker .patch ("udata_hydra.config.BACKOFF_NB_REQ" , 1 )
2828 mocker .patch ("udata_hydra.config.BACKOFF_PERIOD" , 0.25 )
2929 rurl = RESOURCE_URL
3030 rmock .head (rurl , status = 200 )
31- event_loop . run_until_complete ( start_checks (iterations = 1 ) )
31+ await start_checks (iterations = 1 )
3232 # verify that we actually backed-off
3333 assert ("HEAD" , URL (rurl )) not in rmock .requests
3434
@@ -44,9 +44,7 @@ async def test_backoff_nb_req(setup_catalog, event_loop, rmock, mocker, fake_che
4444 (0 , - 1 , False ),
4545 ],
4646)
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 ):
5048 remain , limit , should_backoff = ratelimit
5149 await fake_check (
5250 resource = 2 ,
@@ -59,7 +57,7 @@ async def test_backoff_rate_limiting(
5957 rurl = RESOURCE_URL
6058 rmock .head (rurl , status = 200 )
6159 rmock .get (rurl , status = 200 )
62- event_loop . run_until_complete ( start_checks (iterations = 1 ) )
60+ await start_checks (iterations = 1 )
6361 # verify that we actually backed-off
6462 if should_backoff :
6563 assert ("HEAD" , URL (rurl )) not in rmock .requests
@@ -68,7 +66,7 @@ async def test_backoff_rate_limiting(
6866
6967
7068async 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
7270):
7371 await fake_check (
7472 resource = 2 ,
@@ -97,7 +95,7 @@ async def test_backoff_rate_limiting_lifted(
9795
9896
9997async 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
10199):
102100 await fake_check (
103101 resource = 2 ,
@@ -133,9 +131,7 @@ async def test_backoff_rate_limiting_cooled_off(
133131 assert ("HEAD" , URL (rurl )) in rmock .requests
134132
135133
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 ):
139135 await fake_check (resource = 2 , resource_id = "c5187912-24a5-49ea-a725-5e1e3d472efe" )
140136 mocker .patch ("udata_hydra.config.BACKOFF_NB_REQ" , 1 )
141137 mocker .patch ("udata_hydra.config.BACKOFF_PERIOD" , 0.25 )
@@ -155,7 +151,7 @@ async def test_backoff_nb_req_lifted(
155151
156152
157153async 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
159155):
160156 resource_id = "c5187912-24a5-49ea-a725-5e1e3d472efe"
161157 await fake_check (resource = 2 , resource_id = resource_id )
@@ -186,16 +182,14 @@ async def test_backoff_on_429_status_code(
186182 assert ("HEAD" , URL (rurl )) in rmock .requests
187183
188184
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 ):
192186 await fake_check (resource = 2 )
193187 mocker .patch ("udata_hydra.config.BACKOFF_NB_REQ" , 1 )
194188 mocker .patch ("udata_hydra.config.NO_BACKOFF_DOMAINS" , ["example.com" ])
195189 magic = MagicMock ()
196190 mocker .patch ("udata_hydra.context.monitor" ).return_value = magic
197191 rurl = RESOURCE_URL
198192 rmock .get (rurl , status = 200 )
199- event_loop . run_until_complete ( start_checks (iterations = 1 ) )
193+ await start_checks (iterations = 1 )
200194 # verify that we actually did not back-off
201195 assert not magic .add_backoff .called
0 commit comments