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

tests/test_integration/test_flask_integration.py

@@ -11,25 +11,30 @@ License: MIT
 """
 
 import json
-import tempfile
 import os
-from unittest.mock import patch, MagicMock
-from typing import Dict, Any
+import tempfile
+from typing import Any, Dict
+from unittest.mock import MagicMock, patch
 
 import pytest
-from flask.testing import FlaskClient
 from cryptography import x509
 from cryptography.hazmat.primitives import hashes, serialization
 from cryptography.hazmat.primitives.asymmetric import rsa
+from flask.testing import FlaskClient
 
-from mcp_security_framework.examples.flask_example import FlaskExample
 from mcp_security_framework.core.security_manager import SecurityManager
-from mcp_security_framework.schemas.config import SecurityConfig, AuthConfig, RateLimitConfig, SSLConfig
+from mcp_security_framework.examples.flask_example import FlaskExample
+from mcp_security_framework.schemas.config import (
+    AuthConfig,
+    RateLimitConfig,
+    SecurityConfig,
+    SSLConfig,
+)
 
 
 class TestFlaskIntegration:
     """Integration tests for Flask with security framework."""
-    
+
     def setup_method(self):
         """Set up test fixtures before each test method."""
         # Create temporary configuration
@@ -40,227 +45,226 @@ class TestFlaskIntegration:
                 "api_keys": {
                     "admin_key_123": {"username": "admin", "roles": ["admin", "user"]},
                     "user_key_456": {"username": "user", "roles": ["user"]},
-                    "readonly_key_789": {"username": "readonly", "roles": ["readonly"]}
-                }
-            },
-            "rate_limit": {
-                "enabled": True,
-                "default_requests_per_minute": 100
-            },
-            "ssl": {
-                "enabled": False
-            },
-            "permissions": {
-                "enabled": True,
-                "roles_file": "test_roles.json"
-            },
-            "certificates": {
-                "enabled": False
+                    "readonly_key_789": {"username": "readonly", "roles": ["readonly"]},
+                },
             },
-            "logging": {
-                "level": "INFO",
-                "format": "standard"
-            }
+            "rate_limit": {"enabled": True, "default_requests_per_minute": 100},
+            "ssl": {"enabled": False},
+            "permissions": {"enabled": True, "roles_file": "test_roles.json"},
+            "certificates": {"enabled": False},
+            "logging": {"level": "INFO", "format": "standard"},
         }
-        
+
         # Create temporary config file
-        self.config_fd, self.config_path = tempfile.mkstemp(suffix='.json')
-        with os.fdopen(self.config_fd, 'w') as f:
+        self.config_fd, self.config_path = tempfile.mkstemp(suffix=".json")
+        with os.fdopen(self.config_fd, "w") as f:
             json.dump(self.test_config, f)
-        
+
         # Create temporary roles file
         self.roles_config = {
             "roles": {
                 "admin": {
                     "permissions": ["read", "write", "delete", "admin"],
-                    "description": "Administrator role"
+                    "description": "Administrator role",
                 },
                 "user": {
                     "permissions": ["read", "write"],
-                    "description": "Regular user role"
+                    "description": "Regular user role",
                 },
                 "readonly": {
                     "permissions": ["read"],
-                    "description": "Read-only user role"
-                }
+                    "description": "Read-only user role",
+                },
             }
         }
-        
-        self.roles_fd, self.roles_path = tempfile.mkstemp(suffix='.json')
-        with os.fdopen(self.roles_fd, 'w') as f:
+
+        self.roles_fd, self.roles_path = tempfile.mkstemp(suffix=".json")
+        with os.fdopen(self.roles_fd, "w") as f:
             json.dump(self.roles_config, f)
-        
+
         # Update config to use roles file
         self.test_config["permissions"]["roles_file"] = self.roles_path
-        
+
         # Recreate config file with updated roles path
-        with open(self.config_path, 'w') as f:
+        with open(self.config_path, "w") as f:
             json.dump(self.test_config, f)
-    
+
     def teardown_method(self):
         """Clean up after each test method."""
         # Remove temporary files
-        if hasattr(self, 'config_path') and os.path.exists(self.config_path):
+        if hasattr(self, "config_path") and os.path.exists(self.config_path):
             os.unlink(self.config_path)
-        if hasattr(self, 'roles_path') and os.path.exists(self.roles_path):
+        if hasattr(self, "roles_path") and os.path.exists(self.roles_path):
             os.unlink(self.roles_path)
-    
+
     def test_flask_full_integration(self):
         """Test complete Flask integration with security framework."""
         # Create Flask example
         example = FlaskExample(config_path=self.config_path)
-        
+
         # Test that the app is properly configured
         assert example.app is not None
-        assert hasattr(example.app, 'wsgi_app')
-        
+        assert hasattr(example.app, "wsgi_app")
+
         # Test that security manager is configured
         assert example.security_manager is not None
         assert isinstance(example.security_manager, SecurityManager)
-        
+
         # Test that configuration is loaded
         assert example.config is not None
         assert example.config.auth.enabled is True
         assert example.config.rate_limit.enabled is True
-    
+
     def test_flask_authentication_flow(self):
         """Test complete authentication flow in Flask."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test unauthenticated access to protected endpoint
         response = client.get("/api/v1/users/me")
         assert response.status_code == 401
