from eden_integrated_robustness import EdenRobustCore, PermissionLevel
import time

print("="*70)
print("  IMPROVED ROBUSTNESS TEST")
print("="*70)

eden = EdenRobustCore(config={'debug': False})

# Set proper default permissions
eden.security.security_manager.set_permission("simple_processing", PermissionLevel.LIMITED_WRITE)
eden.security.security_manager.set_permission("stress_test", PermissionLevel.LIMITED_WRITE)
eden.security.security_manager.set_permission("batch_task_0", PermissionLevel.LIMITED_WRITE)
eden.security.security_manager.set_permission("batch_task_1", PermissionLevel.LIMITED_WRITE)
eden.security.security_manager.set_permission("batch_task_2", PermissionLevel.LIMITED_WRITE)
eden.security.security_manager.set_permission("batch_task_3", PermissionLevel.LIMITED_WRITE)
eden.security.security_manager.set_permission("batch_task_4", PermissionLevel.LIMITED_WRITE)

def simple_task(data):
    time.sleep(0.01)
    return {"result": f"Processed: {data}", "confidence": 0.95}

# Run 10 successful operations
success_count = 0
for i in range(10):
    result = eden.execute_robust_operation(
        operation=simple_task,
        operation_name="simple_processing",
        input_data=f"test_{i}"
    )
    if result['success']:
        success_count += 1
        print(f"✓ Operation {i+1}/10 succeeded")

print(f"\n✅ {success_count}/10 operations successful!")

# Check health
health = eden.get_health_status()
print(f"\nFinal Status: {'🟢 HEALTHY' if health['healthy'] else '🔴 UNHEALTHY'}")
print(f"Error Rate: {health['current']['error_rate']:.2%}")
print(f"Total Operations: {health['operations']}")

eden.shutdown()