Public/fgAI
SHA256
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix poll mcp

Browse Source
This commit is contained in:
larssand
2026-07-02 21:03:37 +02:00
parent 825bd0247a
commit 7d1bdd35c9
6 changed files with 58 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_config.py
View File

@@ -38,6 +38,16 @@ class ConfigTests(unittest.TestCase):
self.assertEqual(public["graylog_fetch_mode"], "aggregate")
self.assertEqual(public["graylog_raw_sample_events"], 2500)
def test_graylog_mcp_timeout_settings_are_numeric_and_bounded(self):
with tempfile.TemporaryDirectory() as directory:
store = ConfigStore(str(Path(directory) / "config.json"))
public = store.update({"graylog_mcp_call_timeout_seconds": "0", "graylog_mcp_poll_timeout_seconds": "30"})
self.assertEqual(public["graylog_mcp_call_timeout_seconds"], 1)
self.assertEqual(public["graylog_mcp_poll_timeout_seconds"], 60)
public = store.update({"graylog_mcp_call_timeout_seconds": "6", "graylog_mcp_poll_timeout_seconds": "180"})
self.assertEqual(public["graylog_mcp_call_timeout_seconds"], 6)
self.assertEqual(public["graylog_mcp_poll_timeout_seconds"], 180)
def test_graylog_tls_verify_can_be_disabled(self):
with tempfile.TemporaryDirectory() as directory:
store = ConfigStore(str(Path(directory) / "config.json"))

5
tests/test_graylog_source.py
View File

@@ -62,6 +62,11 @@ class GraylogSourceTests(unittest.TestCase):
self.assertEqual(client.arguments["limit"], 2)
self.assertTrue(status["truncated"])
def test_uses_larger_single_page_for_raw_sample(self):
client = _Client()
GraylogStreamSource(client, "vpn").fetch(max_events=5000)
self.assertEqual(client.arguments["limit"], 5000)
def test_returns_partial_status_instead_of_raising_on_search_error(self):
events, status = GraylogStreamSource(_ErrorClient(), "vpn").fetch()
self.assertEqual(events, [])