mcp-proxy-adapter 6.3.3__py3-none-any.whl → 6.3.5__py3-none-any.whl

mcp_proxy_adapter/examples/security_test_client.py

@@ -34,6 +34,7 @@ sys.path.insert(0, str(parent_dir))
 try:
     from mcp_security_framework import SSLManager, CertificateManager
     from mcp_security_framework.schemas.config import SSLConfig
+
     _MCP_SECURITY_AVAILABLE = True
     print("✅ mcp_security_framework available")
 except ImportError:
@@ -44,14 +45,18 @@ except ImportError:
 try:
     from cryptography import x509
     from cryptography.hazmat.primitives import serialization
+
     _CRYPTOGRAPHY_AVAILABLE = True
     print("✅ cryptography available")
 except ImportError:
     _CRYPTOGRAPHY_AVAILABLE = False
     print("⚠️ cryptography not available, SSL validation will be limited")
+
+
 @dataclass
 class TestResult:
     """Test result data class."""
+
     test_name: str
     server_url: str
     auth_type: str
@@ -60,8 +65,11 @@ class TestResult:
     response_data: Optional[Dict] = None
     error_message: Optional[str] = None
     duration: float = 0.0
+
+
 class SecurityTestClient:
     """Security test client for comprehensive testing."""
+
     def __init__(self, base_url: str = "http://localhost:8000"):
         """Initialize security test client."""
         self.base_url = base_url
@@ -82,7 +90,7 @@ class SecurityTestClient:
                     key_file=None,
                     ca_cert_file=None,
                     verify_mode="CERT_NONE",  # For testing
-                    min_tls_version="TLSv1.2"
+                    min_tls_version="TLSv1.2",
                 )
                 self.ssl_manager = SSLManager(ssl_config)
                 print("✅ SSL Manager initialized with mcp_security_framework")
@@ -101,23 +109,24 @@ class SecurityTestClient:
             "readonly": "readonly-token-123",
             "guest": "guest-token-123",
             "proxy": "proxy-token-123",
-            "invalid": "invalid-token-999"
+            "invalid": "invalid-token-999",
         }
         # Test certificates
         self.test_certificates = {
             "admin": {
                 "cert": "mcp_proxy_adapter/examples/certs/admin_cert.pem",
-                "key": "mcp_proxy_adapter/examples/certs/admin_key.pem"
+                "key": "mcp_proxy_adapter/examples/certs/admin_key.pem",
             },
             "user": {
                 "cert": "mcp_proxy_adapter/examples/certs/user_cert.pem",
-                "key": "mcp_proxy_adapter/examples/certs/user_key.pem"
+                "key": "mcp_proxy_adapter/examples/certs/user_key.pem",
             },
             "readonly": {
                 "cert": "mcp_proxy_adapter/examples/certs/readonly_cert.pem",
-                "key": "mcp_proxy_adapter/examples/certs/readonly_key.pem"
-            }
+                "key": "mcp_proxy_adapter/examples/certs/readonly_key.pem",
+            },
         }
+
     async def __aenter__(self):
         """Async context manager entry."""
         timeout = ClientTimeout(total=30)
@@ -126,6 +135,7 @@ class SecurityTestClient:
         connector = TCPConnector(ssl=ssl_context)
         self.session = ClientSession(timeout=timeout, connector=connector)
         return self
+
     def create_ssl_context_for_mtls(self) -> ssl.SSLContext:
         """Create SSL context for mTLS connections."""
         if self.ssl_manager and self._security_available:
@@ -140,10 +150,12 @@ class SecurityTestClient:
                     client_cert_file=cert_file if os.path.exists(cert_file) else None,
                     client_key_file=key_file if os.path.exists(key_file) else None,
                     verify_mode="CERT_NONE",  # For testing
-                    min_version="TLSv1.2"
+                    min_version="TLSv1.2",
                 )
             except Exception as e:
-                print(f"⚠️ Failed to create mTLS context with mcp_security_framework: {e}")
+                print(
+                    f"⚠️ Failed to create mTLS context with mcp_security_framework: {e}"
+                )
                 print("ℹ️ Falling back to standard SSL")
 
         # Fallback to standard SSL
@@ -160,26 +172,41 @@ class SecurityTestClient:
         if os.path.exists(ca_cert_file):
             ssl_context.load_verify_locations(cafile=ca_cert_file)
         return ssl_context
