Skip to content

Commit e5bdff2

Browse files
committed
Update tests
1 parent ffa70ff commit e5bdff2

File tree

2 files changed

+5
-11
lines changed

2 files changed

+5
-11
lines changed

tests/integration/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def apify_client_async(apify_token: str) -> ApifyClientAsync:
107107
return ApifyClientAsync(apify_token, api_url=api_url)
108108

109109

110-
@pytest.fixture(params=[['single', 'shared']])
110+
@pytest.fixture(params=['single', 'shared'])
111111
async def request_queue_apify(
112112
apify_token: str, monkeypatch: pytest.MonkeyPatch, request: pytest.FixtureRequest
113113
) -> AsyncGenerator[RequestQueue]:

tests/integration/test_request_queue.py

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1174,24 +1174,18 @@ async def test_cache_initialization(apify_token: str, monkeypatch: pytest.Monkey
11741174
await rq.drop()
11751175

11761176

1177-
async def test_request_queue_has_stats(request_queue_force_cloud: RequestQueue) -> None:
1177+
async def test_request_queue_has_stats(request_queue_apify: RequestQueue) -> None:
11781178
"""Test that Apify based request queue has stats in metadata."""
1179-
1179+
rq = request_queue_apify
11801180
add_request_count = 3
1181-
read_request_count = 2
11821181

1183-
await request_queue_force_cloud.add_requests(
1184-
[Request.from_url(f'http://example.com/{i}') for i in range(add_request_count)]
1185-
)
1186-
for _ in range(read_request_count):
1187-
await request_queue_force_cloud.get_request(Request.from_url('http://example.com/1').unique_key)
1182+
await rq.add_requests([Request.from_url(f'http://example.com/{i}') for i in range(add_request_count)])
11881183

11891184
# Wait for stats to become stable
11901185
await asyncio.sleep(10)
11911186

1192-
metadata = await request_queue_force_cloud.get_metadata()
1187+
metadata = await rq.get_metadata()
11931188

11941189
assert hasattr(metadata, 'stats')
11951190
apify_metadata = cast('ApifyRequestQueueMetadata', metadata)
1196-
assert apify_metadata.stats.read_count == read_request_count
11971191
assert apify_metadata.stats.write_count == add_request_count

0 commit comments

Comments
 (0)