70 lines
2.1 KiB
Python
70 lines
2.1 KiB
Python
"""
|
|
Shared helpers for binary-protocol service tests.
|
|
"""
|
|
|
|
import os
|
|
import threading
|
|
from types import ModuleType
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
from hypothesis import HealthCheck
|
|
|
|
|
|
_FUZZ_SETTINGS = dict(
|
|
max_examples=int(os.environ.get("HYPOTHESIS_MAX_EXAMPLES", "200")),
|
|
deadline=2000,
|
|
suppress_health_check=[HealthCheck.function_scoped_fixture],
|
|
)
|
|
|
|
|
|
def make_fake_syslog_bridge() -> ModuleType:
|
|
mod = ModuleType("syslog_bridge")
|
|
mod.syslog_line = MagicMock(return_value="")
|
|
mod.write_syslog_file = MagicMock()
|
|
mod.forward_syslog = MagicMock()
|
|
mod.SEVERITY_WARNING = 4
|
|
mod.SEVERITY_INFO = 6
|
|
# encode_secret returns the universal cred SD shape; tests don't
|
|
# care about the exact bytes, just that the key set is correct.
|
|
mod.encode_secret = MagicMock(
|
|
return_value={"secret_printable": "", "secret_b64": ""}
|
|
)
|
|
# classify_authorization returns None for unknown / absent auth so
|
|
# services that call **(cred or {}) get a no-op spread.
|
|
mod.classify_authorization = MagicMock(return_value=None)
|
|
return mod
|
|
|
|
|
|
def load_real_instance_seed() -> ModuleType:
|
|
"""Load the real instance_seed helper so templates under test see the
|
|
actual per-instance seeding behavior, not a stub. Tests that need
|
|
determinism should pin NODE_NAME via monkeypatch before loading a
|
|
template."""
|
|
import importlib.util
|
|
spec = importlib.util.spec_from_file_location(
|
|
"instance_seed", "decnet/templates/instance_seed.py"
|
|
)
|
|
mod = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(mod)
|
|
return mod
|
|
|
|
|
|
def run_with_timeout(fn, *args, timeout: float = 2.0) -> None:
|
|
"""Run fn(*args) in a daemon thread. pytest.fail if it doesn't return in time."""
|
|
exc_box: list[BaseException] = []
|
|
|
|
def _target():
|
|
try:
|
|
fn(*args)
|
|
except Exception as e:
|
|
exc_box.append(e)
|
|
|
|
t = threading.Thread(target=_target, daemon=True)
|
|
t.start()
|
|
t.join(timeout)
|
|
if t.is_alive():
|
|
pytest.fail(f"data_received hung for >{timeout}s — likely infinite loop")
|
|
if exc_box:
|
|
raise exc_box[0]
|