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

mcp connect issue cert

Browse Source
This commit is contained in:
larssand
2026-06-30 21:37:17 +02:00
parent 6a63de0e37
commit 24561ac984
2 changed files with 28 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
tests/test_graylog_mcp.py
View File

@@ -1,7 +1,7 @@
import json
import unittest
from urllib.error import HTTPError
from unittest.mock import patch
from unittest.mock import Mock, patch
from fgai.graylog_mcp import GraylogMcpClient
@@ -34,7 +34,7 @@ class GraylogMcpTests(unittest.TestCase):
_Response({}),
_Response({"result": {"tools": [{"name": "search_messages"}, {"name": "aggregate_messages"}]}}),
])
with patch("fgai.graylog_mcp.request.urlopen", side_effect=responses):
with patch("fgai.graylog_mcp.GraylogMcpClient._open", side_effect=responses):
status = GraylogMcpClient("http://graylog/api/mcp", "raw-token").probe()
self.assertEqual(status["status"], "connected")
self.assertIn("search_messages", status["tools"])
@@ -48,7 +48,7 @@ class GraylogMcpTests(unittest.TestCase):
client = GraylogMcpClient("http://graylog/api/mcp", "raw-token")
client.session_id = "existing-session"
with patch("fgai.graylog_mcp.request.urlopen", side_effect=responses):
with patch("fgai.graylog_mcp.GraylogMcpClient._open", side_effect=responses):
client.probe()
self.assertEqual(client.session_id, "session-1")
@@ -64,12 +64,25 @@ class GraylogMcpTests(unittest.TestCase):
client = GraylogMcpClient("http://graylog/api/mcp", "raw-token")
client.session_id = "old-session"
with patch("fgai.graylog_mcp.request.urlopen", side_effect=responses):
with patch("fgai.graylog_mcp.GraylogMcpClient._open", side_effect=responses):
result = client.call_tool("search_messages", {"query": "*"})
self.assertEqual(result["result"]["content"], [])
self.assertEqual(client.session_id, "session-2")
def test_default_open_bypasses_environment_proxies(self):
client = GraylogMcpClient("http://graylog/api/mcp", "raw-token")
req = object()
opener = Mock()
opener.open.return_value = "response"
with patch("fgai.graylog_mcp.request.build_opener", return_value=opener) as build_opener:
result = client._open(req, None)
self.assertEqual(result, "response")
self.assertEqual(opener.open.call_args.kwargs["timeout"], 15)
self.assertEqual(build_opener.call_args.args[0].proxies, {})
if __name__ == "__main__":
unittest.main()