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

tests/test_cli/test_cert_cli.py

@@ -4,10 +4,11 @@ Certificate CLI Tests
 This module contains tests for the certificate management CLI commands.
 """
 
-import pytest
-import tempfile
 import os
-from unittest.mock import Mock, patch, mock_open, MagicMock
+import tempfile
+from unittest.mock import MagicMock, Mock, mock_open, patch
+
+import pytest
 from click.testing import CliRunner
 
 from mcp_security_framework.cli.cert_cli import cert_cli
@@ -17,34 +18,35 @@ from mcp_security_framework.schemas.models import CertificatePair, CertificateTy
 
 class TestCertCLI:
     """Test suite for certificate CLI commands."""
-    
+
     def setup_method(self):
         """Set up test fixtures."""
         self.runner = CliRunner()
         self.temp_dir = tempfile.mkdtemp()
-        
+
         # Create test certificate files
         self.test_cert_path = os.path.join(self.temp_dir, "test.crt")
         self.test_key_path = os.path.join(self.temp_dir, "test.key")
-        
-        with open(self.test_cert_path, 'w') as f:
+
+        with open(self.test_cert_path, "w") as f:
             f.write("-----BEGIN CERTIFICATE-----\nTEST CERT\n-----END CERTIFICATE-----")
-        
-        with open(self.test_key_path, 'w') as f:
+
+        with open(self.test_key_path, "w") as f:
             f.write("-----BEGIN PRIVATE KEY-----\nTEST KEY\n-----END PRIVATE KEY-----")
-    
+
     def teardown_method(self):
         """Clean up test fixtures."""
         import shutil
+
         shutil.rmtree(self.temp_dir, ignore_errors=True)
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_ca_success(self, mock_cert_manager_class):
         """Test successful CA certificate creation."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
-        
+
         # Mock certificate pair
         mock_cert_pair = Mock(spec=CertificatePair)
         mock_cert_pair.certificate_path = "/path/to/ca.crt"
@@ -52,101 +54,107 @@ class TestCertCLI:
         mock_cert_pair.serial_number = "123456789"
         mock_cert_pair.not_after = "2025-01-01"
         mock_cert_manager.create_root_ca.return_value = mock_cert_pair
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'create-ca',
-            '--common-name', 'Test CA',
-            '--organization', 'Test Org',
-            '--country', 'US'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "create-ca",
+                "--common-name",
+                "Test CA",
+                "--organization",
+                "Test Org",
+                "--country",
+                "US",
+            ],
+        )
+
         # Assertions
         assert result.exit_code == 0
         assert "✅ CA certificate created successfully!" in result.output
         # Note: The output doesn't include the input parameters, only the result
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_ca_failure(self, mock_cert_manager_class):
         """Test CA certificate creation failure."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.create_root_ca.side_effect = Exception("Test error")
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'create-ca',
-            '--common-name', 'Test CA',
-            '--organization', 'Test Org',
-            '--country', 'US'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "create-ca",
+                "--common-name",
+                "Test CA",
+                "--organization",
+                "Test Org",
+                "--country",
+                "US",
+            ],
+        )
+
         # Assertions
         assert result.exit_code != 0
         assert "❌ Failed to create CA certificate" in result.output
-    
+
     def test_create_server_success(self):
         """Test successful server certificate creation."""
         # Create a simple test that doesn't require complex mocking
-        result = self.runner.invoke(cert_cli, ['--help'])
-        
+        result = self.runner.invoke(cert_cli, ["--help"])
+
         # Assertions - just check that CLI is working
         assert result.exit_code == 0
         assert "Certificate Management CLI" in result.output
-    
+
     def test_create_client_success(self):
         """Test successful client certificate creation."""
         # Create a simple test that doesn't require complex mocking
-        result = self.runner.invoke(cert_cli, ['create-client', '--help'])
-        
+        result = self.runner.invoke(cert_cli, ["create-client", "--help"])
+
         # Assertions - just check that CLI command is available
         assert result.exit_code == 0
         assert "Create a client certificate" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_validate_success(self, mock_cert_manager_class):
         """Test successful certificate validation."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.validate_certificate_chain.return_value = True
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'validate',
-            self.test_cert_path
-        ])
-        
+        result = self.runner.invoke(cert_cli, ["validate", self.test_cert_path])
+
         # Assertions
         assert result.exit_code == 0
         assert "✅ Certificate is valid!" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_validate_failure(self, mock_cert_manager_class):
         """Test certificate validation failure."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.validate_certificate_chain.return_value = False
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'validate',
-            self.test_cert_path
-        ])
-        
+        result = self.runner.invoke(cert_cli, ["validate", self.test_cert_path])
+
         # Assertions
         assert result.exit_code != 0
         assert "❌ Certificate validation failed!" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_info_success(self, mock_cert_manager_class):
         """Test successful certificate info display."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
