wehub-resource-sync
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import socket
|
|
from collections.abc import Callable
|
|
from dataclasses import dataclass
|
|
from typing import Any, cast
|
|
|
|
import pytest
|
|
from _pytest.config import Config
|
|
|
|
from tests.e2e_ui.url_safety import DEV_PORTS, unsafe_ui_base_url_reason
|
|
|
|
|
|
@dataclass
|
|
class _ConfigStub:
|
|
options: dict[str, Any]
|
|
|
|
def getoption(self, name: str, default: Any = None) -> Any:
|
|
return self.options.get(name, default)
|
|
|
|
|
|
def _pytest_configure() -> Callable[[Config], None]:
|
|
pytest.importorskip("playwright", exc_type=ImportError)
|
|
from tests.e2e_ui.conftest import pytest_configure
|
|
|
|
return pytest_configure
|
|
|
|
|
|
@pytest.mark.parametrize("port", sorted(DEV_PORTS))
|
|
def test_unsafe_ui_base_url_reason_refuses_known_dev_ports(port: int) -> None:
|
|
reason = unsafe_ui_base_url_reason(f"http://example.com:{port}")
|
|
|
|
assert reason == f"port {port} is a known Omnigent/Vite dev port"
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("ui_base_url", "expected"),
|
|
[
|
|
("http://127.0.0.1:54321", "loopback address"),
|
|
("http://localhost:54321", "local dev host"),
|
|
("http://10.0.0.5:54321", "private-network address"),
|
|
("http://[::1]:54321", "loopback address"),
|
|
("not-a-url", "absolute http(s) URL"),
|
|
],
|
|
)
|
|
def test_unsafe_ui_base_url_reason_refuses_dev_hosts(
|
|
ui_base_url: str,
|
|
expected: str,
|
|
) -> None:
|
|
reason = unsafe_ui_base_url_reason(ui_base_url)
|
|
|
|
assert reason is not None
|
|
assert expected in reason
|
|
|
|
|
|
def test_unsafe_ui_base_url_reason_allows_public_non_dev_port(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.setattr(socket, "getaddrinfo", lambda *args, **kwargs: [])
|
|
|
|
assert unsafe_ui_base_url_reason("https://example.com:443") is None
|
|
|
|
|
|
def test_unsafe_ui_base_url_reason_handles_missing_port_explicitly() -> None:
|
|
reason = unsafe_ui_base_url_reason("http://127.0.0.1")
|
|
|
|
assert reason == "a loopback address"
|
|
|
|
|
|
def test_pytest_configure_rejects_headed_in_ci(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("CI", "1")
|
|
|
|
with pytest.raises(pytest.UsageError, match="must run headless in CI"):
|
|
_pytest_configure()(cast(Config, _ConfigStub({"--ui-base-url": None, "--headed": True})))
|
|
|
|
|
|
def test_pytest_configure_rejects_dev_ui_base_url(
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
) -> None:
|
|
monkeypatch.delenv("OMNIGENT_E2E_ALLOW_DEV_BASE_URL", raising=False)
|
|
monkeypatch.delenv("CI", raising=False)
|
|
|
|
with pytest.raises(pytest.UsageError, match="Refusing --ui-base-url"):
|
|
_pytest_configure()(
|
|
cast(
|
|
Config,
|
|
_ConfigStub({"--ui-base-url": "http://127.0.0.1:5173", "--headed": False}),
|
|
)
|
|
)
|