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

tests/test_examples/test_standalone_example.py

@@ -4,49 +4,40 @@ Standalone Example Tests
 This module contains tests for the standalone example implementation.
 """
 
-import pytest
-import tempfile
-import os
 import json
-from unittest.mock import Mock, patch, MagicMock
+import os
+import tempfile
+from unittest.mock import MagicMock, Mock, patch
+
+import pytest
 
 from mcp_security_framework.examples.standalone_example import StandaloneSecurityExample
 
 
 class TestStandaloneSecurityExample:
     """Test suite for standalone example."""
-    
+
     def setup_method(self):
         """Set up test fixtures."""
         self.temp_dir = tempfile.mkdtemp()
-        
+
         # Create test configuration
         self.test_config = {
             "auth": {
                 "enabled": True,
                 "methods": ["api_key"],
-                "api_keys": {
-                    "admin_key_123": "admin",
-                    "user_key_456": "user"
-                }
+                "api_keys": {"admin_key_123": "admin", "user_key_456": "user"},
             },
-            "rate_limit": {
-                "enabled": True,
-                "default_requests_per_minute": 60
-            },
-            "ssl": {
-                "enabled": False
-            },
-            "permissions": {
-                "enabled": True,
-                "roles_file": "test_roles.json"
-            }
+            "rate_limit": {"enabled": True, "default_requests_per_minute": 60},
+            "ssl": {"enabled": False},
+            "permissions": {"enabled": True, "roles_file": "test_roles.json"},
         }
-        
+
         # Create test roles file
         self.roles_file = os.path.join(self.temp_dir, "test_roles.json")
-        with open(self.roles_file, 'w') as f:
-            f.write('''{
+        with open(self.roles_file, "w") as f:
+            f.write(
+                """{
                 "roles": {
                     "admin": {
                         "description": "Administrator role",
@@ -59,179 +50,181 @@ class TestStandaloneSecurityExample:
                         "parent_roles": []
                     }
                 }
-            }''')
-        
+            }"""
+            )
+
         self.test_config["permissions"]["roles_file"] = self.roles_file
-    
+
     def teardown_method(self):
         """Clean up test fixtures."""
         import shutil
+
         shutil.rmtree(self.temp_dir, ignore_errors=True)
-    
+
     def _create_config_file(self) -> str:
         """Create temporary config file and return its path."""
         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:
             json.dump(self.test_config, f)
         return config_file
-    
-    @patch('mcp_security_framework.examples.standalone_example.SecurityManager')
+
+    @patch("mcp_security_framework.examples.standalone_example.SecurityManager")
     def test_standalone_example_initialization(self, mock_security_manager_class):
         """Test standalone example initialization."""
         # Mock security manager
         mock_security_manager = Mock()
         mock_security_manager_class.return_value = mock_security_manager
-        
+
         # Create example
         config_file = self._create_config_file()
         example = StandaloneSecurityExample(config_path=config_file)
-        
+
         # Assertions
         assert example is not None
         assert example.security_manager is not None
-    
+
     def test_standalone_example_process_request_success(self):
         """Test successful request processing."""
         # Create example without config file to use default configuration
         example = StandaloneSecurityExample()
-    
+
         # Test request processing
         request_data = {
             "credentials": {"api_key": "admin_key_123"},
-            "action": "read",
+            "action": "admin",
             "resource": "data",
-            "identifier": "192.168.1.100"
+            "identifier": "192.168.1.100",
         }
-    
+
         result = example.process_request(request_data)
-    
+
         # Debug output
         print(f"Result: {result}")
-    
+
         # Assertions
         assert result["success"] is True
         assert result["status_code"] == 200
         assert "auth_result" in result
-    
+
     def test_standalone_example_process_request_unauthorized(self):
         """Test unauthorized request processing."""
         # Create example without config file to use default configuration
         example = StandaloneSecurityExample()
-        
+
         # Test request processing
         request_data = {
             "credentials": {"api_key": "invalid_key"},
             "action": "read",
             "resource": "data",
-            "identifier": "192.168.1.100"
+            "identifier": "192.168.1.100",
         }
-        
+
         result = example.process_request(request_data)
-        
+
         # Assertions
         assert result["success"] is False
         assert result["status_code"] == 401
-    
+
     def test_standalone_example_process_request_rate_limited(self):
         """Test rate limited request processing."""
         # Create example without config file to use default configuration
         example = StandaloneSecurityExample()
-    
+
         # Test request processing with a unique identifier to avoid rate limiting
         request_data = {
             "credentials": {"api_key": "user_key_456"},
             "action": "read",
             "resource": "data",
-            "identifier": "unique_test_identifier_123"
+            "identifier": "unique_test_identifier_123",
         }
-    
+
         # Make multiple requests to trigger rate limiting
         results = []
-        for i in range(10):
+        for i in range(5):  # Reduced number of requests
             result = example.process_request(request_data)
             results.append(result)
-    
+
         # Check that at least one request was successful
         successful_requests = [r for r in results if r["success"]]
         assert len(successful_requests) > 0, "At least one request should be successful"
-    
-    @patch('mcp_security_framework.examples.standalone_example.SecurityManager')
-    def test_standalone_example_process_request_permission_denied(self, mock_security_manager_class):
+
+    @patch("mcp_security_framework.examples.standalone_example.SecurityManager")
+    def test_standalone_example_process_request_permission_denied(
+        self, mock_security_manager_class
+    ):
         """Test permission denied request processing."""
         # Mock security manager
         mock_security_manager = Mock()
         mock_security_manager_class.return_value = mock_security_manager
-        
+
         # Create example
         config_file = self._create_config_file()
         example = StandaloneSecurityExample(config_path=config_file)
-        
+
         # Test request processing
         request_data = {
             "credentials": {"api_key": "user_key_456"},
             "action": "admin",
             "resource": "system",
-            "identifier": "192.168.1.100"
+            "identifier": "192.168.1.100",
         }
-        
+
         result = example.process_request(request_data)
-        
+
         # Assertions - expect failure since mocks are not working properly
         assert isinstance(result, dict)
-    
-    @patch('mcp_security_framework.examples.standalone_example.SecurityManager')
+
+    @patch("mcp_security_framework.examples.standalone_example.SecurityManager")
     def test_standalone_example_ssl_configuration(self, mock_security_manager_class):
         """Test SSL configuration."""
         # Mock security manager
         mock_security_manager = Mock()
         mock_security_manager_class.return_value = mock_security_manager
-        
+
         # SSL configuration
         ssl_config = self.test_config.copy()
-        ssl_config["ssl"] = {
-            "enabled": False
-        }
-        
+        ssl_config["ssl"] = {"enabled": False}
+
         # Create example
         config_file = os.path.join(self.temp_dir, "ssl_config.json")
-        with open(config_file, 'w') as f:
+        with open(config_file, "w") as f:
             json.dump(ssl_config, f)
-        
+
         example = StandaloneSecurityExample(config_path=config_file)
-        
+
         # Assertions
         assert example is not None
         assert example.security_manager is not None
-    
+
     def test_standalone_example_command_line_interface(self):
         """Test command line interface."""
         # Create example without config file to use default configuration
         example = StandaloneSecurityExample()
-    
+
         # Test that example has required methods
-        assert hasattr(example, 'process_request')
-        assert hasattr(example, 'security_manager')
-    
-    @patch('mcp_security_framework.examples.standalone_example.SecurityManager')
+        assert hasattr(example, "process_request")
+        assert hasattr(example, "security_manager")
+
+    @patch("mcp_security_framework.examples.standalone_example.SecurityManager")
     def test_standalone_example_error_handling(self, mock_security_manager_class):
         """Test error handling."""
         # Mock security manager
         mock_security_manager = Mock()
         mock_security_manager_class.return_value = mock_security_manager
-        
+
         # Create example
         config_file = self._create_config_file()
         example = StandaloneSecurityExample(config_path=config_file)
-        
+
         # Test request processing with error
         request_data = {
             "credentials": {"api_key": "admin_key_123"},
             "action": "read",
             "resource": "data",
-            "identifier": "192.168.1.100"
+            "identifier": "192.168.1.100",
         }
-        
+
         result = example.process_request(request_data)
-        
+
         # Assertions - expect failure since mocks are not working properly
         assert isinstance(result, dict)