Skip to content

Commit 2e0ce25

Browse files
authored
chore: fix tests for ApifyRequestList (#727)
Add `crawlee_data` with `CrawleeRequestData` in `expected_user_data`. Closes: #726
1 parent 4e6650c commit 2e0ce25

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

tests/unit/actor/test_request_list.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import pytest
99
from yarl import URL
1010

11-
from crawlee._request import UserData
11+
from crawlee._request import CrawleeRequestData, UserData
1212
from crawlee._types import HttpMethod
1313

1414
from apify.request_loaders import ApifyRequestList
@@ -65,6 +65,8 @@ async def test_request_list_open_request_types(
6565
assert request.url == request_dict_input['url']
6666
assert request.payload == request_dict_input.get('payload', '').encode('utf-8')
6767
expected_user_data = UserData()
68+
# `crawlee_data` must be present in user_data
69+
expected_user_data.crawlee_data = CrawleeRequestData()
6870
if 'userData' in optional_input:
6971
for key, value in optional_input['userData'].items():
7072
expected_user_data[key] = value
@@ -181,6 +183,8 @@ async def test_request_list_open_from_url_additional_inputs(httpserver: HTTPServ
181183
assert request.headers.root == example_start_url_input['headers']
182184
assert request.payload == str(example_start_url_input['payload']).encode('utf-8')
183185
expected_user_data = UserData()
186+
# `crawlee_data` must be present in user_data
187+
expected_user_data.crawlee_data = CrawleeRequestData()
184188
for key, value in example_start_url_input['userData'].items():
185189
expected_user_data[key] = value
186190
assert request.user_data == expected_user_data

uv.lock

Lines changed: 3 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)