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.
ADN-DMR-Peer-Server/tests/application/test_subscription_table_sto...

139 lines
5.5 KiB

# ADN DMR Peer Server - tests application subscription table store ops
#
# Copyright (C) 2026 Rodrigo Pérez, CE5RPY <ce5rpy@qmd.cl>
#
###############################################################################
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
###############################################################################
"""Store-native bridge table and OBP source ops."""
from __future__ import annotations
from tests.harness.deterministic import active_routing_table, minimal_config
from adn_server.application.subscription.obp_source_ops import ensure_obp_source_for_tg_store
from adn_server.application.subscription.store_sync import replace_store_from_routing_table
from adn_server.application.subscription.subscription_table_ops import (
ensure_dynamic_relay_store,
make_static_tg_store,
)
from adn_server.domain import bytes_3
from fakes.subscription_store import InMemorySubscriptionStore
def _store(bridges: dict | None = None) -> InMemorySubscriptionStore:
store = InMemorySubscriptionStore()
if bridges:
replace_store_from_routing_table(store, bridges)
return store
def test_make_static_tg_store_creates_active_off_leg() -> None:
config = minimal_config(("MASTER-A",))
store = _store()
make_static_tg_store(
store,
52090,
2,
10.0,
"MASTER-A",
config["SYSTEMS"],
now=1000.0,
single_mode=False,
)
leg = next(s for s in store.snapshot() if s.system.value == "MASTER-A" and int(s.channel.slot) == 2)
assert leg.is_active()
def test_ensure_obp_source_activates_ts1_leg() -> None:
bridges = active_routing_table(7305, (("OBP-CL", 1), ("MASTER-A", 2)))
for entry in bridges["7305"]:
if entry["SYSTEM"] == "OBP-CL":
entry["ACTIVE"] = False
store = _store(bridges)
ensure_obp_source_for_tg_store(store, "OBP-CL", "7305", bytes_3(7305), 7305, now=500.0)
obp = next(s for s in store.snapshot() if s.system.value == "OBP-CL")
assert obp.is_active()
def test_ensure_dynamic_relay_store_obp_leg() -> None:
config = minimal_config(("MASTER-A",))
add_obp = {
"OBP-CL": {
"MODE": "OPENBRIDGE",
"ENABLED": True,
"IP": "127.0.0.1",
"PORT": 0,
}
}
config["SYSTEMS"].update(add_obp)
store = _store()
ensure_dynamic_relay_store(store, 7305, "MASTER-A", 2, 10.0, config["SYSTEMS"], now=1.0)
obp = next(s for s in store.snapshot() if s.system.value == "OBP-CL")
assert obp.is_active()
def test_master_dynamic_tg_slots_single_mode() -> None:
"""SINGLE=1 UA session is detected for a dynamic TG not in OPTIONS."""
from adn_server.application.routing.helpers import master_dynamic_tg_slots
from adn_server.domain import bytes_4
sys_cfg = {"_PEER_UA_SESSIONS": {bytes_4(730039101): {2: {"tgid": 7300, "expires": 9999999.0}}}}
assert master_dynamic_tg_slots(sys_cfg, 7300, now=1000.0) == {2}
def test_master_dynamic_tg_slots_multi_mode() -> None:
"""SINGLE=0 multi-dynamic TG set is detected."""
from adn_server.application.routing.helpers import master_dynamic_tg_slots
from adn_server.domain import bytes_4
sys_cfg = {"_PEER_UA_MULTI_TGS": {bytes_4(730039101): {2: {7300, 7305}}}}
assert master_dynamic_tg_slots(sys_cfg, 7300, now=1000.0) == {2}
def test_master_dynamic_tg_slots_expired_single() -> None:
"""Expired SINGLE=1 session is not detected."""
from adn_server.application.routing.helpers import master_dynamic_tg_slots
from adn_server.domain import bytes_4
sys_cfg = {"_PEER_UA_SESSIONS": {bytes_4(730039101): {2: {"tgid": 7300, "expires": 500.0}}}}
assert master_dynamic_tg_slots(sys_cfg, 7300, now=1000.0) == set()
def test_apply_static_tg_to_bridge_activates_dynamic_tg() -> None:
"""A dynamic TG not in OPTIONS is activated when a peer has an active UA session for it."""
from tests.harness.deterministic import DeterministicScenario, add_openbridge_system
from adn_server.domain import bytes_4
config = minimal_config(("MASTER-A",))
add_openbridge_system(config, "OBP-CL")
config["SYSTEMS"]["MASTER-A"]["_PEER_UA_SESSIONS"] = {
bytes_4(730039101): {2: {"tgid": 7300, "expires": 9999999.0}}
}
scenario = DeterministicScenario(config)
scenario.routing.ensure_dynamic_relay(bytes_3(7300), "MASTER-A", 2, 10.0)
# Simulate OBP-CL/TS1 source active (as _ensure_obp_source_for_tg would do)
scenario.routing._ensure_obp_source_for_tg("OBP-CL", "7300", bytes_3(7300), 7300)
# Before fix: SYSTEM/7300/TS2 is IDLE, OBP traffic can't bridge
scenario.routing.apply_static_tg_to_bridge(7300)
legs = [
s for s in scenario.subscription_store.snapshot()
if s.system.value == "MASTER-A" and int(s.target_tgid) == 7300
]
ts2 = next(s for s in legs if int(s.channel.slot) == 2)
assert ts2.is_active(), "SYSTEM TS2 leg for dynamic TG 7300 should be ACTIVE"

Powered by TurnKey Linux.