155 lines
4.7 KiB
Python
155 lines
4.7 KiB
Python
"""Tests for tui_gateway.event_replay — per-session event seq + replay ring."""
|
|
|
|
import threading
|
|
|
|
import pytest
|
|
|
|
from tui_gateway import event_replay
|
|
from tui_gateway.event_replay import (
|
|
latest_seq,
|
|
reset_replay_state,
|
|
events_since,
|
|
replay_stats,
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _clean():
|
|
reset_replay_state()
|
|
yield
|
|
reset_replay_state()
|
|
|
|
|
|
def _frame(sid, etype="message.delta"):
|
|
return {
|
|
"jsonrpc": "2.0",
|
|
"method": "event",
|
|
"params": {"type": etype, "session_id": sid, "payload": {}},
|
|
}
|
|
|
|
|
|
def test_stamp_adds_monotonic_seq_per_session():
|
|
f1 = _frame("s1")
|
|
f2 = _frame("s1")
|
|
other = _frame("s2")
|
|
|
|
event_replay._stamp_event(f1)
|
|
event_replay._stamp_event(other)
|
|
event_replay._stamp_event(f2)
|
|
|
|
assert f1["params"]["seq"] == 1
|
|
assert f2["params"]["seq"] == 2 # per-session counter, unaffected by s2
|
|
assert other["params"]["seq"] == 1
|
|
|
|
|
|
def test_stamp_ignores_non_event_and_sessionless_frames():
|
|
rpc = {"jsonrpc": "2.0", "id": 1, "result": {}}
|
|
no_sid = {"jsonrpc": "2.0", "method": "event", "params": {"type": "skin.changed"}}
|
|
|
|
event_replay._stamp_event(rpc)
|
|
event_replay._stamp_event(no_sid)
|
|
|
|
assert "seq" not in rpc
|
|
assert "seq" not in no_sid["params"]
|
|
assert replay_stats()["events"] == 0
|
|
|
|
|
|
def test_events_since_returns_only_newer_frames_in_order():
|
|
frames = [_frame("s1") for _ in range(5)]
|
|
for f in frames:
|
|
event_replay._stamp_event(f)
|
|
|
|
got = events_since("s1", 3)
|
|
assert [e["seq"] for e in got] == [4, 5]
|
|
assert events_since("s1", 0) == [f["params"] for f in frames]
|
|
assert events_since("s1", 99) == []
|
|
assert latest_seq("s1") == 5
|
|
|
|
|
|
def test_events_since_returns_client_dispatchable_event_objects():
|
|
"""Cross-language contract: the client's replay loop dispatches an element
|
|
only when it has a TOP-LEVEL ``type`` (json-rpc-gateway.ts fetchReplay:
|
|
``if (!event?.type) continue``). Returning full JSON-RPC envelopes here
|
|
makes every replayed event silently droppable — the original #94219 bug.
|
|
"""
|
|
event_replay._stamp_event(_frame("s1"))
|
|
(event,) = events_since("s1", 0)
|
|
|
|
# Bare event object, not an envelope.
|
|
assert event["type"] == "message.delta"
|
|
assert event["session_id"] == "s1"
|
|
assert event["seq"] == 1
|
|
assert "jsonrpc" not in event
|
|
assert "method" not in event
|
|
assert "params" not in event
|
|
|
|
|
|
def test_unknown_session_returns_empty():
|
|
assert events_since("nope", 0) == []
|
|
assert latest_seq("nope") == 0
|
|
|
|
|
|
def test_ring_buffer_is_bounded():
|
|
for i in range(event_replay._REPLAY_BUFFER_MAX + 50):
|
|
event_replay._stamp_event(_frame("s1"))
|
|
|
|
stats = replay_stats()
|
|
assert stats["events"] == event_replay._REPLAY_BUFFER_MAX
|
|
# Oldest evicted: last_seen=0 must report truncation via the RPC contract.
|
|
buf = event_replay._replay_buffers["s1"]
|
|
assert buf[0][0] > 1
|
|
|
|
|
|
def test_session_count_bounded_with_fifo_eviction():
|
|
for i in range(event_replay._REPLAY_SESSIONS_MAX + 10):
|
|
event_replay._stamp_event(_frame(f"s{i}"))
|
|
|
|
stats = replay_stats()
|
|
assert stats["sessions"] == event_replay._REPLAY_SESSIONS_MAX
|
|
assert events_since("s0", 0) == [] # oldest session fully evicted
|
|
assert latest_seq(f"s{event_replay._REPLAY_SESSIONS_MAX + 9}") == 1
|
|
|
|
|
|
def test_concurrent_stamping_never_drops_or_duplicates_seq():
|
|
errors = []
|
|
|
|
def worker(sid):
|
|
try:
|
|
seen = set()
|
|
for _ in range(200):
|
|
f = _frame(sid)
|
|
event_replay._stamp_event(f)
|
|
seq = f["params"]["seq"]
|
|
assert seq not in seen
|
|
seen.add(seq)
|
|
except AssertionError as exc: # pragma: no cover
|
|
errors.append(exc)
|
|
|
|
threads = [threading.Thread(target=worker, args=(f"t{i}",)) for i in range(8)]
|
|
for t in threads:
|
|
t.start()
|
|
for t in threads:
|
|
t.join()
|
|
|
|
assert not errors
|
|
assert replay_stats()["events"] == 8 * 200
|
|
|
|
|
|
def test_truncation_detection_semantics():
|
|
"""The RPC handler's truncated flag: gap between last_seen and buffer start."""
|
|
# Overflow the ring so the oldest events are genuinely evicted.
|
|
for _ in range(event_replay._REPLAY_BUFFER_MAX + 10):
|
|
event_replay._stamp_event(_frame("s1"))
|
|
|
|
with event_replay._replay_lock:
|
|
oldest = event_replay._replay_buffers["s1"][0][0]
|
|
|
|
assert oldest > 1 # eviction happened
|
|
|
|
# Client saw everything up to just before the buffer → NOT truncated.
|
|
assert not event_replay.is_truncated("s1", oldest - 1)
|
|
# Client saw seq 5, buffer starts later → truncated.
|
|
assert event_replay.is_truncated("s1", 5)
|
|
# Unknown session: nothing evicted, nothing truncated.
|
|
assert not event_replay.is_truncated("nope", 0)
|