Function main

tests/test_report_tool.py:119–160

Main test function

()

async def main():
    """Main test function"""
    print("\n" + "="*60)
    print("📊 Report Tool Test Suite")
    print("="*60)
    
    # Check environment variables
    print("\n🔍 Checking environment variables...")
    required_vars = ["OPENAI_API_KEY", "TAVILY_API_KEY"]
    missing_vars = [var for var in required_vars if not os.getenv(var)]
    
    if missing_vars:
        print(f"⚠️  Missing environment variables: {', '.join(missing_vars)}")
        print("Please set them in your .env file or environment")
        print("\nRequired variables:")
        print("  - OPENAI_API_KEY (or REPORT_ENGINE_API_KEY)")
        print("  - TAVILY_API_KEY")
        print("\nOptional variables:")
        print("  - OPENAI_BASE_URL (or REPORT_ENGINE_BASE_URL)")
        print("  - OPENAI_MODEL_NAME (or REPORT_ENGINE_MODEL_NAME)")
    else:
        print("✅ All required environment variables are set")
    
    # Run tests
    try:
        # Test 1: Direct tool invocation
        await test_direct_tool_invocation()
        
        # Test 2: Custom output path
        await test_with_custom_output_path()
        
        # Test 3: Standalone execution info
        test_standalone_execution()
        
    except Exception as e:
        print(f"\n❌ Test suite error: {e}")
        import traceback
        traceback.print_exc()
    
    print("\n" + "="*60)
    print("✅ Test suite completed!")
    print("="*60)


if __name__ == "__main__":