136 lines
5.2 KiB
Python
136 lines
5.2 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from typing import Any, cast
|
|
from unittest.mock import Mock
|
|
|
|
from . import gateway as gateway_module
|
|
from . import xiaohongshu as xiaohongshu_module
|
|
|
|
Gateway = gateway_module.Gateway
|
|
RequestError = gateway_module.RequestError
|
|
valid_xhs_post_url = gateway_module.valid_xhs_post_url
|
|
valid_xhs_url = gateway_module.valid_xhs_url
|
|
valid_xiaohongshu_url = gateway_module.valid_xiaohongshu_url
|
|
valid_xiaohongshu_media_url = gateway_module.valid_xiaohongshu_media_url
|
|
valid_xiaohongshu_generation = gateway_module.valid_xiaohongshu_generation
|
|
valid_xiaohongshu_page_url = gateway_module.valid_xiaohongshu_page_url
|
|
valid_xiaohongshu_source_url = gateway_module.valid_xiaohongshu_source_url
|
|
|
|
|
|
class XiaohongshuValidationTests(unittest.TestCase):
|
|
def test_read_urls_use_explicit_host_path_and_query_allowlist(self) -> None:
|
|
self.assertTrue(
|
|
valid_xhs_url("https://edith.xiaohongshu.com/api/sns/web/v2/user/me")
|
|
)
|
|
self.assertTrue(
|
|
valid_xiaohongshu_url(
|
|
"https://edith.xiaohongshu.com/api/sns/web/v2/user/me"
|
|
)
|
|
)
|
|
self.assertTrue(
|
|
valid_xiaohongshu_page_url("https://www.xiaohongshu.com/user/profile/u-1")
|
|
)
|
|
self.assertTrue(valid_xiaohongshu_source_url("https://xhslink.com/a/abc"))
|
|
self.assertTrue(
|
|
valid_xhs_url(
|
|
"https://edith.xiaohongshu.com/api/sns/web/v1/user_posted?user_id=u-1&cursor=&num=30&xsec_source=pc_user"
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
valid_xhs_url(
|
|
"https://edith.xiaohongshu.com/api/sns/web/v1/user_posted?user_id=u-1&num=10"
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
valid_xhs_url("https://edith.xiaohongshu.com.evil/api/sns/web/v2/user/me")
|
|
)
|
|
self.assertTrue(
|
|
valid_xhs_post_url("https://so.xiaohongshu.com/api/sns/web/v2/search/notes")
|
|
)
|
|
self.assertTrue(
|
|
valid_xhs_post_url("https://edith.xiaohongshu.com/api/sns/web/v1/feed")
|
|
)
|
|
self.assertTrue(
|
|
valid_xiaohongshu_media_url(
|
|
"https://www.xiaohongshu.com/explore/n-1?xsec_source=pc_search"
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
valid_xiaohongshu_media_url(
|
|
"https://www.xiaohongshu.com/explore/n-1#fragment"
|
|
)
|
|
)
|
|
|
|
def test_generation_shape_matches_existing_browser_fence(self) -> None:
|
|
self.assertTrue(
|
|
valid_xiaohongshu_generation(
|
|
{
|
|
"binding_version": 1,
|
|
"runtime_id": "a" * 64,
|
|
"network_id": "network",
|
|
"network_exit_id": "",
|
|
}
|
|
)
|
|
)
|
|
self.assertFalse(
|
|
valid_xiaohongshu_generation(
|
|
{"binding_version": 1, "runtime_id": "runtime", "network_id": "network"}
|
|
)
|
|
)
|
|
|
|
def test_facade_exports_shared_browser(self) -> None:
|
|
self.assertIs(
|
|
xiaohongshu_module.XiaohongshuBrowser,
|
|
gateway_module.XiaohongshuBrowser,
|
|
)
|
|
|
|
|
|
class XiaohongshuRouteTests(unittest.TestCase):
|
|
def test_read_only_routes_dispatch_without_action_or_event_routes(self) -> None:
|
|
handler = gateway_module.GatewayHandler.__new__(gateway_module.GatewayHandler)
|
|
gateway = Mock()
|
|
gateway.get_xiaohongshu.return_value = {"status": 200}
|
|
gateway.post_xiaohongshu.return_value = {"status": 200}
|
|
gateway.get_xiaohongshu_media.return_value = {"status": 200}
|
|
gateway.xiaohongshu_identity.return_value = {"uid": "u-1"}
|
|
gateway.resolve_xiaohongshu.return_value = {
|
|
"url": "https://www.xiaohongshu.com/explore/n-1"
|
|
}
|
|
server = Mock()
|
|
server.gateway = gateway
|
|
cast(Any, handler).server = server
|
|
cast(Any, handler).server_as_gateway = lambda: server
|
|
|
|
self.assertEqual(
|
|
handler._route("POST", "/v1/browsers/account-a/xiaohongshu/get", {}, {}),
|
|
{"status": 200},
|
|
)
|
|
self.assertEqual(
|
|
handler._route("POST", "/v1/browsers/account-a/xiaohongshu/post", {}, {}),
|
|
{"status": 200},
|
|
)
|
|
self.assertEqual(
|
|
handler._route("POST", "/v1/browsers/account-a/xiaohongshu/media", {}, {}),
|
|
{"status": 200},
|
|
)
|
|
self.assertEqual(
|
|
handler._route(
|
|
"POST", "/v1/browsers/account-a/xiaohongshu/identity", {}, {}
|
|
),
|
|
{"uid": "u-1"},
|
|
)
|
|
self.assertEqual(
|
|
handler._route(
|
|
"POST", "/v1/browsers/account-a/xiaohongshu/resolve", {}, {}
|
|
),
|
|
{"url": "https://www.xiaohongshu.com/explore/n-1"},
|
|
)
|
|
with self.assertRaises(RequestError):
|
|
handler._route("POST", "/v1/browsers/account-a/xiaohongshu/action", {}, {})
|
|
gateway.get_xiaohongshu.assert_called_once_with("account-a", {})
|
|
gateway.post_xiaohongshu.assert_called_once_with("account-a", {})
|
|
gateway.get_xiaohongshu_media.assert_called_once_with("account-a", {})
|
|
gateway.xiaohongshu_identity.assert_called_once_with("account-a", {})
|
|
gateway.resolve_xiaohongshu.assert_called_once_with("account-a", {})
|