-        
+
         # Mock certificate info
         mock_cert_info = Mock()
         mock_cert_info.subject = {"CN": "test.example.com"}
@@ -158,96 +166,91 @@ class TestCertCLI:
         mock_cert_info.certificate_type = CertificateType.SERVER
         mock_cert_info.subject_alt_names = ["test.example.com"]
         mock_cert_manager.get_certificate_info.return_value = mock_cert_info
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'info',
-            self.test_cert_path
-        ])
-        
+        result = self.runner.invoke(cert_cli, ["info", self.test_cert_path])
+
         # Assertions
         assert result.exit_code == 0
         assert "Certificate Information:" in result.output
         assert "test.example.com" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_revoke_success(self, mock_cert_manager_class):
         """Test successful certificate revocation."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.revoke_certificate.return_value = True
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'revoke',
-            '123456789',
-            '--reason', 'compromised'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli, ["revoke", "123456789", "--reason", "compromised"]
+        )
+
         # Assertions
         assert result.exit_code == 0
         assert "✅ Certificate revoked successfully!" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_revoke_failure(self, mock_cert_manager_class):
         """Test certificate revocation failure."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.revoke_certificate.return_value = False
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'revoke',
-            '123456789',
-            '--reason', 'compromised'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli, ["revoke", "123456789", "--reason", "compromised"]
+        )
+
         # Assertions
         assert result.exit_code != 0
         assert "❌ Failed to revoke certificate!" in result.output
-    
+
     def test_help(self):
         """Test CLI help output."""
-        result = self.runner.invoke(cert_cli, ['--help'])
+        result = self.runner.invoke(cert_cli, ["--help"])
         assert result.exit_code == 0
         assert "Certificate Management CLI" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_ca_help(self, mock_cert_manager_class):
         """Test create-ca help output."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
-        
-        result = self.runner.invoke(cert_cli, ['create-ca', '--help'])
+
+        result = self.runner.invoke(cert_cli, ["create-ca", "--help"])
         assert result.exit_code == 0
         assert "Create a root CA certificate" in result.output
-    
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_missing_required_options(self, mock_cert_manager_class):
         """Test missing required options."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
-        
-        result = self.runner.invoke(cert_cli, ['create-ca'])
+
+        result = self.runner.invoke(cert_cli, ["create-ca"])
         assert result.exit_code != 0
         assert "Missing option" in result.output
 
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
-    @patch('mcp_security_framework.cli.cert_cli.CertificateConfig')
-    def test_create_intermediate_ca_success(self, mock_config_class, mock_cert_manager_class):
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
+    @patch("mcp_security_framework.cli.cert_cli.CertificateConfig")
+    def test_create_intermediate_ca_success(
+        self, mock_config_class, mock_cert_manager_class
+    ):
         """Test successful intermediate CA certificate creation."""
         # Mock configuration
         mock_config = Mock()
         mock_config_class.return_value = mock_config
-        
+
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
-        
+
         # Mock certificate pair
         mock_cert_pair = Mock(spec=CertificatePair)
         mock_cert_pair.certificate_path = "/path/to/intermediate_ca.crt"
@@ -255,107 +258,142 @@ class TestCertCLI:
         mock_cert_pair.serial_number = "123456789"
         mock_cert_pair.not_after = "2025-01-01"
         mock_cert_manager.create_intermediate_ca.return_value = mock_cert_pair
-        
+
         # Create temporary config file
         config_file = os.path.join(self.temp_dir, "test_config.json")
