Skip to content

Commit 8c286c0

Browse files
committed
Address review feedback
1 parent 54dba8f commit 8c286c0

1 file changed

Lines changed: 9 additions & 34 deletions

File tree

tests/integration/test_request_queue.py

Lines changed: 9 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -77,12 +77,7 @@ async def test_request_queue_lock(client: ApifyClient | ApifyClientAsync, *, is_
7777
# Add requests and check if correct number of requests was locked
7878
for i in range(15):
7979
await maybe_await(
80-
rq.add_request(
81-
cast(
82-
'RequestInputDict',
83-
{'url': f'http://test-lock.com/{i}', 'unique_key': f'http://test-lock.com/{i}'},
84-
)
85-
)
80+
rq.add_request({'url': f'http://test-lock.com/{i}', 'unique_key': f'http://test-lock.com/{i}'})
8681
)
8782

8883
# Poll until all requests are available for locking (eventual consistency)
@@ -227,9 +222,7 @@ async def test_request_queue_list_head(client: ApifyClient | ApifyClientAsync, *
227222
# Add multiple requests
228223
for i in range(5):
229224
await maybe_await(
230-
rq_client.add_request(
231-
cast('RequestInputDict', {'url': f'https://example.com/page-{i}', 'unique_key': f'page-{i}'})
232-
)
225+
rq_client.add_request({'url': f'https://example.com/page-{i}', 'unique_key': f'page-{i}'})
233226
)
234227

235228
# Poll until requests are available (eventual consistency)
@@ -259,9 +252,7 @@ async def test_request_queue_list_requests(client: ApifyClient | ApifyClientAsyn
259252
# Add multiple requests
260253
for i in range(5):
261254
await maybe_await(
262-
rq_client.add_request(
263-
cast('RequestInputDict', {'url': f'https://example.com/item-{i}', 'unique_key': f'item-{i}'})
264-
)
255+
rq_client.add_request({'url': f'https://example.com/item-{i}', 'unique_key': f'item-{i}'})
265256
)
266257

267258
# Poll until all requests are available (eventual consistency)
@@ -290,9 +281,7 @@ async def test_request_queue_delete_request(client: ApifyClient | ApifyClientAsy
290281
try:
291282
# Add a request
292283
result = await maybe_await(
293-
rq_client.add_request(
294-
cast('RequestInputDict', {'url': 'https://example.com/to-delete', 'unique_key': 'delete-me'})
295-
)
284+
rq_client.add_request({'url': 'https://example.com/to-delete', 'unique_key': 'delete-me'})
296285
)
297286
add_result = cast('RequestRegistration', result)
298287

@@ -362,9 +351,7 @@ async def test_request_queue_batch_delete_requests(client: ApifyClient | ApifyCl
362351
# Add requests
363352
for i in range(10):
364353
await maybe_await(
365-
rq_client.add_request(
366-
cast('RequestInputDict', {'url': f'https://example.com/delete-{i}', 'unique_key': f'delete-{i}'})
367-
)
354+
rq_client.add_request({'url': f'https://example.com/delete-{i}', 'unique_key': f'delete-{i}'})
368355
)
369356

370357
# Poll until all requests are available (eventual consistency)
@@ -431,9 +418,7 @@ async def test_request_queue_list_and_lock_head(client: ApifyClient | ApifyClien
431418
# Add multiple requests
432419
for i in range(5):
433420
await maybe_await(
434-
rq_client.add_request(
435-
cast('RequestInputDict', {'url': f'https://example.com/lock-{i}', 'unique_key': f'lock-{i}'})
436-
)
421+
rq_client.add_request({'url': f'https://example.com/lock-{i}', 'unique_key': f'lock-{i}'})
437422
)
438423

439424
# Poll until requests are available for locking (eventual consistency)
@@ -466,11 +451,7 @@ async def test_request_queue_prolong_request_lock(client: ApifyClient | ApifyCli
466451

467452
try:
468453
# Add a request
469-
await maybe_await(
470-
rq_client.add_request(
471-
cast('RequestInputDict', {'url': 'https://example.com/prolong', 'unique_key': 'prolong-test'})
472-
)
473-
)
454+
await maybe_await(rq_client.add_request({'url': 'https://example.com/prolong', 'unique_key': 'prolong-test'}))
474455

475456
# Poll until the request is available for locking (eventual consistency)
476457
lock_response: LockedRequestQueueHead | None = None
@@ -508,11 +489,7 @@ async def test_request_queue_delete_request_lock(client: ApifyClient | ApifyClie
508489

509490
try:
510491
# Add a request
511-
await maybe_await(
512-
rq_client.add_request(
513-
cast('RequestInputDict', {'url': 'https://example.com/unlock', 'unique_key': 'unlock-test'})
514-
)
515-
)
492+
await maybe_await(rq_client.add_request({'url': 'https://example.com/unlock', 'unique_key': 'unlock-test'}))
516493

517494
# Poll until the request is available for locking (eventual consistency)
518495
lock_response: LockedRequestQueueHead | None = None
@@ -550,9 +527,7 @@ async def test_request_queue_unlock_requests(client: ApifyClient | ApifyClientAs
550527
# Add multiple requests
551528
for i in range(5):
552529
await maybe_await(
553-
rq_client.add_request(
554-
cast('RequestInputDict', {'url': f'https://example.com/unlock-{i}', 'unique_key': f'unlock-{i}'})
555-
)
530+
rq_client.add_request({'url': f'https://example.com/unlock-{i}', 'unique_key': f'unlock-{i}'})
556531
)
557532

558533
# Poll until requests are available for locking (eventual consistency)

0 commit comments

Comments
 (0)