+
     async def __aexit__(self, exc_type, exc_val, exc_tb):
         """Async context manager exit."""
         if self.session:
             await self.session.close()
-    def create_ssl_context(self, cert_file: Optional[str] = None,
-                          key_file: Optional[str] = None,
-                          ca_cert_file: Optional[str] = None) -> ssl.SSLContext:
+
+    def create_ssl_context(
+        self,
+        cert_file: Optional[str] = None,
+        key_file: Optional[str] = None,
+        ca_cert_file: Optional[str] = None,
+    ) -> ssl.SSLContext:
         """Create SSL context for client."""
         if self.ssl_manager and self._security_available:
             try:
                 # Use mcp_security_framework for SSL context creation
                 return self.ssl_manager.create_client_context(
-                    ca_cert_file=ca_cert_file if ca_cert_file and os.path.exists(ca_cert_file) else None,
-                    client_cert_file=cert_file if cert_file and os.path.exists(cert_file) else None,
-                    client_key_file=key_file if key_file and os.path.exists(key_file) else None,
+                    ca_cert_file=(
+                        ca_cert_file
+                        if ca_cert_file and os.path.exists(ca_cert_file)
+                        else None
+                    ),
+                    client_cert_file=(
+                        cert_file if cert_file and os.path.exists(cert_file) else None
+                    ),
+                    client_key_file=(
+                        key_file if key_file and os.path.exists(key_file) else None
+                    ),
                     verify_mode="CERT_NONE",  # For testing
-                    min_version="TLSv1.2"
+                    min_version="TLSv1.2",
                 )
             except Exception as e:
-                print(f"⚠️ Failed to create SSL context with mcp_security_framework: {e}")
+                print(
+                    f"⚠️ Failed to create SSL context with mcp_security_framework: {e}"
+                )
                 print("ℹ️ Falling back to standard SSL")
 
         # Fallback to standard SSL
@@ -194,6 +221,7 @@ class SecurityTestClient:
             # For testing, still don't verify
             ssl_context.verify_mode = ssl.CERT_NONE
         return ssl_context
+
     def create_auth_headers(self, auth_type: str, **kwargs) -> Dict[str, str]:
         """Create authentication headers."""
         headers = {"Content-Type": "application/json"}
@@ -206,6 +234,7 @@ class SecurityTestClient:
             username = kwargs.get("username", "admin")
             password = kwargs.get("password", "password")
             import base64
+
             credentials = base64.b64encode(f"{username}:{password}".encode()).decode()
             headers["Authorization"] = f"Basic {credentials}"
         elif auth_type == "certificate":
@@ -213,13 +242,18 @@ class SecurityTestClient:
             # This is handled by SSL context, not headers
             pass
         return headers
-    async def test_health_check(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_health_check(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test health check endpoint."""
         start_time = time.time()
         test_name = f"Health Check ({auth_type})"
         try:
             headers = self.create_auth_headers(auth_type, **kwargs)
-            async with self.session.get(f"{server_url}/health", headers=headers) as response:
+            async with self.session.get(
+                f"{server_url}/health", headers=headers
+            ) as response:
                 duration = time.time() - start_time
                 if response.status == 200:
                     data = await response.json()
@@ -230,7 +264,7 @@ class SecurityTestClient:
                         success=True,
                         status_code=response.status,
                         response_data=data,
-                        duration=duration
+                        duration=duration,
                     )
                 else:
                     error_text = await response.text()
@@ -241,7 +275,7 @@ class SecurityTestClient:
                         success=False,
                         status_code=response.status,
                         error_message=f"Health check failed: {error_text}",
-                        duration=duration
+                        duration=duration,
                     )
         except Exception as e:
             duration = time.time() - start_time
@@ -251,9 +285,12 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Health check error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
-    async def test_echo_command(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_echo_command(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test echo command."""
         start_time = time.time()
         test_name = f"Echo Command ({auth_type})"
@@ -262,14 +299,12 @@ class SecurityTestClient:
             data = {
                 "jsonrpc": "2.0",
                 "method": "echo",
-                "params": {
-                    "message": "Hello from security test client!"
-                },
-                "id": 1
+                "params": {"message": "Hello from security test client!"},
+                "id": 1,
             }
-            async with self.session.post(f"{server_url}/cmd",
-                                        headers=headers,
-                                        json=data) as response:
+            async with self.session.post(
+                f"{server_url}/cmd", headers=headers, json=data
+            ) as response:
                 duration = time.time() - start_time
                 if response.status == 200:
                     data = await response.json()
@@ -280,7 +315,7 @@ class SecurityTestClient:
                         success=True,
                         status_code=response.status,
                         response_data=data,
-                        duration=duration
+                        duration=duration,
                     )
                 else:
                     error_text = await response.text()
@@ -291,7 +326,7 @@ class SecurityTestClient:
                         success=False,
                         status_code=response.status,
                         error_message=f"Echo command failed: {error_text}",
-                        duration=duration
+                        duration=duration,
                     )
         except Exception as e:
             duration = time.time() - start_time
