test_advanced_features.py

#!/usr/bin/env python3
"""
Test script for advanced features
"""

import sys
from pathlib import Path

# Add current directory to path
sys.path.insert(0, str(Path(__file__).parent))

from trend_analyzer import TrendAnalyzer
from reports_generator import ReportGenerator
from performance_metrics import PerformanceMetrics
from ml_analyzer import MLAnalyzer


def test_trend_analyzer():
    """Test TrendAnalyzer"""
    print("Testing TrendAnalyzer...")

    analyzer = TrendAnalyzer("test_history.json")

    # Add sample data
    sample_findings = [
        {'severity': 'CRITICAL', 'category': 'Disk', 'message': 'Disk full'},
        {'severity': 'HIGH', 'category': 'Ceph-OSD', 'message': 'OSD down'},
        {'severity': 'MEDIUM', 'category': 'LVM', 'message': 'Thin pool usage high'}
    ]

    analyzer.add_analysis(sample_findings, metadata={'hostname': 'test-host'})

    # Get summary
    summary = analyzer.get_summary()

    print(f"  ✅ Total analyses: {summary['total_analyses']}")

    # Clean up
    import os
    if os.path.exists("test_history.json"):
        os.remove("test_history.json")

    return True


def test_reports_generator():
    """Test ReportGenerator"""
    print("Testing ReportGenerator...")

    generator = ReportGenerator()

    sample_findings = [
        {
            'severity': 'CRITICAL',
            'category': 'Disk',
            'message': 'Disk /dev/sda is full',
            'details': 'Usage: 98%'
        },
        {
            'severity': 'HIGH',
            'category': 'Ceph-OSD',
            'message': 'OSD.5 is down',
            'details': 'Check OSD status'
        }
    ]

    # Generate HTML report
    html = generator.generate_html_report(sample_findings, title="Test Report")

    print(f"  ✅ HTML report generated ({len(html)} bytes)")

    # Generate text report
    text = generator.generate_text_report(sample_findings, title="Test Report")

    print(f"  ✅ Text report generated ({len(text)} bytes)")

    return True


def test_performance_metrics():
    """Test PerformanceMetrics"""
    print("Testing PerformanceMetrics...")

    metrics = PerformanceMetrics("test_metrics.json")

    sample_findings = [
        {'severity': 'CRITICAL', 'category': 'Disk', 'message': 'Disk full'},
        {'severity': 'HIGH', 'category': 'Ceph-OSD', 'message': 'OSD down'},
        {'severity': 'MEDIUM', 'category': 'LVM', 'message': 'Thin pool usage high'}
    ]

    # Add metrics
    metrics.add_metrics(sample_findings)

    # Get health score
    health_score = metrics.calculate_health_score(sample_findings)

    print(f"  ✅ Health score calculated: {health_score}/100")

    # Get summary
    summary = metrics.get_performance_summary()

    if summary.get('status') != 'no_data':
        print(f"  ✅ Latest health score: {summary['latest_health_score']}")
        print(f"  ✅ Risk level: {summary['latest_risk_level']}")

    # Clean up
    import os
    if os.path.exists("test_metrics.json"):
        os.remove("test_metrics.json")

    return True


def test_ml_analyzer():
    """Test MLAnalyzer"""
    print("Testing MLAnalyzer...")

    analyzer = MLAnalyzer("test_ml_data.json")

    sample_findings = [
        {'severity': 'CRITICAL', 'category': 'Disk', 'message': 'Disk full'},
        {'severity': 'HIGH', 'category': 'Ceph-OSD', 'message': 'OSD down'},
        {'severity': 'MEDIUM', 'category': 'LVM', 'message': 'Thin pool usage high'}
    ]

    # Add training sample
    analyzer.add_training_sample(sample_findings)

    # Prioritize issues
    prioritized = analyzer.prioritize_issues(sample_findings)

    print(f"  ✅ Prioritized {len(prioritized)} issues")

    if prioritized:
        print(f"  ✅ Top priority: {prioritized[0].get('message', 'N/A')} "
              f"(score: {prioritized[0].get('priority_score', 0)})")

    # Get failure predictions
    predictions = analyzer.predict_failures(sample_findings)

    print(f"  ✅ Risk level: {predictions.get('risk_level', 'UNKNOWN')}")

    # Clean up
    import os
    if os.path.exists("test_ml_data.json"):
        os.remove("test_ml_data.json")

    return True


def main():
    """Run all tests"""
    print("\n" + "=" * 60)
    print("Testing Advanced Features")
    print("=" * 60 + "\n")

    tests = [
        ("Trend Analyzer", test_trend_analyzer),
        ("Reports Generator", test_reports_generator),
        ("Performance Metrics", test_performance_metrics),
        ("ML Analyzer", test_ml_analyzer)
    ]

    results = []

    for name, test_func in tests:
        try:
            success = test_func()
            results.append((name, success))
            print(f"✅ {name} test passed\n")
        except Exception as e:
            results.append((name, False))
            print(f"❌ {name} test failed: {str(e)}\n")

    # Summary
    print("=" * 60)
    print("Test Summary")
    print("=" * 60)

    passed = sum(1 for _, success in results if success)
    total = len(results)

    for name, success in results:
        status = "✅ PASS" if success else "❌ FAIL"
        print(f"{status:10} {name}")

    print("\n" + "=" * 60)
    print(f"Total: {passed}/{total} tests passed")
    print("=" * 60)

    return passed == total


if __name__ == '__main__':
    success = main()
    sys.exit(0 if success else 1)