-        
+
         # Test authenticated access with valid API key
         headers = {"X-API-Key": "admin_key_123"}
         response = client.get("/api/v1/users/me", headers=headers)
         assert response.status_code == 200  # Should be authenticated
-        
+
         # Test authenticated access with different user
         headers = {"X-API-Key": "user_key_456"}
         response = client.get("/api/v1/users/me", headers=headers)
         assert response.status_code == 200  # User should be authenticated
-    
+
     def test_flask_authorization_flow(self):
         """Test complete authorization flow in Flask."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test admin access to admin-only endpoint
         headers = {"X-API-Key": "admin_key_123"}
         response = client.get("/api/v1/admin/users", headers=headers)
         assert response.status_code == 200  # Admin should have access
-        
+
         # Test regular user access to admin-only endpoint (should be denied)
         headers = {"X-API-Key": "user_key_456"}
         response = client.get("/api/v1/admin/users", headers=headers)
         assert response.status_code == 403  # User should be denied admin access
-        
+
         # Test readonly user access to write endpoint (should be denied)
         headers = {"X-API-Key": "readonly_key_789"}
-        response = client.post("/api/v1/data",
-                             headers=headers,
-                             json={"name": "test", "value": "test_value"})
-        assert response.status_code == 403  # Readonly user should be denied write access
-    
+        response = client.post(
+            "/api/v1/data",
+            headers=headers,
+            json={"name": "test", "value": "test_value"},
+        )
+        assert (
+            response.status_code == 403
+        )  # Readonly user should be denied write access
+
     def test_flask_rate_limiting(self):
         """Test rate limiting in Flask."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         headers = {"X-API-Key": "user_key_456"}
-        
+
         # Make multiple requests to trigger rate limiting
         responses = []
         for i in range(105):  # Exceed the 100 requests per minute limit
             response = client.get("/api/v1/users/me", headers=headers)
             responses.append(response.status_code)
-        
+
         # Check that some requests were rate limited
         # Note: Rate limiting may not be triggered in test environment
         # but the requests should still be processed
         assert len(responses) == 105, "All requests should be processed"
-        assert all(status in [200, 429] for status in responses), "Responses should be either 200 or 429"
-    
+        assert all(
+            status in [200, 429] for status in responses
+        ), "Responses should be either 200 or 429"
+
     def test_flask_ssl_integration(self):
         """Test SSL/TLS integration in Flask."""
         # Create config with SSL enabled
         ssl_config = self.test_config.copy()
-        ssl_config["ssl"] = {
-            "enabled": False  # Disable SSL for testing
-        }
-        
+        ssl_config["ssl"] = {"enabled": False}  # Disable SSL for testing
+
         # Create temporary SSL config file
