You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
2.1 KiB
64 lines
2.1 KiB
"""Peer disconnect clears internal UA sessions; no synthetic bridge events."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock
|
|
|
|
from adn_server.application.routing.helpers import (
|
|
export_peer_ua_sessions,
|
|
register_peer_ua_session,
|
|
)
|
|
from adn_server.domain import bytes_4
|
|
from adn_server.infrastructure.twisted_adapters.udp_hbp import HBPProtocol
|
|
|
|
|
|
def _master_protocol(*, report: MagicMock | None = None) -> HBPProtocol:
|
|
peer_id = bytes_4(730039101)
|
|
config = {
|
|
"REPORTS": {"REPORT": True},
|
|
"SYSTEMS": {
|
|
"SYSTEM": {
|
|
"MODE": "MASTER",
|
|
"ENABLED": True,
|
|
"PEERS": {
|
|
peer_id: {
|
|
"CONNECTION": "YES",
|
|
"OPTIONS": b"TS2=730,7305;SINGLE=1;TIMER=5;",
|
|
}
|
|
},
|
|
"_PEER_UA_SESSIONS": {},
|
|
}
|
|
},
|
|
}
|
|
proto = HBPProtocol("SYSTEM", config, report_factory=report)
|
|
proto._peers = config["SYSTEMS"]["SYSTEM"]["PEERS"]
|
|
return proto
|
|
|
|
|
|
def test_disconnect_clears_internal_sessions_without_bridge_event() -> None:
|
|
report = MagicMock()
|
|
proto = _master_protocol(report=report)
|
|
peer_id = bytes_4(730039101)
|
|
sys_cfg = proto._config
|
|
register_peer_ua_session(
|
|
proto._peers[peer_id], peer_id, 2, 7305, sys_cfg, now=1_000_000.0
|
|
)
|
|
assert export_peer_ua_sessions(sys_cfg, peer_id, now=1_000_100.0)
|
|
|
|
proto._on_peer_disconnected(peer_id)
|
|
|
|
report.send_routing_event.assert_not_called()
|
|
assert export_peer_ua_sessions(sys_cfg, peer_id, now=1_000_100.0) == {}
|
|
|
|
|
|
def test_export_peer_ua_sessions_omits_expired() -> None:
|
|
peer_id = bytes_4(730039101)
|
|
sys_cfg: dict = {"_PEER_UA_SESSIONS": {}}
|
|
register_peer_ua_session(
|
|
{"OPTIONS": b"SINGLE=1;TIMER=5;"}, peer_id, 2, 7305, sys_cfg, now=1_000_000.0
|
|
)
|
|
active = export_peer_ua_sessions(sys_cfg, peer_id, now=1_000_100.0)
|
|
expired = export_peer_ua_sessions(sys_cfg, peer_id, now=9_999_999.0)
|
|
assert active["2"]["tgid"] == 7305
|
|
assert expired == {}
|