snowy-yang
bcfc1022eb
- granian[reload,rloop,uvloop,winloop] 改为默认依赖，uv sync 即具备完整启动栈 - auto 在启动器内解析为具体循环：granian Windows→winloop / Unix→rloop（缺省回退 uvloop/asyncio）； uvicorn Unix→uvloop / Windows→asyncio - 启动日志打印实际 server+loop；测试覆盖平台默认与回退；README 默认矩阵更新
114 lines
4.8 KiB
Python
114 lines
4.8 KiB
Python
"""启动器（scripts/serve.py）的循环/服务器选择逻辑测试（不启动真实服务器）。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
|
|
|
|
from scripts import serve
|
|
|
|
|
|
def test_pick_server_auto_prefers_granian(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name == "granian")
|
|
assert serve.pick_server("auto") == "granian"
|
|
|
|
monkeypatch.setattr(serve, "has_module", lambda name: False)
|
|
assert serve.pick_server("auto") == "uvicorn"
|
|
|
|
|
|
def test_available_loops_follow_installed_packages(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name == "winloop")
|
|
assert serve.available_loops("granian") == ["auto", "asyncio", "winloop"]
|
|
assert serve.available_loops("uvicorn") == ["auto", "asyncio"]
|
|
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name in ("rloop", "uvloop"))
|
|
assert serve.available_loops("granian") == ["auto", "asyncio", "rloop", "uvloop"]
|
|
assert serve.available_loops("uvicorn") == ["auto", "asyncio", "uvloop"]
|
|
|
|
|
|
def test_explicit_unavailable_loop_fails_with_hint(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "has_module", lambda name: False)
|
|
with pytest.raises(SystemExit) as error:
|
|
serve.resolve_loop("granian", "rloop")
|
|
message = str(error.value)
|
|
assert "rloop" in message and "可用" in message
|
|
|
|
|
|
def test_uvicorn_rejects_granian_only_loops() -> None:
|
|
with pytest.raises(SystemExit):
|
|
serve.resolve_loop("uvicorn", "winloop")
|
|
with pytest.raises(SystemExit):
|
|
serve.resolve_loop("uvicorn", "rloop")
|
|
|
|
|
|
def test_uvicorn_auto_and_asyncio_allowed(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", True)
|
|
assert serve.resolve_loop("uvicorn", "auto") == "asyncio"
|
|
assert serve.resolve_loop("uvicorn", "asyncio") == "asyncio"
|
|
|
|
|
|
def test_default_loop_granian_windows(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", True)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name == "winloop")
|
|
assert serve.default_loop("granian") == "winloop"
|
|
|
|
|
|
def test_default_loop_granian_unix_prefers_rloop(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", False)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name in ("rloop", "uvloop"))
|
|
assert serve.default_loop("granian") == "rloop"
|
|
|
|
|
|
def test_default_loop_granian_unix_falls_back(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", False)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name == "uvloop")
|
|
assert serve.default_loop("granian") == "uvloop"
|
|
|
|
monkeypatch.setattr(serve, "has_module", lambda name: False)
|
|
assert serve.default_loop("granian") == "asyncio"
|
|
|
|
|
|
def test_default_loop_uvicorn(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", False)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name == "uvloop")
|
|
assert serve.default_loop("uvicorn") == "uvloop"
|
|
|
|
# Windows 上 uvicorn 只有 asyncio（即使装了 winloop 也不会用）
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", True)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name == "winloop")
|
|
assert serve.default_loop("uvicorn") == "asyncio"
|
|
|
|
|
|
def test_default_loop_windows_without_winloop(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", True)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: False)
|
|
assert serve.default_loop("granian") == "asyncio"
|
|
|
|
|
|
def test_resolve_loop_auto_uses_platform_default(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "IS_WINDOWS", False)
|
|
monkeypatch.setattr(serve, "has_module", lambda name: name in ("rloop", "uvloop"))
|
|
assert serve.resolve_loop("granian", "auto") == "rloop"
|
|
assert serve.resolve_loop("uvicorn", "auto") == "uvloop"
|
|
|
|
|
|
def test_build_command_granian(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setattr(serve, "has_module", lambda name: True)
|
|
command = serve.build_command("granian", host="0.0.0.0", port=16688, workers=2, loop="winloop", reload=True)
|
|
assert command[:3] == [sys.executable, "-m", "granian"]
|
|
assert command[command.index("--loop") + 1] == "winloop"
|
|
assert command[command.index("--workers") + 1] == "2"
|
|
assert "--reload" in command
|
|
assert command[-1] == serve.APP_TARGET
|
|
|
|
|
|
def test_build_command_uvicorn() -> None:
|
|
command = serve.build_command("uvicorn", host="127.0.0.1", port=9000, workers=1, loop="auto", reload=False)
|
|
assert command[:3] == [sys.executable, "-m", "uvicorn"]
|
|
assert command[3] == serve.APP_TARGET
|
|
assert "--reload" not in command
|