anti
f2b3393669
Replaces LICENSE (GPLv3 -> AGPLv3) and prepends `SPDX-License-Identifier: AGPL-3.0-or-later` to every source file across decnet/, decnet_web/, tests/, scripts/, and tools/. Rationale: closes the GPLv3 ASP loophole so any party operating a modified DECNET as a network service must offer their modified source. Personal copyright (Samuel Paschuan) + inbound=outbound contributions make a future unilateral relicense infeasible. - LICENSE: full AGPL-3.0 text (gnu.org/licenses/agpl-3.0.txt) - COPYRIGHT: project copyright notice - tools/add_spdx_headers.py: idempotent header injector (shebang- and PEP 263-aware) Touches 1565 source files (.py, .ts, .tsx, .js, .jsx, .css, .sh). No behavior change; comments only.
133 lines
4.3 KiB
Python
133 lines
4.3 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""End-to-end tests for :class:`UnixSocketBus` against a real :class:`BusServer`.
|
|
|
|
These tests run in the dev loop (no pytest marker) because they only need
|
|
the tmp filesystem — no Docker, no external broker.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import pathlib
|
|
import stat
|
|
|
|
import pytest
|
|
|
|
from decnet.bus.unix_client import UnixSocketBus
|
|
from decnet.bus.unix_server import BusServer
|
|
|
|
|
|
async def _drain(sub, n: int, timeout: float = 1.5) -> list:
|
|
out = []
|
|
try:
|
|
async with asyncio.timeout(timeout):
|
|
async for event in sub:
|
|
out.append(event)
|
|
if len(out) >= n:
|
|
break
|
|
except TimeoutError:
|
|
pass
|
|
return out
|
|
|
|
|
|
class TestEndToEnd:
|
|
async def test_pub_sub_exact(self, unix_bus) -> None:
|
|
server, client = unix_bus
|
|
sub = client.subscribe("topology.abc.status")
|
|
# Give the SUB frame a tick to register on the server.
|
|
await asyncio.sleep(0.05)
|
|
async with sub:
|
|
await client.publish("topology.abc.status", {"status": "active"})
|
|
events = await _drain(sub, 1)
|
|
# A publisher doesn't see its own events — use a second client.
|
|
assert events == []
|
|
|
|
async def test_pub_sub_across_two_clients(
|
|
self, tmp_path: pathlib.Path,
|
|
) -> None:
|
|
sock = tmp_path / "bus.sock"
|
|
server = BusServer(sock, group=None)
|
|
await server.start()
|
|
serve_task = asyncio.create_task(server.serve_forever())
|
|
|
|
publisher = UnixSocketBus(sock, client_name="publisher")
|
|
subscriber = UnixSocketBus(sock, client_name="subscriber")
|
|
await publisher.connect()
|
|
await subscriber.connect()
|
|
|
|
try:
|
|
sub = subscriber.subscribe("topology.*.mutation.*")
|
|
await asyncio.sleep(0.05) # let SUB register
|
|
|
|
async with sub:
|
|
await publisher.publish(
|
|
"topology.t1.mutation.applied", {"id": 1}, event_type="applied",
|
|
)
|
|
await publisher.publish(
|
|
"decky.xyz.state", {"state": "running"}, # should not match
|
|
)
|
|
await publisher.publish(
|
|
"topology.t2.mutation.failed", {"id": 2}, event_type="failed",
|
|
)
|
|
events = await _drain(sub, 2)
|
|
ids = {e.payload["id"] for e in events}
|
|
assert ids == {1, 2}
|
|
finally:
|
|
await publisher.close()
|
|
await subscriber.close()
|
|
serve_task.cancel()
|
|
try:
|
|
await serve_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
await server.close()
|
|
|
|
async def test_socket_file_mode(self, tmp_path: pathlib.Path) -> None:
|
|
sock = tmp_path / "bus.sock"
|
|
server = BusServer(sock, group=None)
|
|
await server.start()
|
|
try:
|
|
mode = stat.S_IMODE(sock.stat().st_mode)
|
|
assert mode == 0o660
|
|
finally:
|
|
await server.close()
|
|
|
|
async def test_server_close_wakes_subscribers(
|
|
self, tmp_path: pathlib.Path,
|
|
) -> None:
|
|
sock = tmp_path / "bus.sock"
|
|
server = BusServer(sock, group=None)
|
|
await server.start()
|
|
serve_task = asyncio.create_task(server.serve_forever())
|
|
|
|
client = UnixSocketBus(sock, client_name="watcher")
|
|
await client.connect()
|
|
sub = client.subscribe("system.>")
|
|
await asyncio.sleep(0.05)
|
|
|
|
async def consume() -> list:
|
|
out = []
|
|
async for event in sub:
|
|
out.append(event)
|
|
return out
|
|
|
|
consumer = asyncio.create_task(consume())
|
|
await asyncio.sleep(0.05)
|
|
|
|
serve_task.cancel()
|
|
try:
|
|
await serve_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
await server.close()
|
|
|
|
# The consumer must unblock within a reasonable time.
|
|
events = await asyncio.wait_for(consumer, timeout=1.0)
|
|
assert events == []
|
|
await client.close()
|
|
|
|
async def test_start_rejects_missing_parent(self, tmp_path: pathlib.Path) -> None:
|
|
sock = tmp_path / "nonexistent-dir" / "bus.sock"
|
|
server = BusServer(sock, group=None)
|
|
with pytest.raises(FileNotFoundError):
|
|
await server.start()
|