@simbimbo/brainstem 0.0.3 → 0.0.4

package/tests/test_adapters.py

@@ -0,0 +1,94 @@
+from brainstem.adapters import (
+    RawInputAdapter,
+    SyslogRawInputAdapter,
+    build_raw_input_envelope,
+    get_raw_input_adapter,
+    list_raw_input_source_types,
+    register_raw_input_adapter,
+)
+import pytest
+from brainstem.ingest import parse_file_line
+from brainstem.ingest import parse_syslog_envelope
+from brainstem.models import RawInputEnvelope
+
+
+def test_syslog_adapter_registry_is_populated() -> None:
+    sources = list_raw_input_source_types()
+    assert "syslog" in sources
+    assert "file" in sources
+    assert isinstance(get_raw_input_adapter("syslog"), RawInputAdapter)
+
+
+def test_file_adapter_builds_raw_envelope() -> None:
+    raw_line = "vpn tunnel dropped and recovered"
+    raw = build_raw_input_envelope(
+        "file",
+        raw_line,
+        tenant_id="client-a",
+        source_path="/tmp/sample.log",
+    )
+    assert isinstance(raw, RawInputEnvelope)
+    assert raw.source_type == "file"
+    assert raw.message_raw == raw_line
+    assert raw.metadata["raw_line"] == raw_line
+    assert raw.source_path == "/tmp/sample.log"
+
+
+def test_syslog_adapter_builds_raw_envelope() -> None:
+    raw_line = "Mar 22 00:00:01 fw-01 charon: VPN tunnel dropped and recovered"
+    raw = build_raw_input_envelope(
+        "syslog",
+        raw_line,
+        tenant_id="client-a",
+        source_path="/var/log/syslog",
+    )
+    assert isinstance(raw, RawInputEnvelope)
+    assert raw.source_type == "syslog"
+    assert raw.host == "fw-01"
+    assert raw.service == "charon"
+    assert raw.message_raw == "VPN tunnel dropped and recovered"
+    assert raw.metadata["raw_line"] == raw_line
+    assert raw.source_path == "/var/log/syslog"
+
+
+def test_file_adapter_parse_path() -> None:
+    raw = parse_file_line("first line", tenant_id="client-a", source_path="/tmp/file.log")
+    assert raw.message_raw == "first line"
+    assert raw.source_type == "file"
+    assert raw.metadata["raw_line"] == "first line"
+
+
+def test_syslog_dispatch_is_used_by_ingest_parse_path() -> None:
+    class StubAdapter:
+        source_type = "syslog"
+        calls = 0
+
+        def parse_raw_input(self, payload, *, tenant_id: str, source_path: str = "") -> RawInputEnvelope:
+            StubAdapter.calls += 1
+            return RawInputEnvelope(
+                tenant_id=tenant_id,
+                source_type="syslog",
+                timestamp="2026-03-22T00:00:00Z",
+                message_raw="stub-message",
+                host="host",
+                service="service",
+                source_path=source_path,
+            )
+
+    original = get_raw_input_adapter("syslog")
+    try:
+        register_raw_input_adapter(StubAdapter())
+        raw = parse_syslog_envelope("ignored-line", tenant_id="client-a", source_path="/tmp/test.log")
+        assert raw.message_raw == "stub-message"
+        assert StubAdapter.calls == 1
+    finally:
+        if original is not None:
+            register_raw_input_adapter(original)
+        else:
+            # defensive: restore default adapter if registration state changed unexpectedly
+            register_raw_input_adapter(SyslogRawInputAdapter())
+
+
+def test_unknown_source_type_raises_clear_error() -> None:
+    with pytest.raises(ValueError, match="unsupported source_type"):
+        build_raw_input_envelope("not-a-source", "payload", tenant_id="client-a")