| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,74 @@ | ||
| from unittest.mock import AsyncMock, Mock | ||
|
|
||
| import pytest | ||
|
|
||
| import mcp.types as types | ||
| from mcp.server.lowlevel.server import Server | ||
| from mcp.server.session import ServerSession | ||
| from mcp.shared.session import RequestResponder | ||
|
|
||
|
|
||
| @pytest.mark.anyio | ||
| async def test_exception_handling_with_raise_exceptions_true(): | ||
| """Test that exceptions are re-raised when raise_exceptions=True""" | ||
| server = Server("test-server") | ||
| session = Mock(spec=ServerSession) | ||
| session.send_log_message = AsyncMock() | ||
|
|
||
| test_exception = RuntimeError("Test error") | ||
|
|
||
| with pytest.raises(RuntimeError, match="Test error"): | ||
| await server._handle_message(test_exception, session, {}, raise_exceptions=True) | ||
|
|
||
| session.send_log_message.assert_called_once() | ||
|
|
||
|
|
||
| @pytest.mark.anyio | ||
| @pytest.mark.parametrize( | ||
| "exception_class,message", | ||
| [ | ||
| (ValueError, "Test validation error"), | ||
| (RuntimeError, "Test runtime error"), | ||
| (KeyError, "Test key error"), | ||
| (Exception, "Basic error"), | ||
| ], | ||
| ) | ||
| async def test_exception_handling_with_raise_exceptions_false(exception_class: type[Exception], message: str): | ||
| """Test that exceptions are logged when raise_exceptions=False""" | ||
| server = Server("test-server") | ||
| session = Mock(spec=ServerSession) | ||
| session.send_log_message = AsyncMock() | ||
|
|
||
| test_exception = exception_class(message) | ||
|
|
||
| await server._handle_message(test_exception, session, {}, raise_exceptions=False) | ||
|
|
||
| # Should send log message | ||
| session.send_log_message.assert_called_once() | ||
| call_args = session.send_log_message.call_args | ||
|
|
||
| assert call_args.kwargs["level"] == "error" | ||
| assert call_args.kwargs["data"] == "Internal Server Error" | ||
| assert call_args.kwargs["logger"] == "mcp.server.exception_handler" | ||
|
|
||
|
|
||
| @pytest.mark.anyio | ||
| async def test_normal_message_handling_not_affected(): | ||
| """Test that normal messages still work correctly""" | ||
| server = Server("test-server") | ||
| session = Mock(spec=ServerSession) | ||
|
|
||
| # Create a mock RequestResponder | ||
| responder = Mock(spec=RequestResponder) | ||
| responder.request = types.ClientRequest(root=types.PingRequest(method="ping")) | ||
| responder.__enter__ = Mock(return_value=responder) | ||
| responder.__exit__ = Mock(return_value=None) | ||
|
|
||
| # Mock the _handle_request method to avoid complex setup | ||
| server._handle_request = AsyncMock() | ||
|
|
||
| # Should handle normally without any exception handling | ||
| await server._handle_message(responder, session, {}, raise_exceptions=False) | ||
|
|
||
| # Verify _handle_request was called | ||
| server._handle_request.assert_called_once() |
| Back | FazBrowse Home | New Git URL |
Uh oh!
There was an error while loading. Please reload this page.