Skip to content

Commit bc61fee

Browse files
committed
Add test for RequestQueue.is_finished
1 parent ca72313 commit bc61fee

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

tests/integration/test_actor_request_queue.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,3 +86,27 @@ async def test_force_cloud(
8686
assert request_queue_request['url'] == 'http://example.com'
8787
finally:
8888
await request_queue_client.delete()
89+
90+
91+
async def test_request_queue_is_finished(
92+
apify_client_async: ApifyClientAsync,
93+
monkeypatch: pytest.MonkeyPatch,
94+
) -> None:
95+
assert apify_client_async.token is not None
96+
monkeypatch.setenv(ApifyEnvVars.TOKEN, apify_client_async.token)
97+
98+
request_queue_name = generate_unique_resource_name('request_queue')
99+
100+
async with Actor:
101+
request_queue = await Actor.open_request_queue(name=request_queue_name, force_cloud=True)
102+
await request_queue.add_request(Request.from_url('http://example.com'))
103+
assert not await request_queue.is_finished()
104+
105+
request = await request_queue.fetch_next_request()
106+
assert request is not None
107+
assert not await request_queue.is_finished(), (
108+
'RequestQueue should not be finished unless the request is marked as handled.'
109+
)
110+
111+
await request_queue.mark_request_as_handled(request)
112+
assert await request_queue.is_finished()

0 commit comments

Comments
 (0)