@@ -301,23 +336,21 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Echo command error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
-    async def test_security_command(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_security_command(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test security command."""
         start_time = time.time()
         test_name = f"Security Command ({auth_type})"
         try:
             headers = self.create_auth_headers(auth_type, **kwargs)
-            data = {
-                "jsonrpc": "2.0",
-                "method": "health",
-                "params": {},
-                "id": 2
-            }
-            async with self.session.post(f"{server_url}/cmd",
-                                        headers=headers,
-                                        json=data) as response:
+            data = {"jsonrpc": "2.0", "method": "health", "params": {}, "id": 2}
+            async with self.session.post(
+                f"{server_url}/cmd", headers=headers, json=data
+            ) as response:
                 duration = time.time() - start_time
                 if response.status == 200:
                     data = await response.json()
@@ -328,7 +361,7 @@ class SecurityTestClient:
                         success=True,
                         status_code=response.status,
                         response_data=data,
-                        duration=duration
+                        duration=duration,
                     )
                 else:
                     error_text = await response.text()
@@ -339,7 +372,7 @@ class SecurityTestClient:
                         success=False,
                         status_code=response.status,
                         error_message=f"Security command failed: {error_text}",
-                        duration=duration
+                        duration=duration,
                     )
         except Exception as e:
             duration = time.time() - start_time
@@ -349,14 +382,17 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Security command error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
+
     async def test_health(self) -> TestResult:
         """Test health endpoint."""
         return await self.test_health_check(self.base_url, "none")
+
     async def test_command_execution(self) -> TestResult:
         """Test command execution."""
         return await self.test_echo_command(self.base_url, "none")
+
     async def test_authentication(self) -> TestResult:
         """Test authentication."""
         if "api_key" in self.auth_methods:
@@ -372,15 +408,22 @@ class SecurityTestClient:
                 server_url=self.base_url,
                 auth_type="none",
                 success=False,
-                error_message="No authentication method available"
+                error_message="No authentication method available",
             )
+
     async def test_negative_authentication(self) -> TestResult:
         """Test negative authentication (should fail)."""
-        return await self.test_echo_command(self.base_url, "api_key", token="invalid-token")
+        return await self.test_echo_command(
+            self.base_url, "api_key", token="invalid-token"
+        )
+
     async def test_no_auth_required(self) -> TestResult:
         """Test that no authentication is required."""
         return await self.test_echo_command(self.base_url, "none")
-    async def test_negative_auth(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_negative_auth(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test negative authentication scenarios."""
         start_time = time.time()
         test_name = f"Negative Auth ({auth_type})"
@@ -399,20 +442,26 @@ class SecurityTestClient:
                 ca_cert_file = "./certs/mcp_proxy_adapter_ca_ca.crt"
                 if os.path.exists(ca_cert_file):
                     ssl_context.load_verify_locations(cafile=ca_cert_file)
-                ssl_context.verify_mode = ssl.CERT_NONE  # Don't verify server cert for testing
+                ssl_context.verify_mode = (
+                    ssl.CERT_NONE
+                )  # Don't verify server cert for testing
 
                 connector = TCPConnector(ssl=ssl_context)
                 timeout = ClientTimeout(total=10)  # Shorter timeout
 
                 try:
-                    async with aiohttp.ClientSession(timeout=timeout, connector=connector) as temp_session:
+                    async with aiohttp.ClientSession(
+                        timeout=timeout, connector=connector
+                    ) as temp_session:
                         data = {
                             "jsonrpc": "2.0",
                             "method": "echo",
                             "params": {"message": "Should fail without certificate"},
-                            "id": 3
+                            "id": 3,
                         }
-                        async with temp_session.post(f"{server_url}/cmd", json=data) as response:
+                        async with temp_session.post(
+                            f"{server_url}/cmd", json=data
+                        ) as response:
                             duration = time.time() - start_time
                             # If we get here, the server accepted the connection without proper certificate
                             # This is actually a security issue - server should reject
@@ -423,9 +472,13 @@ class SecurityTestClient:
                                 success=False,
                                 status_code=response.status,
                                 error_message=f"SECURITY ISSUE: mTLS server accepted connection without client certificate (status: {response.status})",
-                                duration=duration
+                                duration=duration,
                             )
-                except (aiohttp.ClientError, aiohttp.ServerDisconnectedError, asyncio.TimeoutError) as e:
+                except (
+                    aiohttp.ClientError,
+                    aiohttp.ServerDisconnectedError,
+                    asyncio.TimeoutError,
+                ) as e:
                     # This is expected - server should reject connections without proper certificate
                     duration = time.time() - start_time
                     return TestResult(
@@ -434,8 +487,11 @@ class SecurityTestClient:
                         auth_type=auth_type,
                         success=True,
                         status_code=0,
-                        response_data={"expected": "connection_rejected", "error": str(e)},
-                        duration=duration
+                        response_data={
+                            "expected": "connection_rejected",
+                            "error": str(e),
+                        },
+                        duration=duration,
                     )
             else:
                 # For other auth types, use invalid token
@@ -444,11 +500,11 @@ class SecurityTestClient:
                     "jsonrpc": "2.0",
                     "method": "echo",
                     "params": {"message": "Should fail"},
-                    "id": 3
+                    "id": 3,
                 }
