remove infra.md.example, infra.md is the source of truth

ayn-antivirus/tests/test_event_bus.py

@@ -0,0 +1,117 @@
+"""Tests for the event bus pub/sub system."""
+import pytest
+
+from ayn_antivirus.core.event_bus import EventBus, EventType
+
+
+def test_subscribe_and_publish():
+    bus = EventBus()
+    received = []
+    bus.subscribe(EventType.THREAT_FOUND, lambda et, data: received.append(data))
+    bus.publish(EventType.THREAT_FOUND, {"test": True})
+    assert len(received) == 1
+    assert received[0]["test"] is True
+
+
+def test_multiple_subscribers():
+    bus = EventBus()
+    r1, r2 = [], []
+    bus.subscribe(EventType.SCAN_STARTED, lambda et, d: r1.append(d))
+    bus.subscribe(EventType.SCAN_STARTED, lambda et, d: r2.append(d))
+    bus.publish(EventType.SCAN_STARTED, "go")
+    assert len(r1) == 1
+    assert len(r2) == 1
+
+
+def test_unsubscribe():
+    bus = EventBus()
+    received = []
+    cb = lambda et, d: received.append(d)
+    bus.subscribe(EventType.FILE_SCANNED, cb)
+    bus.unsubscribe(EventType.FILE_SCANNED, cb)
+    bus.publish(EventType.FILE_SCANNED, "data")
+    assert len(received) == 0
+
+
+def test_unsubscribe_nonexistent():
+    """Unsubscribing a callback that was never registered should not crash."""
+    bus = EventBus()
+    bus.unsubscribe(EventType.FILE_SCANNED, lambda et, d: None)
+
+
+def test_publish_no_subscribers():
+    """Publishing with no subscribers should not crash."""
+    bus = EventBus()
+    bus.publish(EventType.SCAN_COMPLETED, "no crash")
+
+
+def test_subscriber_exception_isolated():
+    """A failing subscriber must not prevent other subscribers from running."""
+    bus = EventBus()
+    received = []
+    bus.subscribe(EventType.THREAT_FOUND, lambda et, d: 1 / 0)  # will raise
+    bus.subscribe(EventType.THREAT_FOUND, lambda et, d: received.append(d))
+    bus.publish(EventType.THREAT_FOUND, "data")
+    assert len(received) == 1
+
+
+def test_all_event_types():
+    """Every EventType value can be published without error."""
+    bus = EventBus()
+    for et in EventType:
+        bus.publish(et, None)
+
+
+def test_clear_all():
+    bus = EventBus()
+    received = []
+    bus.subscribe(EventType.THREAT_FOUND, lambda et, d: received.append(d))
+    bus.subscribe(EventType.SCAN_STARTED, lambda et, d: received.append(d))
+    bus.clear()
+    bus.publish(EventType.THREAT_FOUND, "a")
+    bus.publish(EventType.SCAN_STARTED, "b")
+    assert len(received) == 0
+
+
+def test_clear_single_event():
+    bus = EventBus()
+    r1, r2 = [], []
+    bus.subscribe(EventType.THREAT_FOUND, lambda et, d: r1.append(d))
+    bus.subscribe(EventType.SCAN_STARTED, lambda et, d: r2.append(d))
+    bus.clear(EventType.THREAT_FOUND)
+    bus.publish(EventType.THREAT_FOUND, "a")
+    bus.publish(EventType.SCAN_STARTED, "b")
+    assert len(r1) == 0  # cleared
+    assert len(r2) == 1  # still active
+
+
+def test_callback_receives_event_type():
+    """Callback receives (event_type, data) — verify event_type is correct."""
+    bus = EventBus()
+    calls = []
+    bus.subscribe(EventType.QUARANTINE_ACTION, lambda et, d: calls.append((et, d)))
+    bus.publish(EventType.QUARANTINE_ACTION, "payload")
+    assert calls[0][0] is EventType.QUARANTINE_ACTION
+    assert calls[0][1] == "payload"
+
+
+def test_duplicate_subscribe():
+    """Subscribing the same callback twice should only register it once."""
+    bus = EventBus()
+    received = []
+    cb = lambda et, d: received.append(d)
+    bus.subscribe(EventType.SCAN_COMPLETED, cb)
+    bus.subscribe(EventType.SCAN_COMPLETED, cb)
+    bus.publish(EventType.SCAN_COMPLETED, "x")
+    assert len(received) == 1
+
+
+def test_event_type_values():
+    """All expected event types exist."""
+    expected = {
+        "THREAT_FOUND", "SCAN_STARTED", "SCAN_COMPLETED", "FILE_SCANNED",
+        "SIGNATURE_UPDATED", "QUARANTINE_ACTION", "REMEDIATION_ACTION",
+        "DASHBOARD_METRIC",
+    }
+    actual = {et.name for et in EventType}
+    assert expected == actual