AI-Driven Remediation Testing

"""Quick test script for MCP Server.""" import asyncio import logging from pathlib import Path from mcp_server.config import get_settings from mcp_server.logging_config import setup_logging from mcp_server.server import MCPServer logging.basicConfig(level=logging.INFO) logger = logging.getLogger(__name__) async def test_scenario(): """Test running a scenario.""" logger.info("Starting MCP Server test") # Load settings settings = get_settings() setup_logging(settings.log_dir) # Create server server = MCPServer(settings) # Load example scenario scenario_file = Path("scenarios/example_scenario.yaml") if not scenario_file.exists(): logger.error(f"Scenario file not found: {scenario_file}") return scenario_yaml = scenario_file.read_text() logger.info("Running test scenario") # Run scenario result = await server.scenario_service.run_scenario( scenario_yaml=scenario_yaml, bindings={"namespace": "test-namespace", "service": "test-service"}, ) logger.info(f"Run ID: {result['run_id']}") logger.info(f"Status: {result['status']}") logger.info(f"Report URI: {result['report_uri']}") # Wait for scenario to complete logger.info("Waiting for scenario to complete (10 seconds)...") await asyncio.sleep(10) # Check results run_id = result["run_id"] report_file = Path(settings.log_dir) / "runs" / run_id / "report.json" if report_file.exists(): logger.info(f"\nReport generated at: {report_file}") logger.info(f"Report contents:\n{report_file.read_text()}") else: logger.warning(f"Report not found: {report_file}") # Cleanup await server.remediation_client.close() logger.info("Test completed") if __name__ == "__main__": asyncio.run(test_scenario())