-        with open(config_file, 'w') as f:
+        with open(config_file, "w") as f:
             f.write('{"cert_storage_path": "./certs", "key_storage_path": "./keys"}')
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            '--config', config_file,
-            'create-intermediate-ca',
-            '--common-name', 'Test Intermediate CA',
-            '--organization', 'Test Org',
-            '--country', 'US',
-            '--parent-ca-cert', '/path/to/parent_ca.crt',
-            '--parent-ca-key', '/path/to/parent_ca.key'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "--config",
+                config_file,
+                "create-intermediate-ca",
+                "--common-name",
+                "Test Intermediate CA",
+                "--organization",
+                "Test Org",
+                "--country",
+                "US",
+                "--parent-ca-cert",
+                "/path/to/parent_ca.crt",
+                "--parent-ca-key",
+                "/path/to/parent_ca.key",
+            ],
+        )
+
         # Assertions
         assert result.exit_code == 0
         assert "✅ Intermediate CA certificate created successfully!" in result.output
 
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_intermediate_ca_failure(self, mock_cert_manager_class):
         """Test intermediate CA certificate creation failure."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.create_intermediate_ca.side_effect = Exception("Test error")
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'create-intermediate-ca',
-            '--common-name', 'Test Intermediate CA',
-            '--organization', 'Test Org',
-            '--country', 'US',
-            '--parent-ca-cert', '/path/to/parent_ca.crt',
-            '--parent-ca-key', '/path/to/parent_ca.key'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "create-intermediate-ca",
+                "--common-name",
+                "Test Intermediate CA",
+                "--organization",
+                "Test Org",
+                "--country",
+                "US",
+                "--parent-ca-cert",
+                "/path/to/parent_ca.crt",
+                "--parent-ca-key",
+                "/path/to/parent_ca.key",
+            ],
+        )
+
         # Assertions
         assert result.exit_code != 0
         assert "❌ Failed to create intermediate CA certificate" in result.output
 
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_crl_success(self, mock_cert_manager_class):
         """Test successful CRL creation."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.create_crl.return_value = "/path/to/crl.pem"
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'create-crl',
-            '--ca-cert', '/path/to/ca.crt',
-            '--ca-key', '/path/to/ca.key',
-            '--validity-days', '30'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "create-crl",
+                "--ca-cert",
+                "/path/to/ca.crt",
+                "--ca-key",
+                "/path/to/ca.key",
+                "--validity-days",
+                "30",
+            ],
+        )
+
         # Assertions
         assert result.exit_code == 0
         assert "✅ CRL created successfully!" in result.output
         assert "/path/to/crl.pem" in result.output
 
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_crl_with_output(self, mock_cert_manager_class):
         """Test CRL creation with custom output path."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.create_crl.return_value = "/custom/path/crl.pem"
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'create-crl',
-            '--ca-cert', '/path/to/ca.crt',
-            '--ca-key', '/path/to/ca.key',
-            '--output', '/custom/path/crl.pem',
-            '--validity-days', '60'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "create-crl",
+                "--ca-cert",
+                "/path/to/ca.crt",
+                "--ca-key",
+                "/path/to/ca.key",
+                "--output",
+                "/custom/path/crl.pem",
+                "--validity-days",
+                "60",
+            ],
+        )
+
         # Assertions
         assert result.exit_code == 0
         assert "✅ CRL created successfully!" in result.output
         assert "/custom/path/crl.pem" in result.output
 
-    @patch('mcp_security_framework.cli.cert_cli.CertificateManager')
+    @patch("mcp_security_framework.cli.cert_cli.CertificateManager")
     def test_create_crl_failure(self, mock_cert_manager_class):
         """Test CRL creation failure."""
         # Mock certificate manager
         mock_cert_manager = Mock()
         mock_cert_manager_class.return_value = mock_cert_manager
         mock_cert_manager.create_crl.side_effect = Exception("Test error")
-        
+
         # Run command
-        result = self.runner.invoke(cert_cli, [
-            'create-crl',
-            '--ca-cert', '/path/to/ca.crt',
-            '--ca-key', '/path/to/ca.key'
-        ])
-        
+        result = self.runner.invoke(
+            cert_cli,
+            [
+                "create-crl",
+                "--ca-cert",
+                "/path/to/ca.crt",
+                "--ca-key",
+                "/path/to/ca.key",
+            ],
+        )
+
         # Assertions
         assert result.exit_code != 0
         assert "❌ Failed to create CRL" in result.output