test_advanced_features.py

"""
Advanced Features Test Suite
Tests for autonomous loop, social media, CEO reports, and orchestration.
"""

import os
import sys
import time
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent))

from agents.core.claude_agent import ClaudeAgent
from agents.core.approval_workflow import ApprovalWorkflow
from agents.core.skill_orchestrator import SkillOrchestrator
from agents.core.autonomous_loop import AutonomousLoop
from agents.skills.social_media_skills import SocialMediaGenerator
from agents.skills.ceo_report_skills import CEOReportGenerator
from agents.core.advanced_logger import get_logger, ActionType, ActionStatus
from agents.core.error_handler import get_error_handler, with_retry


def test_advanced_logging():
    """Test advanced logging system."""
    print("\n=== Testing Advanced Logging ===")

    logger = get_logger()

    # Log various actions
    logger.log_action(
        ActionType.FILE_READ,
        ActionStatus.SUCCESS,
        {'file': 'test.md', 'size': 1024},
        duration_ms=15.5
    )

    logger.log_action(
        ActionType.TASK_EXECUTED,
        ActionStatus.SUCCESS,
        {'task': 'test_task'},
        duration_ms=250.0
    )

    print("✓ Logged test actions")

    # Get recent actions
    recent = logger.get_recent_actions(limit=5)
    print(f"✓ Retrieved {len(recent)} recent actions")

    # Get error summary
    error_summary = logger.get_error_summary(hours=24)
    print(f"✓ Error summary: {error_summary['total_errors']} errors")


def test_error_handling():
    """Test error handling with retry."""
    print("\n=== Testing Error Handling ===")

    error_handler = get_error_handler()

    # Test successful execution
    def success_func():
        return "Success!"

    success, result, error = error_handler.execute_with_retry(
        success_func,
        operation_name="test_success"
    )

    print(f"✓ Success test: {success}, result: {result}")

    # Test with retry decorator
    @with_retry(max_retries=2, delay_seconds=0.5)
    def flaky_func(attempt=[0]):
        attempt[0] += 1
        if attempt[0] < 2:
            raise Exception("Temporary failure")
        return "Success after retry"

    try:
        result = flaky_func()
        print(f"✓ Retry test: {result}")
    except Exception as e:
        print(f"✗ Retry test failed: {e}")


def test_social_media_generator():
    """Test social media post generation."""
    print("\n=== Testing Social Media Generator ===")

    vault_path = Path("vault")
    claude = ClaudeAgent()
    approval = ApprovalWorkflow(vault_path)
    social_media = SocialMediaGenerator(vault_path, claude, approval)

    # Generate multi-platform posts
    posts = social_media.generate_multi_platform_posts(
        topic="New Product Launch",
        context="We're launching our revolutionary AI-powered analytics platform!",
        require_approval=True
    )

    print(f"✓ Generated posts for {len(posts)} platforms")
    for platform, content in posts.items():
        print(f"  - {platform}: {len(content)} chars")

    # Check social_posts.md
    social_posts_file = vault_path / "Done" / "social_posts.md"
    if social_posts_file.exists():
        print(f"✓ Social posts saved to {social_posts_file}")


def test_ceo_report_generator():
    """Test CEO report generation."""
    print("\n=== Testing CEO Report Generator ===")

    vault_path = Path("vault")
    claude = ClaudeAgent()
    ceo_report = CEOReportGenerator(vault_path, claude)

    # Generate weekly report
    report_path = ceo_report.generate_weekly_report()

    if report_path:
        print(f"✓ Generated CEO report: {report_path}")

        # Check CEO_Report.md exists
        latest_report = vault_path / "CEO_Report.md"
        if latest_report.exists():
            print(f"✓ Latest report saved to {latest_report}")
    else:
        print("✗ Failed to generate CEO report")


def test_skill_orchestrator():
    """Test skill orchestration."""
    print("\n=== Testing Skill Orchestrator ===")

    vault_path = Path("vault")
    claude = ClaudeAgent()
    approval = ApprovalWorkflow(vault_path)
    orchestrator = SkillOrchestrator(vault_path, claude, approval)

    # List available workflows
    workflows = orchestrator.list_workflows()
    print(f"✓ Available workflows: {len(workflows)}")
    for workflow in workflows:
        print(f"  - {workflow['name']}: {workflow['steps']} steps")

    # Test milestone announcement workflow
    result = orchestrator.orchestrate_milestone_announcement(
        milestone="10,000 Users",
        details="We've reached 10,000 active users! Thank you to our amazing community.",
        require_approval=True
    )

    print(f"✓ Milestone workflow: {result['steps_completed']}/{result['steps_completed'] + result['steps_failed']} steps completed")


