Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion tests/unit/actor/test_request_list.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
import pytest
from yarl import URL

from crawlee._request import UserData
from crawlee._request import CrawleeRequestData, UserData
from crawlee._types import HttpMethod

from apify.request_loaders import ApifyRequestList
Expand Down Expand Up @@ -65,6 +65,8 @@ async def test_request_list_open_request_types(
assert request.url == request_dict_input['url']
assert request.payload == request_dict_input.get('payload', '').encode('utf-8')
expected_user_data = UserData()
# `crawlee_data` must be present in user_data
expected_user_data.crawlee_data = CrawleeRequestData()
if 'userData' in optional_input:
for key, value in optional_input['userData'].items():
expected_user_data[key] = value
Expand Down Expand Up @@ -181,6 +183,8 @@ async def test_request_list_open_from_url_additional_inputs(httpserver: HTTPServ
assert request.headers.root == example_start_url_input['headers']
assert request.payload == str(example_start_url_input['payload']).encode('utf-8')
expected_user_data = UserData()
# `crawlee_data` must be present in user_data
expected_user_data.crawlee_data = CrawleeRequestData()
for key, value in example_start_url_input['userData'].items():
expected_user_data[key] = value
assert request.user_data == expected_user_data
Expand Down
6 changes: 3 additions & 3 deletions uv.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading