mcp-security-framework 1.1.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

tests/test_core/test_cert_manager.py

@@ -98,7 +98,8 @@ class TestCertificateManager:
         """Test initialization with missing CA certificate."""
         config = CertificateConfig(
             enabled=True,
-            ca_cert_path="/nonexistent/ca.crt", ca_key_path=self.ca_key_path
+            ca_cert_path="/nonexistent/ca.crt",
+            ca_key_path=self.ca_key_path,
         )
 
         with pytest.raises(CertificateConfigurationError) as exc_info:
@@ -110,7 +111,8 @@ class TestCertificateManager:
         """Test initialization with missing CA key."""
         config = CertificateConfig(
             enabled=True,
-            ca_cert_path=self.ca_cert_path, ca_key_path="/nonexistent/ca.key"
+            ca_cert_path=self.ca_cert_path,
+            ca_key_path="/nonexistent/ca.key",
         )
 
         with pytest.raises(CertificateConfigurationError) as exc_info:
@@ -139,9 +141,9 @@ class TestCertificateManager:
                     mock_cert = Mock()
                     mock_cert.serial_number = 123456789
                     mock_cert.not_valid_before_utc = datetime.now(timezone.utc)
-                    mock_cert.not_valid_after_utc = datetime.now(timezone.utc) + timedelta(
-                        days=3650
-                    )
+                    mock_cert.not_valid_after_utc = datetime.now(
+                        timezone.utc
+                    ) + timedelta(days=3650)
                     mock_cert.public_bytes.return_value = b"-----BEGIN CERTIFICATE-----\nMOCK CERT\n-----END CERTIFICATE-----"
 
                     mock_private_key = Mock()
@@ -239,7 +241,9 @@ class TestCertificateManager:
                                 # Mock the certificate building process
                                 mock_cert = Mock()
                                 mock_cert.serial_number = 987654321
-                                mock_cert.not_valid_before_utc = datetime.now(timezone.utc)
+                                mock_cert.not_valid_before_utc = datetime.now(
+                                    timezone.utc
+                                )
                                 mock_cert.not_valid_after_utc = datetime.now(
                                     timezone.utc
                                 ) + timedelta(days=365)
@@ -340,7 +344,9 @@ class TestCertificateManager:
                                 # Mock the certificate building process
                                 mock_cert = Mock()
                                 mock_cert.serial_number = 555666777
-                                mock_cert.not_valid_before_utc = datetime.now(timezone.utc)
+                                mock_cert.not_valid_before_utc = datetime.now(
+                                    timezone.utc
+                                )
                                 mock_cert.not_valid_after_utc = datetime.now(
                                     timezone.utc
                                 ) + timedelta(days=365)
@@ -600,7 +606,11 @@ class TestCertificateManager:
                                 )
 
         assert isinstance(cert_info, CertificateInfo)
-        assert cert_info.subject == {"CN": "test.client.com", "C": "test.client.com", "O": "test.client.com"}
+        assert cert_info.subject == {
+            "CN": "test.client.com",
+            "C": "test.client.com",
+            "O": "test.client.com",
+        }
         assert cert_info.issuer == {"CN": "Test Root CA"}
         assert cert_info.serial_number == "123456789"
         assert cert_info.roles == ["user"]
@@ -651,14 +661,18 @@ class TestCertificateManager:
     def test_validate_configuration_missing_ca_cert_path(self):
         """Test configuration validation with missing CA certificate path."""
         with pytest.raises(ValidationError) as exc_info:
-            CertificateConfig(enabled=True, ca_cert_path="", ca_key_path=self.ca_key_path)
+            CertificateConfig(
+                enabled=True, ca_cert_path="", ca_key_path=self.ca_key_path
+            )
 
         assert "CA certificate and key paths are required" in str(exc_info.value)
 
     def test_validate_configuration_missing_ca_key_path(self):
         """Test configuration validation with missing CA key path."""
         with pytest.raises(ValidationError) as exc_info:
-            CertificateConfig(enabled=True, ca_cert_path=self.ca_cert_path, ca_key_path="")
+            CertificateConfig(
+                enabled=True, ca_cert_path=self.ca_cert_path, ca_key_path=""
+            )
 
         assert "CA certificate and key paths are required" in str(exc_info.value)
 

tests/test_core/test_security_manager.py

@@ -772,11 +772,11 @@ class TestSecurityManager:
         # Test factory methods
         fastapi_middleware = security_manager.create_fastapi_middleware()
         assert fastapi_middleware is not None
-        assert hasattr(fastapi_middleware, '__call__')
+        assert hasattr(fastapi_middleware, "__call__")
 
         flask_middleware = security_manager.create_flask_middleware()
         assert flask_middleware is not None
-        assert hasattr(flask_middleware, '__call__')
+        assert hasattr(flask_middleware, "__call__")
 
         # Django middleware should raise ImportError since it's not implemented
         with pytest.raises((ImportError, SecurityConfigurationError)):