feat(bus): host-local UNIX-socket pub/sub worker (DEBT-029)
Land the `decnet bus` worker and `get_bus()` factory. Transport is a host-local UNIX-domain socket (0660, group=decnet); authz is the file mode. Wire framing is a tiny verb-line + 4-byte-BE length + orjson body. NATS-style wildcard topics (`*`, `>`). At-most-once, fire-and-forget — DB stays the source of truth. `FakeBus` / `NullBus` for tests and the disabled path. Cross-host federation is deferred to a future `--bridge-tcp` mode; DEBT-030 is master-only and unblocked.
This commit is contained in:
131
tests/bus/test_unix_socket_bus.py
Normal file
131
tests/bus/test_unix_socket_bus.py
Normal file
@@ -0,0 +1,131 @@
|
||||
"""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()
|
||||
Reference in New Issue
Block a user