add more matching

This commit is contained in:
larssand
2026-06-22 19:19:31 +02:00
parent a5489d3c87
commit f2257c74de
8 changed files with 85 additions and 5 deletions

15
tests/test_correlation.py Normal file
View File

@@ -0,0 +1,15 @@
import unittest
from fgai.correlation import correlate_source_ips
from fgai.logs import parse_log_line
class CorrelationTests(unittest.TestCase):
def test_correlates_same_source_across_streams(self):
events = [
parse_log_line("srcip=10.0.0.5 fgai_stream=Fortigate action=blocked"),
parse_log_line("srcip=10.0.0.5 fgai_stream=DNS action=deny"),
]
result = correlate_source_ips(events)
self.assertEqual(result[0]["source_ip"], "10.0.0.5")
self.assertEqual(result[0]["streams"], ["DNS", "Fortigate"])

10
tests/test_metrics.py Normal file
View File

@@ -0,0 +1,10 @@
import unittest
from fgai.metrics import prometheus_metrics
class MetricsTests(unittest.TestCase):
def test_renders_low_cardinality_metrics(self):
output = prometheus_metrics({"summary": {"total": 10}, "anomaly_summary": {"high": 2}, "baseline": {"sources_ready": 3}, "capabilities": {"graylog_mcp": {"status": "connected", "events_fetched": 9}}})
self.assertIn('fgai_events_total{kind="total"} 10', output)
self.assertIn("fgai_graylog_mcp_connected 1", output)