def test_autonomous_loop_single_cycle():
    """Test autonomous loop (single cycle)."""
    print("\n=== Testing Autonomous Loop (Single Cycle) ===")

    vault_path = Path("vault")
    claude = ClaudeAgent()
    approval = ApprovalWorkflow(vault_path)
    orchestrator = SkillOrchestrator(vault_path, claude, approval)

    loop = AutonomousLoop(
        vault_path,
        claude,
        approval,
        orchestrator,
        loop_interval=5
    )

    # Create test business update
    updates_path = vault_path / "Business_Updates"
    updates_path.mkdir(parents=True, exist_ok=True)

    test_update = updates_path / "test_update.md"
    with open(test_update, 'w', encoding='utf-8') as f:
        f.write("""# Q2 Results Announcement

We're thrilled to announce record-breaking Q2 results!

Key highlights:
- 200% revenue growth
- Expanded to 5 new markets
- Launched 4 major features

Thank you to our incredible team and customers!

#Q2Results #Growth #Success
""")

    print(f"✓ Created test business update: {test_update}")

    # Execute one cycle manually
    print("Running single autonomous loop cycle...")
    loop._execute_cycle()

    print("✓ Autonomous loop cycle completed")

    # Get status
    status = loop.get_status()
    print(f"✓ Loop status: {status['iteration_count']} iterations")


def create_sample_data():
    """Create sample data for testing."""
    print("\n=== Creating Sample Data ===")

    vault_path = Path("vault")

    # Create sample business update
    updates_path = vault_path / "Business_Updates"
    updates_path.mkdir(parents=True, exist_ok=True)

    sample_update = updates_path / "product_launch.md"
    with open(sample_update, 'w', encoding='utf-8') as f:
        f.write("""# New AI Analytics Platform Launch

We're excited to announce the launch of our new AI-powered analytics platform!

Features:
- Real-time data processing
- Predictive analytics
- Custom dashboards
- API integrations

Available now for all customers!

#ProductLaunch #AI #Analytics
""")

    print(f"✓ Created sample business update: {sample_update}")

    # Create sample inbox file
    inbox_path = vault_path / "Inbox"
    inbox_path.mkdir(parents=True, exist_ok=True)

    sample_task = inbox_path / "urgent_task.md"
    with open(sample_task, 'w', encoding='utf-8') as f:
        f.write("""# Urgent: Client Presentation

Prepare presentation for major client meeting next week.

Requirements:
- Product demo
- Case studies
- Pricing options
- Q&A preparation

Deadline: End of week

#urgent #client #presentation
""")

    print(f"✓ Created sample inbox file: {sample_task}")


def run_all_tests():
    """Run all advanced feature tests."""
    print("=" * 60)
    print("AI EMPLOYEE SYSTEM - ADVANCED FEATURES TEST SUITE")
    print("=" * 60)

    try:
        test_advanced_logging()
        test_error_handling()
        test_social_media_generator()
        test_ceo_report_generator()
        test_skill_orchestrator()
        create_sample_data()
        test_autonomous_loop_single_cycle()

        print("\n" + "=" * 60)
        print("✓ All advanced feature tests completed!")
        print("=" * 60)

        print("\nWhat's been tested:")
        print("✓ Advanced logging with structured data")
        print("✓ Error handling with automatic retry")
        print("✓ Social media post generation (Facebook, Instagram, Twitter)")
        print("✓ CEO report generation")
        print("✓ Skill orchestration workflows")
        print("✓ Autonomous loop decision making")

        print("\nNext steps:")
        print("1. Start system with autonomous loop:")
        print("   python main.py --autonomous")
        print("2. Check vault/Done/social_posts.md for generated posts")
        print("3. Check vault/CEO_Report.md for executive summary")
        print("4. Monitor logs/system.log for autonomous actions")
        print("5. Check vault/Pending_Approval/ for approval requests")

    except Exception as e:
        print(f"\n✗ Test failed: {e}")
        import traceback
        traceback.print_exc()


if __name__ == "__main__":
    run_all_tests()