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

add correct mcp schema

Browse Source
This commit is contained in:
larssand
2026-06-21 22:27:13 +02:00
parent 35c21332f3
commit 64a7996b65
2 changed files with 20 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/test_graylog_source.py
View File

@@ -12,7 +12,7 @@ class _Client:
def call_tool(self, _name, arguments):
self.arguments = arguments
return {"result": {"content": [{"type": "text", "text": '{"messages":[{"client":"10.0.0.1","server":"8.8.8.8","result":"deny"}]}' }]}}
return {"result": {"content": [{"type": "text", "text": '{"schema":[{"field":"client"},{"field":"server"},{"field":"result"}],"datarows":[["10.0.0.1","8.8.8.8","deny"],["10.0.0.2","8.8.8.8","accept"]]}' }]}}
class GraylogSourceTests(unittest.TestCase):
@@ -21,12 +21,13 @@ class GraylogSourceTests(unittest.TestCase):
events, status = GraylogStreamSource(
client, "vpn", field_mapping='{"srcip":"client","dstip":"server","action":"result"}'
).fetch()
self.assertEqual(status["events_fetched"], 1)
self.assertEqual(status["events_fetched"], 2)
self.assertEqual(events[0].src_ip, "10.0.0.1")
self.assertEqual(events[0].dst_ip, "8.8.8.8")
self.assertEqual(events[0].action, "deny")
self.assertEqual(client.arguments["streams"], ["vpn"])
self.assertEqual(client.arguments["range_seconds"], 300)
self.assertIn("client", client.arguments["fields"])
if __name__ == "__main__":