-        ssl_config_fd, ssl_config_path = tempfile.mkstemp(suffix='.json')
-        with os.fdopen(ssl_config_fd, 'w') as f:
+        ssl_config_fd, ssl_config_path = tempfile.mkstemp(suffix=".json")
+        with os.fdopen(ssl_config_fd, "w") as f:
             json.dump(ssl_config, f)
-        
+
         try:
             # Mock SSL context creation to avoid file requirements
-            with patch('mcp_security_framework.core.ssl_manager.SSLManager.create_server_context') as mock_ssl:
+            with patch(
+                "mcp_security_framework.core.ssl_manager.SSLManager.create_server_context"
+            ) as mock_ssl:
                 mock_ssl.return_value = MagicMock()
-                
+
                 example = FlaskExample(config_path=ssl_config_path)
-                
+
                 # Test that SSL is configured
                 assert example.config.ssl.enabled is False  # SSL disabled for testing
-                
+
         finally:
             os.unlink(ssl_config_path)
-    
+
     def test_flask_error_handling(self):
         """Test error handling in Flask integration."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test invalid API key
         headers = {"X-API-Key": "invalid_key"}
         response = client.get("/api/v1/users/me", headers=headers)
         assert response.status_code == 401
-        
+
         # Test malformed request
         headers = {"X-API-Key": "admin_key_123"}
-        response = client.post("/api/v1/data",
-                             headers=headers,
-                             json={"invalid": "data"})
-        assert response.status_code == 200  # Should succeed with valid auth (Flask returns 200 for POST)
-    
+        response = client.post(
+            "/api/v1/data", headers=headers, json={"invalid": "data"}
+        )
+        assert (
+            response.status_code == 200
+        )  # Should succeed with valid auth (Flask returns 200 for POST)
+
     def test_flask_health_and_metrics(self):
         """Test health check and metrics endpoints."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test health check
         response = client.get("/health")
         assert response.status_code == 200
         data = response.get_json()
         assert "status" in data
         assert data["status"] == "healthy"
-        
+
         # Test metrics
         response = client.get("/metrics")
         assert response.status_code == 200
         data = response.get_json()
         assert "uptime_seconds" in data
         assert "requests_total" in data
-    
+
     def test_flask_data_operations(self):
         """Test data operations with security."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         headers = {"X-API-Key": "admin_key_123"}
-        
+
         # Create data
-        create_response = client.post("/api/v1/data",
-                                    headers=headers,
-                                    json={"name": "test_item", "value": "test_value"})
+        create_response = client.post(
+            "/api/v1/data",
+            headers=headers,
+            json={"name": "test_item", "value": "test_value"},
+        )
 
-        assert create_response.status_code == 200  # Should succeed with valid auth (Flask returns 200 for POST)
+        assert (
+            create_response.status_code == 200
+        )  # Should succeed with valid auth (Flask returns 200 for POST)
         data = create_response.get_json()
         assert "id" in data
-        
+
         # Retrieve data
         data_id = data["id"]
         get_response = client.get(f"/api/v1/data/{data_id}", headers=headers)
@@ -268,133 +272,130 @@ class TestFlaskIntegration:
         retrieved_data = get_response.get_json()
         assert retrieved_data["id"] == data_id
         assert "data" in retrieved_data
-    
+
     def test_flask_middleware_integration(self):
         """Test that security middleware is properly integrated."""
         example = FlaskExample(config_path=self.config_path)
-        
+
         # Check that middleware is configured
         # Note: In test environment, middleware setup is skipped
         # but we can verify the app structure
-        assert hasattr(example.app, 'wsgi_app')
-        
+        assert hasattr(example.app, "wsgi_app")
+
         # Test that routes are properly configured
         routes = []
         for rule in example.app.url_map.iter_rules():
             routes.append(rule.rule)
-        
+
         expected_routes = [
             "/health",
-            "/metrics", 
+            "/metrics",
             "/api/v1/users/me",
             "/api/v1/admin/users",
             "/api/v1/data",
-            "/api/v1/data/<data_id>"
+            "/api/v1/data/<data_id>",
         ]
-        
+
         for route in expected_routes:
             assert route in routes, f"Route {route} not found in app routes"
-    
+
     def test_flask_configuration_validation(self):
         """Test configuration validation in Flask integration."""
         # Test with invalid configuration
-        invalid_config = {
-            "auth": {
-                "enabled": True,
-                "methods": ["invalid_method"]
-            }
-        }
-        
-        invalid_config_fd, invalid_config_path = tempfile.mkstemp(suffix='.json')
-        with os.fdopen(invalid_config_fd, 'w') as f:
+        invalid_config = {"auth": {"enabled": True, "methods": ["invalid_method"]}}
+
+        invalid_config_fd, invalid_config_path = tempfile.mkstemp(suffix=".json")
+        with os.fdopen(invalid_config_fd, "w") as f:
             json.dump(invalid_config, f)
-        
+
         try:
             # Should raise validation error
             with pytest.raises(Exception):
                 FlaskExample(config_path=invalid_config_path)
         finally:
             os.unlink(invalid_config_path)
-    
+
     def test_flask_performance_benchmark(self):
         """Test performance of Flask integration."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         headers = {"X-API-Key": "user_key_456"}
