@@ -25,76 +25,76 @@ def mock_box_lifespan():
2525 return mock_lifespan
2626
2727
28- class TestGetMcpServer :
29- """Test the get_mcp_server function."""
30-
31- @patch ("mcp_server_box.FastMCP" )
32- @patch ("mcp_server_box.box_lifespan" )
33- def test_get_mcp_server_stdio_default (self , mock_lifespan , mock_fastmcp_class ):
34- """Test get_mcp_server with default stdio transport."""
35- mock_instance = Mock ()
36- mock_fastmcp_class .return_value = mock_instance
37-
38- result = get_mcp_server ()
39-
40- mock_fastmcp_class .assert_called_once_with (
41- "Box MCP Server" , lifespan = mock_lifespan
42- )
43- assert result == mock_instance
44-
45- @patch ("mcp_server_box.FastMCP" )
46- @patch ("mcp_server_box.box_lifespan" )
47- def test_get_mcp_server_stdio_explicit (self , mock_lifespan , mock_fastmcp_class ):
48- """Test get_mcp_server with explicit stdio transport."""
49- mock_instance = Mock ()
50- mock_fastmcp_class .return_value = mock_instance
51-
52- result = get_mcp_server (server_name = "Custom Server" , transport = "stdio" )
53-
54- mock_fastmcp_class .assert_called_once_with (
55- "Custom Server" , lifespan = mock_lifespan
56- )
57- assert result == mock_instance
58-
59- @patch ("mcp_server_box.FastMCP" )
60- @patch ("mcp_server_box.box_lifespan" )
61- def test_get_mcp_server_sse_transport (self , mock_lifespan , mock_fastmcp_class ):
62- """Test get_mcp_server with SSE transport."""
63- mock_instance = Mock ()
64- mock_fastmcp_class .return_value = mock_instance
65-
66- result = get_mcp_server (
67- server_name = "SSE Server" , transport = "sse" , host = "192.168.1.1" , port = 9000
68- )
69-
70- mock_fastmcp_class .assert_called_once_with (
71- "SSE Server" ,
72- stateless_http = True ,
73- host = "192.168.1.1" ,
74- port = 9000 ,
75- lifespan = mock_lifespan ,
76- )
77- assert result == mock_instance
78-
79- @patch ("mcp_server_box.FastMCP" )
80- @patch ("mcp_server_box.box_lifespan" )
81- def test_get_mcp_server_http_transport (self , mock_lifespan , mock_fastmcp_class ):
82- """Test get_mcp_server with HTTP transport."""
83- mock_instance = Mock ()
84- mock_fastmcp_class .return_value = mock_instance
85-
86- result = get_mcp_server (
87- server_name = "HTTP Server" , transport = "http" , host = "localhost" , port = 8080
88- )
89-
90- mock_fastmcp_class .assert_called_once_with (
91- "HTTP Server" ,
92- stateless_http = True ,
93- host = "localhost" ,
94- port = 8080 ,
95- lifespan = mock_lifespan ,
96- )
97- assert result == mock_instance
28+ # class TestGetMcpServer:
29+ # """Test the get_mcp_server function."""
30+
31+ # @patch("mcp_server_box.FastMCP")
32+ # @patch("mcp_server_box.box_lifespan")
33+ # def test_get_mcp_server_stdio_default(self, mock_lifespan, mock_fastmcp_class):
34+ # """Test get_mcp_server with default stdio transport."""
35+ # mock_instance = Mock()
36+ # mock_fastmcp_class.return_value = mock_instance
37+
38+ # result = get_mcp_server()
39+
40+ # mock_fastmcp_class.assert_called_once_with(
41+ # "Box MCP Server", lifespan=mock_lifespan
42+ # )
43+ # assert result == mock_instance
44+
45+ # @patch("mcp_server_box.FastMCP")
46+ # @patch("mcp_server_box.box_lifespan")
47+ # def test_get_mcp_server_stdio_explicit(self, mock_lifespan, mock_fastmcp_class):
48+ # """Test get_mcp_server with explicit stdio transport."""
49+ # mock_instance = Mock()
50+ # mock_fastmcp_class.return_value = mock_instance
51+
52+ # result = get_mcp_server(server_name="Custom Server", transport="stdio")
53+
54+ # mock_fastmcp_class.assert_called_once_with(
55+ # "Custom Server", lifespan=mock_lifespan
56+ # )
57+ # assert result == mock_instance
58+
59+ # @patch("mcp_server_box.FastMCP")
60+ # @patch("mcp_server_box.box_lifespan")
61+ # def test_get_mcp_server_sse_transport(self, mock_lifespan, mock_fastmcp_class):
62+ # """Test get_mcp_server with SSE transport."""
63+ # mock_instance = Mock()
64+ # mock_fastmcp_class.return_value = mock_instance
65+
66+ # result = get_mcp_server(
67+ # server_name="SSE Server", transport="sse", host="192.168.1.1", port=9000
68+ # )
69+
70+ # mock_fastmcp_class.assert_called_once_with(
71+ # "SSE Server",
72+ # stateless_http=True,
73+ # host="192.168.1.1",
74+ # port=9000,
75+ # lifespan=mock_lifespan,
76+ # )
77+ # assert result == mock_instance
78+
79+ # @patch("mcp_server_box.FastMCP")
80+ # @patch("mcp_server_box.box_lifespan")
81+ # def test_get_mcp_server_http_transport(self, mock_lifespan, mock_fastmcp_class):
82+ # """Test get_mcp_server with HTTP transport."""
83+ # mock_instance = Mock()
84+ # mock_fastmcp_class.return_value = mock_instance
85+
86+ # result = get_mcp_server(
87+ # server_name="HTTP Server", transport="http", host="localhost", port=8080
88+ # )
89+
90+ # mock_fastmcp_class.assert_called_once_with(
91+ # "HTTP Server",
92+ # stateless_http=True,
93+ # host="localhost",
94+ # port=8080,
95+ # lifespan=mock_lifespan,
96+ # )
97+ # assert result == mock_instance
9898
9999
100100class TestMainExecution :
0 commit comments