buff/backend/test_background_tasks.py

#!/usr/bin/env python3
"""
测试后台任务功能
"""
import asyncio
import sys
import os

# 添加项目路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from app.background_tasks import BackgroundTaskManager


async def test_background_tasks():
    """测试后台任务管理器"""
    print("🔄 开始测试后台任务管理器...")

    manager = BackgroundTaskManager()

    # 测试启动任务
    print("🚀 启动后台任务...")
    await manager.start_all_tasks()

    # 等待几秒让任务运行
    print("⏳ 等待任务运行 3 秒...")
    await asyncio.sleep(3)

    # 检查任务状态
    print(f"📊 当前任务数量: {len(manager.tasks)}")
    print(f"📊 运行状态: {manager.running}")

    for i, task in enumerate(manager.tasks):
        status = "✅ 运行中" if not task.done() else "❌ 已完成"
        print(f"   任务 {i+1}: {status}")

    # 测试停止任务
    print("🛑 停止后台任务...")
    await manager.stop_all_tasks()

    print("✅ 后台任务测试完成！")


if __name__ == "__main__":
    asyncio.run(test_background_tasks())