Update tests

Author: ddworken

tests/server/fastmcp/test_integration.py

@@ -22,7 +22,7 @@
 from mcp.client.session import ClientSession
 from mcp.client.sse import sse_client
 from mcp.client.streamable_http import streamablehttp_client
-from mcp.server.fastmcp import FastMCP
+from mcp.server.fastmcp import Context, FastMCP
 from mcp.server.fastmcp.resources import FunctionResource
 from mcp.server.transport_security import TransportSecuritySettings
 from mcp.shared.context import RequestContext
@@ -101,10 +101,7 @@ def echo(message: str) -> str:
 
 
 def make_everything_fastmcp() -> FastMCP:
-    """Create a FastMCP server with all features enabled for testing."""
-    from mcp.server.fastmcp import Context
-    from mcp.server.transport_security import TransportSecuritySettings
-    
+    """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:*"]

tests/server/test_streamable_http_security.py

@@ -263,23 +263,6 @@ async def test_streamable_http_security_custom_allowed_hosts(server_port: int):
             )
             # Should connect successfully with custom host
             assert response.status_code == 200
-            
-        # Test with non-allowed host
-        headers = {
-            "Host": "evil.com",
-            "Accept": "application/json, text/event-stream",
-            "Content-Type": "application/json",
-        }
-        
-        async with httpx.AsyncClient(timeout=5.0) as client:
-            response = await client.post(
-                f"http://127.0.0.1:{server_port}/",
-                json={"jsonrpc": "2.0", "method": "initialize", "id": 1, "params": {}},
-                headers=headers
-            )
-            assert response.status_code == 400, response.text
-            assert response.text == "Invalid Host header"
-                
     finally:
         process.terminate()
         process.join()