Fix formatting

Author: ddworken

src/mcp/server/fastmcp/server.py

@@ -119,7 +119,7 @@ class Settings(BaseSettings, Generic[LifespanResultT]):
     )
 
     auth: AuthSettings | None = None
-    
+
     # Transport security settings (DNS rebinding protection)
     transport_security: TransportSecuritySettings | None = None
 

src/mcp/server/sse.py

@@ -77,13 +77,11 @@ class SseServerTransport:
     _read_stream_writers: dict[UUID, MemoryObjectSendStream[SessionMessage | Exception]]
     _security: TransportSecurityMiddleware
 
-    def __init__(
-        self, endpoint: str, security_settings: TransportSecuritySettings | None = None
-    ) -> None:
+    def __init__(self, endpoint: str, security_settings: TransportSecuritySettings | None = None) -> None:
         """
         Creates a new SSE server transport, which will direct the client to POST
         messages to the relative or absolute URL given.
-        
+
         Args:
             endpoint: The relative or absolute URL for POST messages.
             security_settings: Optional security settings for DNS rebinding protection.
@@ -178,7 +176,7 @@ async def response_wrapper(scope: Scope, receive: Receive, send: Send):
     async def handle_post_message(self, scope: Scope, receive: Receive, send: Send) -> None:
         logger.debug("Handling POST message")
         request = Request(scope, receive)
-        
+
         # Validate request headers for DNS rebinding protection
         error_response = await self._security.validate_request(request, is_post=True)
         if error_response:

src/mcp/server/streamable_http.py

@@ -256,14 +256,14 @@ async def _clean_up_memory_streams(self, request_id: RequestId) -> None:
     async def handle_request(self, scope: Scope, receive: Receive, send: Send) -> None:
         """Application entry point that handles all HTTP requests"""
         request = Request(scope, receive)
-        
+
         # Validate request headers for DNS rebinding protection
         is_post = request.method == "POST"
         error_response = await self._security.validate_request(request, is_post=is_post)
         if error_response:
             await error_response(scope, receive, send)
             return
-            
+
         if self._terminated:
             # If the session has been terminated, return 404 Not Found
             response = self._create_error_response(

src/mcp/server/transport_security.py

@@ -11,7 +11,7 @@
 
 class TransportSecuritySettings(BaseModel):
     """Settings for MCP transport security features.
-    
+
     These settings help protect against DNS rebinding attacks by validating
     incoming request headers.
     """
@@ -40,9 +40,7 @@ class TransportSecurityMiddleware:
     def __init__(self, settings: TransportSecuritySettings | None = None):
         # If not specified, disable DNS rebinding protection by default
         # for backwards compatibility
-        self.settings = settings or TransportSecuritySettings(
-            enable_dns_rebinding_protection=False
-        )
+        self.settings = settings or TransportSecuritySettings(enable_dns_rebinding_protection=False)
 
     def _validate_host(self, host: str | None) -> bool:
         """Validate the Host header against allowed values."""
@@ -101,9 +99,7 @@ def _validate_content_type(self, content_type: str | None) -> bool:
 
         return True
 
-    async def validate_request(
-        self, request: Request, is_post: bool = False
-    ) -> Response | None:
+    async def validate_request(self, request: Request, is_post: bool = False) -> Response | None:
         """Validate request headers for DNS rebinding protection.
 
         Returns None if validation passes, or an error Response if validation fails.

tests/server/fastmcp/test_integration.py

@@ -84,8 +84,7 @@ def stateless_http_server_url(stateless_http_server_port: int) -> str:
 def make_fastmcp_app():
     """Create a FastMCP server without auth settings."""
     transport_security = TransportSecuritySettings(
-        allowed_hosts=["127.0.0.1:*", "localhost:*"],
-        allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
+        allowed_hosts=["127.0.0.1:*", "localhost:*"], allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
     )
     mcp = FastMCP(name="NoAuthServer", transport_security=transport_security)
 
@@ -101,10 +100,9 @@ def echo(message: str) -> str:
 
 
 def make_everything_fastmcp() -> FastMCP:
-    """Create a FastMCP server with all features enabled for testing."""    
+    """Create a FastMCP server with all features enabled for testing."""
     transport_security = TransportSecuritySettings(
-        allowed_hosts=["127.0.0.1:*", "localhost:*"],
-        allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
+        allowed_hosts=["127.0.0.1:*", "localhost:*"], allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
     )
     mcp = FastMCP(name="EverythingServer", transport_security=transport_security)
 
@@ -237,10 +235,9 @@ def make_everything_fastmcp_app():
 
 
 def make_fastmcp_streamable_http_app():
-    """Create a FastMCP server with StreamableHTTP transport."""    
+    """Create a FastMCP server with StreamableHTTP transport."""
     transport_security = TransportSecuritySettings(
-        allowed_hosts=["127.0.0.1:*", "localhost:*"],
-        allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
+        allowed_hosts=["127.0.0.1:*", "localhost:*"], allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
     )
     mcp = FastMCP(name="NoAuthServer", transport_security=transport_security)
 
@@ -266,16 +263,11 @@ def make_everything_fastmcp_streamable_http_app():
 
 
 def make_fastmcp_stateless_http_app():
-    """Create a FastMCP server with stateless StreamableHTTP transport."""    
+    """Create a FastMCP server with stateless StreamableHTTP transport."""
     transport_security = TransportSecuritySettings(
-        allowed_hosts=["127.0.0.1:*", "localhost:*"],
-        allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
-    )
-    mcp = FastMCP(
-        name="StatelessServer",
-        stateless_http=True,
-        transport_security=transport_security
+        allowed_hosts=["127.0.0.1:*", "localhost:*"], allowed_origins=["http://127.0.0.1:*", "http://localhost:*"]
     )
+    mcp = FastMCP(name="StatelessServer", stateless_http=True, transport_security=transport_security)
 
     # Add a simple tool
     @mcp.tool(description="A simple echo tool")