-                async with self.session.post(f"{server_url}/cmd",
-                                            headers=headers,
-                                            json=data) as response:
+                async with self.session.post(
+                    f"{server_url}/cmd", headers=headers, json=data
+                ) as response:
                     duration = time.time() - start_time
                     # Expect 401 only when auth is enforced
                     expects_auth = auth_type in ("api_key", "certificate", "basic")
@@ -460,7 +516,7 @@ class SecurityTestClient:
                             success=True,
                             status_code=response.status,
                             response_data={"expected": "authentication_failure"},
-                            duration=duration
+                            duration=duration,
                         )
                     elif not expects_auth and response.status == 200:
                         # Security disabled: negative auth should not fail
@@ -471,7 +527,7 @@ class SecurityTestClient:
                             success=True,
                             status_code=response.status,
                             response_data={"expected": "no_auth_required"},
-                            duration=duration
+                            duration=duration,
                         )
                     else:
                         return TestResult(
@@ -481,7 +537,7 @@ class SecurityTestClient:
                             success=False,
                             status_code=response.status,
                             error_message=f"Unexpected status for negative auth: {response.status}",
-                            duration=duration
+                            duration=duration,
                         )
         except Exception as e:
             duration = time.time() - start_time
@@ -491,9 +547,12 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Negative auth error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
-    async def test_role_based_access(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_role_based_access(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test role-based access control."""
         start_time = time.time()
         test_name = f"Role-Based Access ({auth_type})"
@@ -506,11 +565,11 @@ class SecurityTestClient:
                 "jsonrpc": "2.0",
                 "method": "echo",
                 "params": {"message": f"Testing {role} role"},
-                "id": 4
+                "id": 4,
             }
-            async with self.session.post(f"{server_url}/cmd",
-                                        headers=headers,
-                                        json=data) as response:
+            async with self.session.post(
+                f"{server_url}/cmd", headers=headers, json=data
+            ) as response:
                 duration = time.time() - start_time
                 if response.status == 200:
                     data = await response.json()
@@ -521,7 +580,7 @@ class SecurityTestClient:
                         success=True,
                         status_code=response.status,
                         response_data=data,
-                        duration=duration
+                        duration=duration,
                     )
                 else:
                     error_text = await response.text()
@@ -532,7 +591,7 @@ class SecurityTestClient:
                         success=False,
                         status_code=response.status,
                         error_message=f"Role-based access failed: {error_text}",
-                        duration=duration
+                        duration=duration,
                     )
         except Exception as e:
             duration = time.time() - start_time
@@ -542,9 +601,12 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Role-based access error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
-    async def test_role_permissions(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_role_permissions(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test role permissions with role_test command."""
         start_time = time.time()
         test_name = f"Role Permissions Test ({auth_type})"
@@ -558,11 +620,11 @@ class SecurityTestClient:
                 "jsonrpc": "2.0",
                 "method": "role_test",
                 "params": {"action": action},
-                "id": 5
+                "id": 5,
             }
-            async with self.session.post(f"{server_url}/cmd",
-                                        headers=headers,
-                                        json=data) as response:
+            async with self.session.post(
+                f"{server_url}/cmd", headers=headers, json=data
+            ) as response:
                 duration = time.time() - start_time
                 if response.status == 200:
                     data = await response.json()
@@ -573,7 +635,7 @@ class SecurityTestClient:
                         success=True,
                         status_code=response.status,
                         response_data=data,
-                        duration=duration
+                        duration=duration,
                     )
                 else:
                     error_text = await response.text()
