mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-10-18 21:02:20 +02:00
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
"""Behavioral tests for Cookbook port parsing / picking (#4507 follow-up).
|
|
|
|
Driven through `node --input-type=module` (same approach as the other
|
|
*_js.py tests); skips when `node` is not installed.
|
|
"""
|
|
import json
|
|
import shutil
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
_REPO = Path(__file__).resolve().parent.parent
|
|
_HELPER = _REPO / "static" / "js" / "cookbookPorts.js"
|
|
_HAS_NODE = shutil.which("node") is not None
|
|
|
|
|
|
def _run(expr):
|
|
js = (
|
|
f"import {{ portOf, nextFreePort }} from '{_HELPER.as_posix()}';"
|
|
f"console.log(JSON.stringify({expr}));"
|
|
)
|
|
proc = subprocess.run(
|
|
["node", "--input-type=module"],
|
|
input=js, capture_output=True, text=True, cwd=str(_REPO), timeout=30,
|
|
)
|
|
assert proc.returncode == 0, proc.stderr
|
|
return json.loads(proc.stdout.strip())
|
|
|
|
|
|
@pytest.mark.skipif(not _HAS_NODE, reason="node binary not on PATH")
|
|
def test_port_of_handles_all_forms():
|
|
assert _run("portOf('vllm serve m --host 0.0.0.0 --port 8000')") == "8000"
|
|
assert _run("portOf('x --port=8001')") == "8001"
|
|
assert _run("portOf('llama-server -p 8002')") == "8002"
|
|
assert _run("portOf('llama-server -p=8003')") == "8003"
|
|
assert _run("portOf('serve with no port flag')") == ""
|
|
|
|
|
|
@pytest.mark.skipif(not _HAS_NODE, reason="node binary not on PATH")
|
|
def test_next_free_port_skips_taken_including_eq_and_short_flag():
|
|
# a --port= serve and a -p serve are both 'taken'; picker skips them
|
|
taken = "[portOf('a --port=8000'), portOf('b -p 8001')]"
|
|
assert _run(f"nextFreePort({taken})") == "8002"
|
|
assert _run("nextFreePort([])") == "8000"
|
|
assert _run("nextFreePort(['8000', '8002'])") == "8001"
|
|
|
|
|
|
@pytest.mark.skipif(not _HAS_NODE, reason="node binary not on PATH")
|
|
def test_clash_outcome_same_port_flagged_different_ignored():
|
|
# the guard's predicate is portOf(cmd) === target
|
|
assert _run("portOf('m --port 8000') === '8000'") is True
|
|
assert _run("portOf('m --port 8001') === '8000'") is False
|