mirror of
https://github.com/feder-cr/invisible_playwright.git
synced 2026-06-13 08:55:12 +02:00
c2103ed0db
headless=True now hides the window via the binary's own cloak pref (zoom.stealth.cloak_windows) on Windows and macOS instead of the broken thread-level SetThreadDesktop; macOS is now supported. Linux keeps Xvfb. Adds e2e guards that also run per-platform in the release drive-gate: - test_cloak: the window is hidden (Windows DWMWA_CLOAKED / macOS CGWindowAlpha) yet still renders + drives; the macOS leg is where the cocoa cloak patch runs. - a WebGL readPixels masking guard: the gamma noise must stay a smooth gamma remap, not the pixelscan-maskable +-1 spikes.
124 lines
5.1 KiB
Python
124 lines
5.1 KiB
Python
"""Unit tests for the ``_headless`` window-hider dispatcher.
|
|
|
|
``make_virtual_display`` is pure platform routing:
|
|
- Linux: a ``_LinuxVirtualDisplay`` (Xvfb) object the launcher start()s/stop()s.
|
|
- Windows / macOS: ``None`` — the patched binary self-cloaks its chrome windows
|
|
via ``cloak_prefs()`` (injected by the launcher), so nothing host-side spawns.
|
|
- Anything else: a clear ``RuntimeError`` naming the platform.
|
|
|
|
``_LinuxVirtualDisplay`` construction does no I/O (Xvfb is only spawned in
|
|
``start()``), so it's safe to exercise on any host.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
|
|
import invisible_playwright._headless as headless
|
|
from invisible_playwright._headless import (
|
|
CLOAK_PREFS,
|
|
_LinuxVirtualDisplay,
|
|
cloak_prefs,
|
|
make_virtual_display,
|
|
)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_make_virtual_display_returns_none_on_win32(monkeypatch):
|
|
"""Windows hides via the in-binary cloak pref, not a host-side display."""
|
|
monkeypatch.setattr(headless.sys, "platform", "win32")
|
|
assert make_virtual_display() is None
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_make_virtual_display_returns_none_on_darwin(monkeypatch):
|
|
"""macOS is now supported — it hides via the same in-binary cloak pref."""
|
|
monkeypatch.setattr(headless.sys, "platform", "darwin")
|
|
assert make_virtual_display() is None
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_make_virtual_display_returns_linux_xvfb_on_linux(monkeypatch):
|
|
"""``__init__`` of ``_LinuxVirtualDisplay`` does no I/O — only ``start()``
|
|
spawns Xvfb. Exercising the dispatcher here is safe on any host."""
|
|
monkeypatch.setattr(headless.sys, "platform", "linux")
|
|
assert isinstance(make_virtual_display(), _LinuxVirtualDisplay)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_make_virtual_display_accepts_linux_variants(monkeypatch):
|
|
"""``sys.platform`` can be ``linux2`` on older Pythons / WSL builds.
|
|
The dispatcher uses ``startswith("linux")`` to accept all variants."""
|
|
monkeypatch.setattr(headless.sys, "platform", "linux2")
|
|
assert isinstance(make_virtual_display(), _LinuxVirtualDisplay)
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_make_virtual_display_raises_on_unsupported_platform(monkeypatch):
|
|
monkeypatch.setattr(headless.sys, "platform", "freebsd14")
|
|
with pytest.raises(RuntimeError, match="Windows, macOS and Linux"):
|
|
make_virtual_display()
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_make_virtual_display_error_mentions_offending_platform(monkeypatch):
|
|
"""Error message should include the actual ``sys.platform`` so the
|
|
user can diagnose why their CI / weird container is being rejected."""
|
|
monkeypatch.setattr(headless.sys, "platform", "sunos5")
|
|
with pytest.raises(RuntimeError, match="sunos5"):
|
|
make_virtual_display()
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_cloak_prefs_enables_cloak_and_disables_occlusion():
|
|
"""The cloak prefs must turn on the in-binary cloak and turn OFF Windows
|
|
occlusion tracking (so a hidden window keeps painting). Returns a copy."""
|
|
p = cloak_prefs()
|
|
assert p["zoom.stealth.cloak_windows"] is True
|
|
assert p["widget.windows.window_occlusion_tracking.enabled"] is False
|
|
assert p == CLOAK_PREFS and p is not CLOAK_PREFS
|
|
|
|
|
|
# ──────────────────────────────────────────────────────────────────────
|
|
# _LinuxVirtualDisplay — construction-only smoke tests. ``start()`` is
|
|
# E2E because it spawns Xvfb; ``stop()`` is safe to call when no Xvfb
|
|
# was ever started, so we exercise that path explicitly.
|
|
# ──────────────────────────────────────────────────────────────────────
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_linux_virtual_display_initial_state_is_clean():
|
|
"""Construction must not spawn Xvfb or mutate the environment — only
|
|
``start()`` does. Mirrors the Windows construction-state test."""
|
|
vd = _LinuxVirtualDisplay()
|
|
assert vd._proc is None
|
|
assert vd._display is None
|
|
assert vd._saved_env == {}
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_linux_virtual_display_geometry_default():
|
|
"""Default geometry is 1920x1080x24 — matches the profile sampler's
|
|
default screen and avoids the Xvfb default of 1280x1024 which the
|
|
fingerprint pipeline never produces."""
|
|
vd = _LinuxVirtualDisplay()
|
|
assert vd._geometry == "1920x1080x24"
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_linux_virtual_display_custom_geometry():
|
|
"""Caller-supplied width/height feed straight into the Xvfb geometry
|
|
spec; the depth is always 24 (Firefox/ANGLE assume true-color)."""
|
|
vd = _LinuxVirtualDisplay(width=2560, height=1440)
|
|
assert vd._geometry == "2560x1440x24"
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_linux_virtual_display_stop_without_start_is_safe():
|
|
"""``stop()`` before ``start()`` must be a no-op — supports the
|
|
``__exit__`` path on a launcher that failed before Xvfb was spawned.
|
|
Verifies no AttributeError on env restore (saved_env is empty)."""
|
|
vd = _LinuxVirtualDisplay()
|
|
vd.stop()
|
|
vd.stop()
|
|
assert vd._proc is None
|
|
assert vd._display is None
|