fix poll mcp

2026-07-02 21:03:37 +02:00
parent 825bd0247a
commit 7d1bdd35c9
6 changed files with 58 additions and 8 deletions
--- a/tests/test_graylog_source.py
+++ b/tests/test_graylog_source.py
@@ -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, [])