Skip to content

Commit 91ff3fd

Browse files
committed
Format and type check
1 parent 70908b3 commit 91ff3fd

File tree

2 files changed

+11
-4
lines changed

2 files changed

+11
-4
lines changed

src/apify/storage_clients/_apify/_request_queue_client.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -240,7 +240,7 @@ async def add_batch_of_requests(
240240
"""
241241
# Do not try to add previously added requests to avoid pointless expensive calls to API
242242
new_requests: list[Request] = []
243-
already_present_requests: list[ProcessedRequest] = []
243+
already_present_requests: list[dict[str, str | bool]] = []
244244
for request in requests:
245245
if self._requests_cache.get(request.id):
246246
# We are no sure if it was already handled at this point, and it is not worth calling API for it.

tests/integration/test_actor_request_queue.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -129,18 +129,25 @@ async def main() -> None:
129129

130130
async with Actor:
131131
request = Request.from_url('http://example.com')
132-
rq = await Actor.open_request_queue(name='test-deduplication', force_cloud=True)
132+
rq = await Actor.open_request_queue()
133133

134134
await asyncio.sleep(10) # Wait to be sure that metadata are updated
135-
stats_before = (await Actor.apify_client.request_queue(request_queue_id=rq.id).get()).get('stats', {})
135+
136+
# Get raw client, because stats are not exposed in `RequestQueue` class, but are available in raw client
137+
rq_client = Actor.apify_client.request_queue(request_queue_id=rq.id)
138+
_rq = await rq_client.get()
139+
assert _rq
140+
stats_before = _rq.get('stats', {})
136141
Actor.log.info(stats_before)
137142

138143
# Add same request twice
139144
await rq.add_request(request)
140145
await rq.add_request(request)
141146

142147
await asyncio.sleep(10) # Wait to be sure that metadata are updated
143-
stats_after = (await Actor.apify_client.request_queue(request_queue_id=rq.id).get()).get('stats', {})
148+
_rq = await rq_client.get()
149+
assert _rq
150+
stats_after = _rq.get('stats', {})
144151
Actor.log.info(stats_after)
145152

146153
assert (stats_after['writeCount'] - stats_before['writeCount']) == 1

0 commit comments

Comments
 (0)