-        
+
         import time
-        
+
         # Benchmark health check endpoint
         start_time = time.time()
         for _ in range(100):
             response = client.get("/health")
             assert response.status_code == 200
         end_time = time.time()
-        
+
         avg_time = (end_time - start_time) / 100
         assert avg_time < 0.01, f"Health check too slow: {avg_time:.4f}s per request"
-        
+
         # Benchmark authenticated endpoint
         start_time = time.time()
         for _ in range(50):
             response = client.get("/api/v1/users/me", headers=headers)
             assert response.status_code == 200  # Should be authenticated
         end_time = time.time()
-        
+
         avg_time = (end_time - start_time) / 50
-        assert avg_time < 0.02, f"Authenticated endpoint too slow: {avg_time:.4f}s per request"
-    
+        assert (
+            avg_time < 0.02
+        ), f"Authenticated endpoint too slow: {avg_time:.4f}s per request"
+
     def test_flask_session_management(self):
         """Test session management in Flask."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test that sessions are properly configured
-        assert hasattr(example.app, 'config')
-        assert 'SECRET_KEY' in example.app.config
-        
+        assert hasattr(example.app, "config")
+        assert "SECRET_KEY" in example.app.config
+
         # Test session persistence across requests
         headers = {"X-API-Key": "admin_key_123"}
-        
+
         # First request
         response1 = client.get("/api/v1/users/me", headers=headers)
         assert response1.status_code == 200  # Should be authenticated
-        
+
         # Second request with same session
         response2 = client.get("/api/v1/users/me", headers=headers)
         assert response2.status_code == 200  # Should be authenticated
-        
+
         # Verify consistent user data
         data1 = response1.get_json()
         data2 = response2.get_json()
         assert data1["username"] == data2["username"]
-    
+
     def test_flask_cors_integration(self):
         """Test CORS integration in Flask."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test CORS headers are present
         response = client.get("/health")
         assert response.status_code == 200
-        
+
         # Check for CORS headers (if CORS is configured)
         # Note: This depends on CORS configuration in the Flask app
-        cors_headers = ['Access-Control-Allow-Origin', 'Access-Control-Allow-Methods']
+        cors_headers = ["Access-Control-Allow-Origin", "Access-Control-Allow-Methods"]
         # We don't assert specific CORS headers as they may not be configured in test mode
-    
+
     def test_flask_logging_integration(self):
         """Test logging integration in Flask."""
         example = FlaskExample(config_path=self.config_path)
         client = example.app.test_client()
-        
+
         # Test that logging is configured
-        assert hasattr(example.app, 'logger')
-        
+        assert hasattr(example.app, "logger")
+
         # Test that requests are logged
         response = client.get("/health")
         assert response.status_code == 200
-        
+
         # Verify that the app has proper logging configuration
         assert example.app.logger is not None