mirror of
https://github.com/BigBodyCobain/Shadowbroker.git
synced 2026-05-26 17:17:51 +02:00
BigBodyCobain
25a98a9869
Allow local-operator DM invite import without requiring a full admin session. Prioritize bundled/bootstrap seed peers and shorten stale seed cooldowns for faster Infonet recovery. Replace raw DM invite dumps with copyable signed-address controls, contact request handling, and safer sealed-send behavior while the private delivery route connects.
134 lines
3.7 KiB
Python
134 lines
3.7 KiB
Python
from services.mesh.mesh_peer_store import (
|
|
PeerStore,
|
|
make_bootstrap_peer_record,
|
|
make_push_peer_record,
|
|
make_sync_peer_record,
|
|
)
|
|
|
|
|
|
def test_peer_store_preserves_provenance_across_buckets(tmp_path):
|
|
store = PeerStore(tmp_path / "peer_store.json")
|
|
bootstrap = make_bootstrap_peer_record(
|
|
peer_url="https://seed.example",
|
|
transport="clearnet",
|
|
role="seed",
|
|
signer_id="bootstrap-a",
|
|
now=100,
|
|
)
|
|
sync_peer = make_sync_peer_record(
|
|
peer_url="https://seed.example",
|
|
transport="clearnet",
|
|
role="seed",
|
|
source="bootstrap_promoted",
|
|
signer_id="bootstrap-a",
|
|
now=101,
|
|
)
|
|
push_peer = make_push_peer_record(
|
|
peer_url="https://seed.example",
|
|
transport="clearnet",
|
|
role="seed",
|
|
now=102,
|
|
)
|
|
|
|
store.upsert(bootstrap)
|
|
store.upsert(sync_peer)
|
|
store.upsert(push_peer)
|
|
|
|
assert [record.bucket for record in store.records()] == ["bootstrap", "push", "sync"]
|
|
assert [record.source for record in store.records_for_bucket("sync")] == ["bootstrap_promoted"]
|
|
assert [record.source for record in store.records_for_bucket("push")] == ["operator"]
|
|
|
|
|
|
def test_peer_store_save_load_roundtrip(tmp_path):
|
|
path = tmp_path / "peer_store.json"
|
|
store = PeerStore(path)
|
|
store.upsert(
|
|
make_bootstrap_peer_record(
|
|
peer_url="https://seed.example",
|
|
transport="clearnet",
|
|
role="seed",
|
|
signer_id="bootstrap-a",
|
|
now=100,
|
|
)
|
|
)
|
|
store.upsert(
|
|
make_sync_peer_record(
|
|
peer_url="http://alphaexample.onion",
|
|
transport="onion",
|
|
role="relay",
|
|
source="operator",
|
|
now=101,
|
|
)
|
|
)
|
|
store.save()
|
|
|
|
loaded = PeerStore(path)
|
|
records = loaded.load()
|
|
|
|
assert len(records) == 2
|
|
assert [record.bucket for record in records] == ["bootstrap", "sync"]
|
|
assert records[0].signer_id == "bootstrap-a"
|
|
assert records[1].peer_url == "http://alphaexample.onion"
|
|
|
|
|
|
def test_peer_store_failure_and_success_lifecycle(tmp_path):
|
|
store = PeerStore(tmp_path / "peer_store.json")
|
|
store.upsert(
|
|
make_sync_peer_record(
|
|
peer_url="https://sync.example",
|
|
transport="clearnet",
|
|
now=100,
|
|
)
|
|
)
|
|
failed = store.mark_failure(
|
|
"https://sync.example",
|
|
"sync",
|
|
error="timeout",
|
|
cooldown_s=30,
|
|
now=200,
|
|
)
|
|
assert failed.failure_count == 1
|
|
assert failed.cooldown_until == 230
|
|
assert failed.last_error == "timeout"
|
|
|
|
recovered = store.mark_sync_success("https://sync.example", now=250)
|
|
assert recovered.failure_count == 0
|
|
assert recovered.cooldown_until == 0
|
|
assert recovered.last_error == ""
|
|
assert recovered.last_sync_ok_at == 250
|
|
|
|
|
|
def test_upsert_explicit_seed_clears_stale_cooldown(tmp_path):
|
|
store = PeerStore(tmp_path / "peer_store.json")
|
|
store.upsert(
|
|
make_sync_peer_record(
|
|
peer_url="https://node.shadowbroker.info",
|
|
transport="clearnet",
|
|
role="seed",
|
|
source="bundle",
|
|
now=100,
|
|
)
|
|
)
|
|
failed = store.mark_failure(
|
|
"https://node.shadowbroker.info",
|
|
"sync",
|
|
error="timed out",
|
|
cooldown_s=120,
|
|
now=110,
|
|
)
|
|
assert failed.cooldown_until == 230
|
|
|
|
refreshed = store.upsert(
|
|
make_sync_peer_record(
|
|
peer_url="https://node.shadowbroker.info",
|
|
transport="clearnet",
|
|
role="seed",
|
|
source="bundle",
|
|
now=120,
|
|
)
|
|
)
|
|
|
|
assert refreshed.failure_count == 0
|
|
assert refreshed.cooldown_until == 0
|
|
assert refreshed.last_error == ""
|