Skip to content

Commit e8a6186

Browse files
committed
More fixes
1 parent cc5b583 commit e8a6186

File tree

4 files changed

+17
-10
lines changed

4 files changed

+17
-10
lines changed

tests/integration/actor/conftest.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,9 @@ async def _make_actor(
261261
actor = actor_client.get()
262262

263263
if actor is not None and actor.pricing_infos is not None:
264-
new_pricing_infos = [*actor.pricing_infos, {'pricingModel': 'FREE'}]
264+
# Convert Pydantic models to dicts before mixing with plain dict
265+
existing_pricing_infos = [pi.model_dump(by_alias=True, exclude_none=True) for pi in actor.pricing_infos]
266+
new_pricing_infos = [*existing_pricing_infos, {'pricingModel': 'FREE'}]
265267
actor_client.update(pricing_infos=new_pricing_infos)
266268

267269
actor_client.delete()

tests/integration/actor/test_actor_charge.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@ async def main() -> None:
3535
actor_client = await make_actor('ppe', main_func=main)
3636

3737
await actor_client.update(
38-
default_run_build='latest',
3938
pricing_infos=[
4039
{
4140
'pricingModel': 'PAY_PER_EVENT',

tests/integration/apify_api/test_request_queue.py

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
import pytest
1010

11-
from apify_client._models import Data13, ProcessedRequest, UnprocessedRequest
11+
from apify_client._models import Data13, UnprocessedRequest
1212
from apify_shared.consts import ApifyEnvVars
1313
from crawlee import service_locator
1414
from crawlee.crawlers import BasicCrawler
@@ -1171,12 +1171,18 @@ async def test_request_queue_deduplication_unprocessed_requests(
11711171

11721172
def return_unprocessed_requests(requests: list[dict], *_: Any, **__: Any) -> Data13:
11731173
"""Simulate API returning unprocessed requests."""
1174-
return Data13(
1175-
processed_requests=list[ProcessedRequest](),
1176-
unprocessed_requests=[
1177-
UnprocessedRequest(url=request['url'], unique_key=request['uniqueKey'], method=request['method'])
1178-
for request in requests
1179-
],
1174+
unprocessed_requests = [
1175+
UnprocessedRequest.model_construct(
1176+
url=request['url'],
1177+
unique_key=request['uniqueKey'],
1178+
method=request['method'],
1179+
)
1180+
for request in requests
1181+
]
1182+
1183+
return Data13.model_construct(
1184+
processed_requests=[],
1185+
unprocessed_requests=unprocessed_requests,
11801186
)
11811187

11821188
with mock.patch(

uv.lock

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)