84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from daena.config import (
|
|
DaenaConfig,
|
|
MetricsConfig,
|
|
ProcessorConfig,
|
|
SinkConfig,
|
|
SourceConfig,
|
|
)
|
|
from daena.core.layout import preview_message, render_layout
|
|
|
|
|
|
def _config() -> DaenaConfig:
|
|
config = DaenaConfig()
|
|
config.persistence.lmdb.path = "/tmp/daena/data"
|
|
config.sources = [
|
|
SourceConfig(name="dummy", type="dummy", options={"interval": 2.0}),
|
|
SourceConfig(
|
|
name="system",
|
|
type="journald",
|
|
enabled=False,
|
|
options={"units": ["ssh.service"]},
|
|
),
|
|
]
|
|
config.processors = [
|
|
ProcessorConfig(name="host", type="enricher", options={"static": {"env": "local"}}),
|
|
ProcessorConfig(name="keep", type="filter", options={"drop": {"body.priority": "7"}}),
|
|
]
|
|
config.sinks = [
|
|
SinkConfig(name="log", type="dummy"),
|
|
SinkConfig(
|
|
name="loki",
|
|
type="loki",
|
|
options={"url": "http://127.0.0.1:3100/loki/api/v1/push"},
|
|
),
|
|
]
|
|
config.logging.verbose = True
|
|
return config
|
|
|
|
|
|
def test_layout_is_a_single_pipeline_line() -> None:
|
|
text = render_layout(_config(), "./daena.yaml")
|
|
assert "┌" not in text
|
|
assert "▼" not in text
|
|
flow = text.splitlines()[1]
|
|
assert flow.index("dummy") < flow.index("host")
|
|
assert flow.index("host") < flow.index("lmdb")
|
|
assert flow.index("lmdb") < flow.index("log")
|
|
assert "daena 0.2.0 ./daena.yaml" in text
|
|
assert "/tmp/daena/data" in text
|
|
assert "127.0.0.1:8642 127.0.0.1:8643" in text
|
|
|
|
|
|
def test_layout_names_each_plugin_once() -> None:
|
|
text = render_layout(_config(), None)
|
|
flow = text.splitlines()[1]
|
|
assert flow == " → ".join(
|
|
[
|
|
"dummy (2s), system (journald, off)",
|
|
"host (enricher), keep (filter)",
|
|
"lmdb",
|
|
"log (dummy), loki (127.0.0.1:3100)",
|
|
]
|
|
)
|
|
assert "daena 0.2.0 defaults" in text
|
|
|
|
|
|
def test_layout_skips_empty_stages_and_disabled_metrics() -> None:
|
|
config = DaenaConfig()
|
|
config.api.metrics = MetricsConfig(enabled=False)
|
|
text = render_layout(config, None)
|
|
assert "→ lmdb" not in text
|
|
assert text.splitlines()[1] == "lmdb"
|
|
assert text.splitlines()[-1] == "127.0.0.1:8642"
|
|
|
|
|
|
def test_preview_message_uses_message_then_clips() -> None:
|
|
assert preview_message({"message": "daena heartbeat #1"}) == "daena heartbeat #1"
|
|
assert preview_message({"message": "a\nb"}) == "a b"
|
|
assert preview_message({"counter": 1}) == '{"counter":1}'
|
|
clipped = preview_message({"message": "x" * 130}, limit=10)
|
|
assert clipped.endswith("…")
|
|
assert len(clipped) == 10
|