Fix linting issues in resource leak tests

- Remove unused asyncio import
- Remove unused exception variables

Author: certainly-param

test_resource_leak_reproduction.py

@@ -26,10 +26,11 @@
 
 class MockResponse:
     """Simple mock to track if aclose() gets called"""
+
     def __init__(self):
         self.closed = False
         self.close_count = 0
-    
+
     async def aclose(self):
         self.closed = True
         self.close_count += 1
@@ -38,27 +39,29 @@ async def aclose(self):
 
 class MockEventSource:
     """Mock that throws an exception to simulate broken SSE"""
+
     def __init__(self, response):
         self.response = response
-    
+
     def __aiter__(self):
         return self
-    
+
     async def __anext__(self):
         # Simulate what happens when SSE parsing fails
         raise Exception("SSE parsing failed - connection broken")
 
 
 class MockTransport(StreamableHTTPTransport):
     """Mock that shows the same bug as the real code"""
+
     def __init__(self):
         super().__init__("http://test")
         self.mock_response = MockResponse()
-    
+
     async def _handle_sse_response(self, response, ctx, is_initialization=False):
         """
         This mimics the actual bug in the real code.
-        
+
         The problem: when the async for loop throws an exception,
         response.aclose() never gets called because it's only in the success path.
         """
@@ -74,7 +77,7 @@ async def _handle_sse_response(self, response, ctx, is_initialization=False):
             print(f"Exception caught: {e}")
             # Here's the bug - response.aclose() is never called!
             raise
-    
+
     async def _handle_resumption_request(self, ctx):
         """
         Same issue here - the aconnect_sse context manager should handle cleanup,
@@ -85,20 +88,20 @@ async def _handle_resumption_request(self, ctx):
             class MockEventSourceWithResponse:
                 def __init__(self, response):
                     self.response = response
-                
+
                 async def __aenter__(self):
                     return self
-                
+
                 async def __aexit__(self, exc_type, exc_val, exc_tb):
                     # Context manager exits but response might not be closed
                     pass
-                
+
                 def __aiter__(self):
                     return self
-                
+
                 async def __anext__(self):
                     raise Exception("Resumption SSE parsing failed")
-            
+
             async with MockEventSourceWithResponse(self.mock_response) as event_source:
                 async for sse in event_source:
                     # This never runs because the exception happens first
@@ -116,25 +119,25 @@ async def test_resource_leak():
     """Test the resource leak I found"""
     print("Testing resource leak in streamable_http.py")
     print("=" * 50)
-    
+
     transport = MockTransport()
-    
+
     # Create mock context
     class MockContext:
         def __init__(self):
             self.read_stream_writer = None
             self.metadata = None
-    
+
     ctx = MockContext()
-    
+
     print("\nTesting _handle_sse_response method:")
     print("-" * 35)
-    
+
     try:
         await transport._handle_sse_response(transport.mock_response, ctx)
     except Exception as e:
         print(f"Caught expected exception: {e}")
-    
+
     # Check if response was closed
     if transport.mock_response.closed:
         print("No resource leak - response was closed properly")
@@ -151,27 +154,27 @@ async def test_resumption_resource_leak():
     """Test the resource leak in _handle_resumption_request"""
     print("\nTesting _handle_resumption_request method:")
     print("-" * 40)
-    
+
     transport = MockTransport()
-    
+
     # Create mock context with resumption token
     class MockResumptionContext:
         def __init__(self):
             self.read_stream_writer = None
-            self.metadata = type('obj', (object,), {'resumption_token': 'test-token'})()
-            self.session_message = type('obj', (object,), {
-                'message': type('obj', (object,), {
-                    'root': type('obj', (object,), {'id': 'test-id'})()
-                })()
-            })()
-    
+            self.metadata = type("obj", (object,), {"resumption_token": "test-token"})()
+            self.session_message = type(
+                "obj",
+                (object,),
+                {"message": type("obj", (object,), {"root": type("obj", (object,), {"id": "test-id"})()})()},
+            )()
+
     ctx_resumption = MockResumptionContext()
-    
+
     try:
         await transport._handle_resumption_request(ctx_resumption)
     except Exception as e:
         print(f"Caught expected exception: {e}")
-    
+
     # Check if response was closed
     if transport.mock_response.closed:
         print("No resource leak - response was closed properly")
@@ -190,15 +193,15 @@ async def main():
     print("This shows the issue I found where HTTP responses don't get closed")
     print("when SSE streaming fails in the MCP Python SDK.")
     print()
-    
+
     # Test both methods
     sse_leak = await test_resource_leak()
     resumption_leak = await test_resumption_resource_leak()
-    
+
     print("\n" + "=" * 50)
     print("SUMMARY:")
     print("=" * 50)
-    
+
     if sse_leak and resumption_leak:
         print("All tests passed - no resource leaks detected")
         return 0