33import pytest
44import asyncio
55from unittest .mock import Mock , patch , AsyncMock
6- from jupyter_server_docs_mcp .extension import MCPExtensionApp
6+ from jupyter_server_mcp .extension import MCPExtensionApp
77
88
99class TestMCPExtensionApp :
@@ -13,7 +13,7 @@ def test_extension_creation(self):
1313 """Test creating extension with defaults."""
1414 extension = MCPExtensionApp ()
1515
16- assert extension .name == "jupyter_server_docs_mcp "
16+ assert extension .name == "jupyter_server_mcp "
1717 assert extension .mcp_port == 3001
1818 assert extension .mcp_name == "Jupyter MCP Server"
1919
@@ -57,7 +57,7 @@ async def test_start_extension_success(self):
5757 extension .mcp_name = "Test Server"
5858
5959 # Mock the MCP server creation to avoid actual server startup
60- with patch ('jupyter_server_docs_mcp .extension.MCPServer' ) as mock_mcp_class :
60+ with patch ('jupyter_server_mcp .extension.MCPServer' ) as mock_mcp_class :
6161 mock_server = Mock ()
6262 mock_server .start_server = AsyncMock ()
6363 mock_mcp_class .return_value = mock_server
@@ -82,7 +82,7 @@ async def test_start_extension_failure(self):
8282 extension = MCPExtensionApp ()
8383
8484 # Mock server creation to raise an exception
85- with patch ('jupyter_server_docs_mcp .extension.MCPServer' ) as mock_mcp_class :
85+ with patch ('jupyter_server_mcp .extension.MCPServer' ) as mock_mcp_class :
8686 mock_mcp_class .side_effect = Exception ("Server creation failed" )
8787
8888 with pytest .raises (Exception , match = "Server creation failed" ):
@@ -147,7 +147,7 @@ async def test_full_lifecycle(self):
147147 extension .mcp_name = "Lifecycle Test Server"
148148
149149 # Mock the MCP server
150- with patch ('jupyter_server_docs_mcp .extension.MCPServer' ) as mock_mcp_class :
150+ with patch ('jupyter_server_mcp .extension.MCPServer' ) as mock_mcp_class :
151151 mock_server = Mock ()
152152 mock_server .start_server = AsyncMock ()
153153 mock_mcp_class .return_value = mock_server
@@ -261,7 +261,7 @@ def test_register_configured_tools_valid(self):
261261
262262 # Capture log output
263263 import logging
264- with patch ('jupyter_server_docs_mcp .extension.logger' ) as mock_logger :
264+ with patch ('jupyter_server_mcp .extension.logger' ) as mock_logger :
265265 extension ._register_configured_tools ()
266266
267267 # Should register both tools
@@ -278,7 +278,7 @@ def test_register_configured_tools_with_errors(self):
278278 extension .mcp_server_instance = Mock ()
279279 extension .mcp_tools = ["os:getcwd" , "invalid:function" , "math:sqrt" ]
280280
281- with patch ('jupyter_server_docs_mcp .extension.logger' ) as mock_logger :
281+ with patch ('jupyter_server_mcp .extension.logger' ) as mock_logger :
282282 extension ._register_configured_tools ()
283283
284284 # Should register 2 valid tools (os:getcwd and math:sqrt)
@@ -299,7 +299,7 @@ async def test_start_extension_with_tools(self):
299299 extension .mcp_name = "Test Server With Tools"
300300 extension .mcp_tools = ["os:getcwd" , "math:sqrt" ]
301301
302- with patch ('jupyter_server_docs_mcp .extension.MCPServer' ) as mock_mcp_class :
302+ with patch ('jupyter_server_mcp .extension.MCPServer' ) as mock_mcp_class :
303303 mock_server = Mock ()
304304 mock_server .start_server = AsyncMock ()
305305 mock_server ._registered_tools = {"getcwd" : {}, "sqrt" : {}} # Mock registered tools
@@ -324,7 +324,7 @@ async def test_start_extension_no_tools(self):
324324 extension .mcp_port = 3088
325325 extension .mcp_tools = []
326326
327- with patch ('jupyter_server_docs_mcp .extension.MCPServer' ) as mock_mcp_class :
327+ with patch ('jupyter_server_mcp .extension.MCPServer' ) as mock_mcp_class :
328328 mock_server = Mock ()
329329 mock_server .start_server = AsyncMock ()
330330 mock_server ._registered_tools = {}
0 commit comments