feat: 军事科技每日资讯推送系统 - Docker部署 + 日志系统 + 数据目录重组

2026-07-12 20:01:02 +08:00
commit 54ca4b1b6a
267 changed files with 47047 additions and 0 deletions
@@ -0,0 +1,113 @@
+#!/usr/bin/env python3
+"""测试 push_loop 时间逻辑 - 直接调用 main.py"""
+import asyncio
+import sys
+from datetime import datetime, timedelta
+from pathlib import Path
+from unittest.mock import patch
+
+sys.path.insert(0, str(Path(__file__).parent.parent))
+
+from src.config import get_timezone, load_config
+
+# 记录调用时间
+call_times = []
+
+async def mock_run_push_job(config):
+    """模拟 push job"""
+    now = datetime.now(get_timezone(config))
+    call_times.append(now)
+    print(f"\n{'='*40}")
+    print(f"📤 Mock Push Job 被调用 | {now.strftime('%Y-%m-%d %H:%M:%S')}")
+    print(f"{'='*40}\n")
+
+async def main():
+    print("🧪 测试 push_loop 时间逻辑")
+    print("="*50)
+
+    # 加载配置
+    config = load_config()
+    tz = get_timezone(config)
+
+    # 动态设置 cron：约 30 秒后 和 60 秒后触发
+    # 标准 cron 是5字段（分 时 日 月 周），通过计算实现秒级等待
+    now = datetime.now(tz)
+
+    # 计算等待时间：
+    # 第一次：约 30 秒后（下一分钟，等待 = 60 - 当前秒数）
+    # 第二次：约 60 秒后（下两分钟，等待 = 120 - 当前秒数）
+    sec_to_wait_1 = 60 - now.second  # 到下一分钟的剩余秒数
+    sec_to_wait_2 = sec_to_wait_1 + 60  # 再加一分钟
+
+    min_1 = (now.minute + 1) % 60
+    min_2 = (now.minute + 2) % 60
+
+    cron1 = f"{min_1} {now.hour} * * *"
+    cron2 = f"{min_2} {now.hour} * * *"
+
+    config['schedule']['push_cron'] = [cron1, cron2]
+
+    print(f"\n当前时间: {now.strftime('%H:%M:%S')}")
+    print(f"测试配置:")
+    print(f"  - 第1次推送: {cron1} (约 {sec_to_wait_1}s 后)")
+    print(f"  - 第2次推送: {cron2} (约 {sec_to_wait_2}s 后)")
+    print()
+
+    # 使用 patch mock run_push_job，设置超时 3 分钟
+    from src import main as main_module
+
+    test_task = None
+    push_task = None
+
+    async def run_test():
+        with patch.object(main_module, 'run_push_job', mock_run_push_job):
+            await main_module.push_loop(config)
+
+    async def timeout_guard():
+        await asyncio.sleep(180)  # 3分钟超时
+        print("\n⏱️ 测试超时")
+        if push_task:
+            push_task.cancel()
+
+    try:
+        # 同时运行 push_loop 和超时守卫
+        push_task = asyncio.create_task(run_test())
+        timeout_task = asyncio.create_task(timeout_guard())
+
+        # 等待 push_task 完成或超时
+        while push_task and not push_task.done():
+            if len(call_times) >= 2:
+                push_task.cancel()
+                break
+            await asyncio.sleep(0.1)
+
+        timeout_task.cancel()
+
+    except asyncio.CancelledError:
+        pass
+
+    # 验证结果
+    print("\n" + "="*50)
+    print("📊 测试结果验证")
+    print("="*50)
+
+    if len(call_times) >= 2:
+        print(f"✅ 成功调用 {len(call_times)} 次")
+        for i, t in enumerate(call_times, 1):
+            print(f"  第{i}次: {t.strftime('%H:%M:%S')}")
+
+        interval = (call_times[1] - call_times[0]).total_seconds()
+        print(f"\n实际间隔: {interval:.1f} 秒")
+        if 55 <= interval <= 65:
+            print("✅ 间隔正确 (约60秒)")
+        else:
+            print(f"⚠️ 间隔异常 (期望 ~60秒)")
+    else:
+        print(f"❌ 只调用了 {len(call_times)} 次，期望 2 次")
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(main())
+    except KeyboardInterrupt:
+        print("\n\n👋 测试已取消")
+        sys.exit(130)