@@ -584,7 +646,7 @@ class SecurityTestClient:
                         success=False,
                         status_code=response.status,
                         error_message=f"Role permissions test failed: {error_text}",
-                        duration=duration
+                        duration=duration,
                     )
         except Exception as e:
             duration = time.time() - start_time
@@ -594,9 +656,12 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Role permissions test error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
-    async def test_multiple_roles(self, server_url: str, auth_type: str = "none", **kwargs) -> TestResult:
+
+    async def test_multiple_roles(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> TestResult:
         """Test multiple roles with different permissions."""
         start_time = time.time()
         test_name = f"Multiple Roles Test ({auth_type})"
@@ -608,11 +673,11 @@ class SecurityTestClient:
                 "jsonrpc": "2.0",
                 "method": "role_test",
                 "params": {"action": "manage"},
-                "id": 6
+                "id": 6,
             }
-            async with self.session.post(f"{server_url}/cmd",
-                                        headers=admin_headers,
-                                        json=admin_data) as response:
+            async with self.session.post(
+                f"{server_url}/cmd", headers=admin_headers, json=admin_data
+            ) as response:
                 if response.status != 200:
                     return TestResult(
                         test_name=test_name,
@@ -621,19 +686,21 @@ class SecurityTestClient:
                         success=False,
                         status_code=response.status,
                         error_message="Admin role test failed",
-                        duration=time.time() - start_time
+                        duration=time.time() - start_time,
                     )
                 # Test readonly role (should only have read permission)
                 readonly_token = self.test_tokens.get("readonly", "readonly-token-123")
-                readonly_headers = self.create_auth_headers("api_key", token=readonly_token)
+                readonly_headers = self.create_auth_headers(
+                    "api_key", token=readonly_token
+                )
                 readonly_data = {
                     "jsonrpc": "2.0",
                     "method": "role_test",
                     "params": {"action": "write"},
                 }
-                async with self.session.post(f"{server_url}/cmd",
-                                            headers=readonly_headers,
-                                            json=readonly_data) as response:
+                async with self.session.post(
+                    f"{server_url}/cmd", headers=readonly_headers, json=readonly_data
+                ) as response:
                     duration = time.time() - start_time
                     # Readonly should be denied write access
                     if response.status == 403:
@@ -643,8 +710,10 @@ class SecurityTestClient:
                             auth_type=auth_type,
                             success=True,
                             status_code=response.status,
-                            response_data={"message": "Correctly denied write access to readonly role"},
-                            duration=duration
+                            response_data={
+                                "message": "Correctly denied write access to readonly role"
+                            },
+                            duration=duration,
                         )
                     else:
                         return TestResult(
@@ -654,7 +723,7 @@ class SecurityTestClient:
                             success=False,
                             status_code=response.status,
                             error_message="Readonly role incorrectly allowed write access",
-                            duration=duration
+                            duration=duration,
                         )
         except Exception as e:
             duration = time.time() - start_time
@@ -664,9 +733,12 @@ class SecurityTestClient:
                 auth_type=auth_type,
                 success=False,
                 error_message=f"Multiple roles test error: {str(e)}",
