wehub-resource-sync
233 lines
8.5 KiB
Python
233 lines
8.5 KiB
Python
"""Unit tests for build_args timezone/locale injection and timezone alias."""
|
|
|
|
from cloakbrowser.browser import build_args, _resolve_timezone
|
|
|
|
|
|
def test_timezone_injected():
|
|
"""--fingerprint-timezone flag should appear when timezone is set."""
|
|
args = build_args(stealth_args=True, extra_args=None, timezone="America/New_York")
|
|
assert "--fingerprint-timezone=America/New_York" in args
|
|
|
|
|
|
def test_locale_injected():
|
|
"""--lang and --fingerprint-locale flags should appear when locale is set."""
|
|
args = build_args(stealth_args=True, extra_args=None, locale="en-US")
|
|
assert "--lang=en-US" in args
|
|
assert "--fingerprint-locale=en-US" in args
|
|
|
|
|
|
def test_both_injected():
|
|
"""Both flags should appear when both are set."""
|
|
args = build_args(stealth_args=True, extra_args=None, timezone="Europe/Berlin", locale="de-DE")
|
|
assert "--fingerprint-timezone=Europe/Berlin" in args
|
|
assert "--lang=de-DE" in args
|
|
assert "--fingerprint-locale=de-DE" in args
|
|
|
|
|
|
def test_timezone_independent_of_stealth_args():
|
|
"""--fingerprint-timezone should be injected even when stealth_args=False."""
|
|
args = build_args(stealth_args=False, extra_args=None, timezone="America/New_York", locale="en-US")
|
|
assert "--fingerprint-timezone=America/New_York" in args
|
|
assert "--lang=en-US" in args
|
|
assert "--fingerprint-locale=en-US" in args
|
|
# No stealth fingerprint args
|
|
assert not any(a.startswith("--fingerprint=") for a in args)
|
|
|
|
|
|
def test_no_flags_when_not_set():
|
|
"""No timezone/lang/fingerprint-locale flags when params are None."""
|
|
args = build_args(stealth_args=True, extra_args=None)
|
|
assert not any(a.startswith("--fingerprint-timezone=") for a in args)
|
|
assert not any(a.startswith("--lang=") for a in args)
|
|
assert not any(a.startswith("--fingerprint-locale=") for a in args)
|
|
|
|
|
|
def test_extra_args_preserved():
|
|
"""Extra args should still be included alongside timezone/locale."""
|
|
args = build_args(stealth_args=True, extra_args=["--disable-gpu"], timezone="Asia/Tokyo", locale="ja-JP")
|
|
assert "--disable-gpu" in args
|
|
assert "--fingerprint-timezone=Asia/Tokyo" in args
|
|
assert "--lang=ja-JP" in args
|
|
assert "--fingerprint-locale=ja-JP" in args
|
|
|
|
|
|
# --- _resolve_timezone alias ---
|
|
|
|
|
|
def test_resolve_timezone_id_alias():
|
|
"""timezone_id in kwargs should be promoted to timezone."""
|
|
kwargs = {"timezone_id": "Europe/Paris"}
|
|
result = _resolve_timezone(None, kwargs)
|
|
assert result == "Europe/Paris"
|
|
assert "timezone_id" not in kwargs
|
|
|
|
|
|
def test_resolve_timezone_wins_over_alias():
|
|
"""Explicit timezone takes precedence; timezone_id is still popped."""
|
|
kwargs = {"timezone_id": "Europe/Paris"}
|
|
result = _resolve_timezone("UTC", kwargs)
|
|
assert result == "UTC"
|
|
assert "timezone_id" not in kwargs
|
|
|
|
|
|
def test_resolve_no_alias():
|
|
"""No-op when timezone_id is absent."""
|
|
kwargs = {"other": "value"}
|
|
result = _resolve_timezone("UTC", kwargs)
|
|
assert result == "UTC"
|
|
assert "other" in kwargs
|
|
|
|
|
|
def test_resolve_both_none():
|
|
"""Neither param set — returns None."""
|
|
kwargs = {}
|
|
result = _resolve_timezone(None, kwargs)
|
|
assert result is None
|
|
|
|
|
|
# --- Deduplication tests ---
|
|
|
|
|
|
def test_user_fingerprint_overrides_default():
|
|
"""User --fingerprint should override the random default seed."""
|
|
args = build_args(stealth_args=True, extra_args=["--fingerprint=99887"])
|
|
fingerprint_args = [a for a in args if a.startswith("--fingerprint=")]
|
|
assert len(fingerprint_args) == 1
|
|
assert fingerprint_args[0] == "--fingerprint=99887"
|
|
|
|
|
|
def test_user_platform_overrides_default():
|
|
"""User --fingerprint-platform should override the default."""
|
|
args = build_args(stealth_args=True, extra_args=["--fingerprint-platform=linux"])
|
|
platform_args = [a for a in args if a.startswith("--fingerprint-platform=")]
|
|
assert len(platform_args) == 1
|
|
assert platform_args[0] == "--fingerprint-platform=linux"
|
|
|
|
|
|
def test_timezone_param_overrides_user_arg():
|
|
"""Dedicated timezone param should override user arg."""
|
|
args = build_args(
|
|
stealth_args=True,
|
|
extra_args=["--fingerprint-timezone=Europe/London"],
|
|
timezone="America/New_York",
|
|
)
|
|
tz_args = [a for a in args if a.startswith("--fingerprint-timezone=")]
|
|
assert len(tz_args) == 1
|
|
assert tz_args[0] == "--fingerprint-timezone=America/New_York"
|
|
|
|
|
|
def test_locale_param_overrides_user_arg():
|
|
"""Dedicated locale param should override user --lang and --fingerprint-locale args."""
|
|
args = build_args(
|
|
stealth_args=True,
|
|
extra_args=["--lang=de-DE", "--fingerprint-locale=de-DE"],
|
|
locale="en-US",
|
|
)
|
|
lang_args = [a for a in args if a.startswith("--lang=")]
|
|
assert len(lang_args) == 1
|
|
assert lang_args[0] == "--lang=en-US"
|
|
locale_args = [a for a in args if a.startswith("--fingerprint-locale=")]
|
|
assert len(locale_args) == 1
|
|
assert locale_args[0] == "--fingerprint-locale=en-US"
|
|
|
|
|
|
def test_no_duplicate_flags():
|
|
"""No flag key should appear more than once in the output."""
|
|
args = build_args(
|
|
stealth_args=True,
|
|
extra_args=["--fingerprint=99887", "--fingerprint-timezone=UTC", "--lang=fr-FR"],
|
|
timezone="Europe/Berlin",
|
|
locale="de-DE",
|
|
)
|
|
keys = [a.split("=", 1)[0] for a in args]
|
|
assert len(keys) == len(set(keys)), f"Duplicate keys found: {keys}"
|
|
|
|
|
|
def test_non_value_flags_preserved():
|
|
"""Flags without = should be preserved without dedup issues."""
|
|
args = build_args(stealth_args=True, extra_args=["--disable-gpu", "--no-zygote"])
|
|
assert "--disable-gpu" in args
|
|
assert "--no-zygote" in args
|
|
assert "--no-sandbox" in args
|
|
|
|
|
|
def test_override_logs_debug(caplog):
|
|
"""Should log debug message when an override happens."""
|
|
import logging
|
|
|
|
with caplog.at_level(logging.DEBUG, logger="cloakbrowser"):
|
|
build_args(stealth_args=True, extra_args=["--fingerprint=99887"])
|
|
assert any("--fingerprint=" in r.message and "99887" in r.message for r in caplog.records)
|
|
|
|
|
|
# --- WebRTC IP spoofing ---
|
|
|
|
|
|
def test_webrtc_ip_passed_through_args():
|
|
"""--fingerprint-webrtc-ip in args should pass through to output."""
|
|
args = build_args(stealth_args=True, extra_args=["--fingerprint-webrtc-ip=1.2.3.4"])
|
|
assert "--fingerprint-webrtc-ip=1.2.3.4" in args
|
|
|
|
|
|
def test_webrtc_ip_not_present_by_default():
|
|
"""No --fingerprint-webrtc-ip when not in args."""
|
|
args = build_args(stealth_args=True, extra_args=None)
|
|
assert not any(a.startswith("--fingerprint-webrtc-ip") for a in args)
|
|
|
|
|
|
def test_resolve_webrtc_args_auto():
|
|
"""--fingerprint-webrtc-ip=auto should be resolved to an IP."""
|
|
from cloakbrowser.browser import _resolve_webrtc_args
|
|
from unittest.mock import patch
|
|
|
|
with patch("cloakbrowser.geoip._resolve_exit_ip", return_value="5.6.7.8"):
|
|
result = _resolve_webrtc_args(["--fingerprint-webrtc-ip=auto"], "http://proxy:8080")
|
|
assert result == ["--fingerprint-webrtc-ip=5.6.7.8"]
|
|
|
|
|
|
def test_resolve_webrtc_args_explicit_ip_unchanged():
|
|
"""Explicit IP in args should not be touched."""
|
|
from cloakbrowser.browser import _resolve_webrtc_args
|
|
|
|
result = _resolve_webrtc_args(["--fingerprint-webrtc-ip=9.9.9.9"], "http://proxy:8080")
|
|
assert result == ["--fingerprint-webrtc-ip=9.9.9.9"]
|
|
|
|
|
|
def test_resolve_webrtc_args_no_flag():
|
|
"""No webrtc flag in args should return args unchanged."""
|
|
from cloakbrowser.browser import _resolve_webrtc_args
|
|
|
|
result = _resolve_webrtc_args(["--no-sandbox"], "http://proxy:8080")
|
|
assert result == ["--no-sandbox"]
|
|
|
|
|
|
def test_start_maximized_injected_when_gated():
|
|
"""start_maximized=True adds the flag."""
|
|
args = build_args(stealth_args=True, extra_args=None, start_maximized=True)
|
|
assert "--start-maximized" in args
|
|
|
|
|
|
def test_start_maximized_absent_by_default():
|
|
"""Default (gate off) does not add the flag."""
|
|
args = build_args(stealth_args=True, extra_args=None)
|
|
assert "--start-maximized" not in args
|
|
args = build_args(stealth_args=True, extra_args=None, start_maximized=False)
|
|
assert "--start-maximized" not in args
|
|
|
|
|
|
def test_start_maximized_suppressed_by_user_window_size():
|
|
"""A user --window-size means the user chose a geometry; don't also maximize."""
|
|
args = build_args(
|
|
stealth_args=True, extra_args=["--window-size=1000,800"], start_maximized=True
|
|
)
|
|
assert "--start-maximized" not in args
|
|
assert "--window-size=1000,800" in args
|
|
|
|
|
|
def test_start_maximized_not_doubled():
|
|
"""A user-supplied --start-maximized is not duplicated."""
|
|
args = build_args(
|
|
stealth_args=True, extra_args=["--start-maximized"], start_maximized=True
|
|
)
|
|
assert args.count("--start-maximized") == 1
|