292 lines
11 KiB
Python
292 lines
11 KiB
Python
"""test_server_smoke.py — смоук для resilience-логики tradein-browser.
|
||
|
||
Полный браузер (camoufox subprocess) юнит-тестить непрактично, поэтому здесь
|
||
только смоук: (1) модуль импортируется; (2) /fetch и /login отдают 503 когда
|
||
браузер поставщика не поднят (прокси лежит) — критичный resilience-инвариант,
|
||
защищающий от crash-storm. camoufox в окружении теста НЕ нужен: launch замокан
|
||
no-op'ом, имитируя «прокси всё ещё недоступен».
|
||
|
||
Per-provider модель (#1793): состояние keyed по provider (avito/cian/yandex/
|
||
generic), доступ к инстансу сериализован _locks[provider]; разные поставщики
|
||
работают параллельно. Конкурентность проверяется в test_server.py.
|
||
|
||
Запуск (из tradein-mvp/browser/)::
|
||
|
||
python -m pytest test_server_smoke.py -q
|
||
|
||
aiohttp.test_utils.make_mocked_request не поднимает реальный сервер/сокет —
|
||
хендлер вызывается напрямую, без event-loop-сервера.
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import importlib.util
|
||
import json
|
||
from pathlib import Path
|
||
from typing import Any
|
||
|
||
import pytest
|
||
from aiohttp.test_utils import make_mocked_request
|
||
|
||
# server.py — не пакет (отдельный сервис без __init__/pyproject). Грузим по пути.
|
||
_SERVER_PATH = Path(__file__).resolve().parent / "server.py"
|
||
_spec = importlib.util.spec_from_file_location("tradein_browser_server", _SERVER_PATH)
|
||
assert _spec is not None and _spec.loader is not None
|
||
server = importlib.util.module_from_spec(_spec)
|
||
_spec.loader.exec_module(server)
|
||
|
||
|
||
def _json_body(response: Any) -> dict[str, Any]:
|
||
"""Достаёт JSON-тело из aiohttp Response (body — bytes после json_response)."""
|
||
return json.loads(response.body.decode())
|
||
|
||
|
||
@pytest.fixture(autouse=True)
|
||
def _reset_state(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""Чистое per-provider состояние + инициализированный _locks_guard на каждый тест."""
|
||
monkeypatch.setattr(server, "_browsers", {})
|
||
monkeypatch.setattr(server, "_browser_cms", {})
|
||
monkeypatch.setattr(server, "_page_counters", {})
|
||
monkeypatch.setattr(server, "_locks", {})
|
||
monkeypatch.setattr(server, "_retry_tasks", {})
|
||
monkeypatch.setattr(server, "_locks_guard", asyncio.Lock())
|
||
|
||
|
||
def test_module_imports() -> None:
|
||
"""Смоук: модуль импортируется и экспонирует ключевые хендлеры."""
|
||
assert callable(server.fetch_handler)
|
||
assert callable(server.login_handler)
|
||
assert callable(server.health_handler)
|
||
|
||
|
||
def test_health_reports_browsers_not_ready() -> None:
|
||
"""/health всегда 200, browsers[<provider>]=False когда инстансы не подняты."""
|
||
request = make_mocked_request("GET", "/health")
|
||
response = asyncio.run(server.health_handler(request))
|
||
assert response.status == 200
|
||
body = _json_body(response)
|
||
assert body["status"] == "ok"
|
||
assert body["browsers"] == {"avito": False, "cian": False, "yandex": False, "generic": False}
|
||
|
||
|
||
def test_health_reports_partial_readiness() -> None:
|
||
"""/health отражает per-provider readiness: один поднят, остальные нет."""
|
||
server._browsers["avito"] = object()
|
||
request = make_mocked_request("GET", "/health")
|
||
response = asyncio.run(server.health_handler(request))
|
||
body = _json_body(response)
|
||
assert body["browsers"]["avito"] is True
|
||
assert body["browsers"]["cian"] is False
|
||
|
||
|
||
def test_fetch_returns_503_when_browser_unavailable(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
"""/fetch отдаёт 503 (не 500/краш) когда инстанс поставщика не поднят.
|
||
|
||
Имитирует «прокси лежит»: _ensure_browser no-op, оставляющий инстанс None.
|
||
Никакого реального camoufox-launch'а / direct-fetch.
|
||
"""
|
||
|
||
async def _no_ensure(provider: str) -> bool:
|
||
return False # прокси всё ещё недоступен → браузер остаётся None
|
||
|
||
monkeypatch.setattr(server, "_ensure_browser", _no_ensure)
|
||
|
||
request = make_mocked_request("POST", "/fetch")
|
||
|
||
async def _run() -> Any:
|
||
request.json = lambda: _coro({"url": "https://www.avito.ru/x"}) # type: ignore[method-assign]
|
||
return await server.fetch_handler(request)
|
||
|
||
response = asyncio.run(_run())
|
||
assert response.status == 503
|
||
body = _json_body(response)
|
||
assert "browser unavailable" in body["error"]
|
||
|
||
|
||
def test_login_returns_503_when_browser_unavailable(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
"""/login тоже отдаёт 503 когда инстанс не поднят (без credential-логирования)."""
|
||
|
||
async def _no_ensure(provider: str) -> bool:
|
||
return False
|
||
|
||
monkeypatch.setattr(server, "_ensure_browser", _no_ensure)
|
||
|
||
login_body = {
|
||
"url": "https://www.cian.ru/login",
|
||
"email": "u@example.com",
|
||
"password": "secret",
|
||
"email_selector": "#email",
|
||
"password_selector": "#password",
|
||
"submit_selector": "#submit",
|
||
}
|
||
request = make_mocked_request("POST", "/login")
|
||
|
||
async def _run() -> Any:
|
||
request.json = lambda: _coro(login_body) # type: ignore[method-assign]
|
||
return await server.login_handler(request)
|
||
|
||
response = asyncio.run(_run())
|
||
assert response.status == 503
|
||
body = _json_body(response)
|
||
assert "browser unavailable" in body["error"]
|
||
|
||
|
||
# ── per-provider state isolation ─────────────────────────────────────────────────
|
||
|
||
|
||
class _FakePage:
|
||
"""Поддельная page: считает open/close и отмечает пик параллелизма."""
|
||
|
||
def __init__(self, tracker: "_Tracker") -> None:
|
||
self._tracker = tracker
|
||
|
||
async def route(self, pattern: str, handler: Any) -> None:
|
||
return None
|
||
|
||
async def goto(self, url: str, **kwargs: Any) -> None:
|
||
self._tracker.active += 1
|
||
self._tracker.peak = max(self._tracker.peak, self._tracker.active)
|
||
await asyncio.sleep(0.02)
|
||
self._tracker.active -= 1
|
||
|
||
async def wait_for_timeout(self, ms: int) -> None:
|
||
return None
|
||
|
||
async def content(self) -> str:
|
||
return "<html>ok</html>"
|
||
|
||
async def close(self) -> None:
|
||
self._tracker.closed += 1
|
||
|
||
|
||
class _FakeBrowser:
|
||
def __init__(self, tracker: "_Tracker") -> None:
|
||
self._tracker = tracker
|
||
|
||
async def new_page(self) -> _FakePage:
|
||
self._tracker.opened += 1
|
||
return _FakePage(self._tracker)
|
||
|
||
|
||
class _Tracker:
|
||
def __init__(self) -> None:
|
||
self.opened = 0
|
||
self.closed = 0
|
||
self.active = 0
|
||
self.peak = 0
|
||
|
||
|
||
def _make_fetch_request(url: str = "https://www.avito.ru/x") -> Any:
|
||
request = make_mocked_request("POST", "/fetch")
|
||
request.json = lambda: _coro({"url": url}) # type: ignore[method-assign]
|
||
return request
|
||
|
||
|
||
def test_fetch_opens_and_closes_own_page(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""Один /fetch открывает ровно одну собственную страницу и закрывает её."""
|
||
tracker = _Tracker()
|
||
server._browsers["avito"] = _FakeBrowser(tracker)
|
||
monkeypatch.setattr(server, "BROWSER_RECYCLE_PAGES", 10_000)
|
||
|
||
async def _ensure(provider: str) -> bool:
|
||
return True
|
||
|
||
monkeypatch.setattr(server, "_ensure_browser", _ensure)
|
||
|
||
response = asyncio.run(server.fetch_handler(_make_fetch_request()))
|
||
assert response.status == 200
|
||
assert _json_body(response)["html"] == "<html>ok</html>"
|
||
assert tracker.opened == 1
|
||
assert tracker.closed == 1
|
||
|
||
|
||
# ── _relaunch resets per-provider page counter (recycle) ─────────────────────────
|
||
|
||
|
||
def test_fetch_recycles_after_threshold(monkeypatch: pytest.MonkeyPatch) -> None:
|
||
"""Достижение BROWSER_RECYCLE_PAGES → _relaunch_browser + сброс счётчика."""
|
||
tracker = _Tracker()
|
||
server._browsers["avito"] = _FakeBrowser(tracker)
|
||
server._page_counters["avito"] = 0
|
||
monkeypatch.setattr(server, "BROWSER_RECYCLE_PAGES", 1)
|
||
|
||
relaunched = {"n": 0}
|
||
|
||
async def _fake_relaunch(provider: str) -> None:
|
||
relaunched["n"] += 1
|
||
server._page_counters[provider] = 0
|
||
|
||
async def _ensure(provider: str) -> bool:
|
||
return True
|
||
|
||
monkeypatch.setattr(server, "_relaunch_browser", _fake_relaunch)
|
||
monkeypatch.setattr(server, "_ensure_browser", _ensure)
|
||
|
||
asyncio.run(server.fetch_handler(_make_fetch_request()))
|
||
assert relaunched["n"] == 1, "recycle должен сработать на пороге"
|
||
assert server._page_counters["avito"] == 0
|
||
|
||
|
||
# ── _login_once closes page on CancelledError (BaseException leak fix) ────────────
|
||
|
||
|
||
class _CancellingLoginPage:
|
||
"""Page для /login, которая бросает CancelledError внутри логина."""
|
||
|
||
def __init__(self, tracker: "_Tracker") -> None:
|
||
self._tracker = tracker
|
||
self.url = "https://www.cian.ru/login"
|
||
|
||
async def goto(self, url: str, **kwargs: Any) -> None:
|
||
return None
|
||
|
||
async def wait_for_timeout(self, ms: int) -> None:
|
||
return None
|
||
|
||
async def fill(self, selector: str, value: str, **kwargs: Any) -> None:
|
||
raise asyncio.CancelledError()
|
||
|
||
async def close(self) -> None:
|
||
self._tracker.closed += 1
|
||
|
||
|
||
class _CancellingLoginBrowser:
|
||
def __init__(self, tracker: "_Tracker") -> None:
|
||
self._tracker = tracker
|
||
|
||
async def new_page(self) -> _CancellingLoginPage:
|
||
self._tracker.opened += 1
|
||
return _CancellingLoginPage(self._tracker)
|
||
|
||
|
||
def test_login_closes_page_on_cancelled_error(
|
||
monkeypatch: pytest.MonkeyPatch,
|
||
) -> None:
|
||
"""_login_once закрывает страницу даже при CancelledError (finally, не except)."""
|
||
tracker = _Tracker()
|
||
server._browsers["cian"] = _CancellingLoginBrowser(tracker)
|
||
|
||
params = {
|
||
"url": "https://www.cian.ru/login",
|
||
"email": "u@example.com",
|
||
"password": "secret",
|
||
"email_selector": "#email",
|
||
"password_selector": "#password",
|
||
"submit_selector": "#submit",
|
||
}
|
||
|
||
with pytest.raises(asyncio.CancelledError):
|
||
asyncio.run(server._login_once("cian", params))
|
||
|
||
assert tracker.opened == 1
|
||
assert tracker.closed == 1, "страница должна закрыться в finally при CancelledError"
|
||
|
||
|
||
async def _coro(value: Any) -> Any:
|
||
"""Хелпер: оборачивает значение в awaitable для подмены request.json()."""
|
||
return value
|