-                duration=duration
+                duration=duration,
             )
-    async def run_security_tests(self, server_url: str, auth_type: str = "none", **kwargs) -> List[TestResult]:
+
+    async def run_security_tests(
+        self, server_url: str, auth_type: str = "none", **kwargs
+    ) -> List[TestResult]:
         """Run comprehensive security tests."""
         print(f"\n🔒 Running security tests for {server_url} ({auth_type})")
         print("=" * 60)
@@ -675,7 +747,7 @@ class SecurityTestClient:
             self.test_echo_command(server_url, auth_type, **kwargs),
             self.test_security_command(server_url, auth_type, **kwargs),
             self.test_negative_auth(server_url, auth_type, **kwargs),
-            self.test_role_based_access(server_url, auth_type, **kwargs)
+            self.test_role_based_access(server_url, auth_type, **kwargs),
         ]
         results = []
         for test in tests:
@@ -692,44 +764,28 @@ class SecurityTestClient:
                 print(f"   Error: {result.error_message}")
             print()
         return results
+
     async def test_all_scenarios(self) -> Dict[str, List[TestResult]]:
         """Test all security scenarios."""
         scenarios = {
-            "basic_http": {
-                "url": "http://localhost:8000",
-                "auth": "none"
-            },
-            "http_token": {
-                "url": "http://localhost:8001",
-                "auth": "api_key"
-            },
-            "https": {
-                "url": "https://localhost:8443",
-                "auth": "none"
-            },
-            "https_token": {
-                "url": "https://localhost:8444",
-                "auth": "api_key"
-            },
-            "mtls": {
-                "url": "https://localhost:8445",
-                "auth": "certificate"
-            }
+            "basic_http": {"url": "http://localhost:8000", "auth": "none"},
+            "http_token": {"url": "http://localhost:8001", "auth": "api_key"},
+            "https": {"url": "https://localhost:8443", "auth": "none"},
+            "https_token": {"url": "https://localhost:8444", "auth": "api_key"},
+            "mtls": {"url": "https://localhost:8445", "auth": "certificate"},
         }
         all_results = {}
         for scenario_name, config in scenarios.items():
             print(f"\n🚀 Testing scenario: {scenario_name.upper()}")
             print("=" * 60)
             try:
-                results = await self.run_security_tests(
-                    config["url"],
-                    config["auth"]
-                )
+                results = await self.run_security_tests(config["url"], config["auth"])
                 all_results[scenario_name] = results
             except Exception as e:
                 print(f"❌ Failed to test {scenario_name}: {e}")
                 all_results[scenario_name] = []
         return all_results
+
     def print_summary(self):
         """Print test summary."""
         print("\n" + "=" * 80)
@@ -753,16 +809,27 @@ class SecurityTestClient:
         for result in self.test_results:
             if result.success:
                 print(f"  - {result.test_name} ({result.server_url})")
+
+
 async def main():
     """Main function."""
     import argparse
-    parser = argparse.ArgumentParser(description="Security Test Client for MCP Proxy Adapter")
-    parser.add_argument("--server", default="http://localhost:8000",
-                       help="Server URL to test")
-    parser.add_argument("--auth", choices=["none", "api_key", "basic", "certificate"],
-                       default="none", help="Authentication type")
-    parser.add_argument("--all-scenarios", action="store_true",
-                       help="Test all security scenarios")
+
+    parser = argparse.ArgumentParser(
+        description="Security Test Client for MCP Proxy Adapter"
+    )
+    parser.add_argument(
+        "--server", default="http://localhost:8000", help="Server URL to test"
+    )
+    parser.add_argument(
+        "--auth",
+        choices=["none", "api_key", "basic", "certificate"],
+        default="none",
+        help="Authentication type",
+    )
+    parser.add_argument(
+        "--all-scenarios", action="store_true", help="Test all security scenarios"
+    )
     parser.add_argument("--token", help="API token for authentication")
     parser.add_argument("--cert", help="Client certificate file")
     parser.add_argument("--key", help="Client private key file")
@@ -778,5 +845,7 @@ async def main():
         async with SecurityTestClient(args.server) as client:
             await client.run_security_tests(args.server, args.auth, token=args.token)
             client.print_summary()
+
+
 if __name__ == "__main__":
     asyncio.run(main())