Skip to content

Commit ec2a9f0

Browse files
committed
Fix 2 tests in RQ Apify storage client
1 parent 9edac0f commit ec2a9f0

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/apify/storage_clients/_apify/_request_queue_client.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ async def get_request(self, request_id: str) -> Request | None:
215215
if response is None:
216216
return None
217217

218-
return Request.model_validate(**response)
218+
return Request.model_validate(response)
219219

220220
@override
221221
async def fetch_next_request(self) -> Request | None:
@@ -256,6 +256,15 @@ async def fetch_next_request(self) -> Request | None:
256256
)
257257
return None
258258

259+
# Use get request to ensure we have the full request object.
260+
request = await self.get_request(request.id)
261+
if request is None:
262+
logger.debug(
263+
'Request fetched from the beginning of queue was not found in the RQ',
264+
extra={'nextRequestId': next_request_id},
265+
)
266+
return None
267+
259268
return request
260269

261270
@override

0 commit comments

Comments
 (0)