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

2026-07-12 20:01:02 +08:00
commit 54ca4b1b6a
267 changed files with 47047 additions and 0 deletions
@@ -0,0 +1,209 @@
+"""推送模块测试"""
+
+import os
+import pytest
+import sys
+from pathlib import Path
+from unittest.mock import patch, AsyncMock, MagicMock
+
+sys.path.insert(0, str(Path(__file__).parent.parent.parent / "src"))
+
+from push.discord import DiscordPlatform
+from push.feishu import FeishuPlatform
+from push import create_platform
+
+
+class TestDiscordPlatform:
+    """测试Discord推送"""
+
+    def test_validate_config_valid(self):
+        config = {
+            "enabled": True,
+            "apiKeyName": "DISCORD_WEBHOOK_URL",
+        }
+        with patch.dict(
+            os.environ,
+            {"DISCORD_WEBHOOK_URL": "https://discord.com/api/webhooks/123456/abcdef"},
+        ):
+            platform = DiscordPlatform(config)
+            assert platform.validate_config(config) is True
+
+    def test_validate_config_disabled(self):
+        config = {
+            "enabled": False,
+            "apiKeyName": "DISCORD_WEBHOOK_URL",
+        }
+        with patch.dict(
+            os.environ,
+            {"DISCORD_WEBHOOK_URL": "https://discord.com/api/webhooks/123456/abcdef"},
+        ):
+            platform = DiscordPlatform(config)
+            assert platform.validate_config(config) is False
+
+    def test_validate_config_missing_webhook(self):
+        config = {"enabled": True, "apiKeyName": "DISCORD_WEBHOOK_URL"}
+        with patch.dict(os.environ, {"DISCORD_WEBHOOK_URL": ""}):
+            platform = DiscordPlatform(config)
+            assert platform.validate_config(config) is False
+
+    def test_validate_config_invalid_url(self):
+        config = {"enabled": True, "apiKeyName": "DISCORD_WEBHOOK_URL"}
+        with patch.dict(os.environ, {"DISCORD_WEBHOOK_URL": "not-a-valid-url"}):
+            platform = DiscordPlatform(config)
+            assert platform.validate_config(config) is False
+
+    def test_validate_config_wrong_domain(self):
+        config = {"enabled": True, "apiKeyName": "DISCORD_WEBHOOK_URL"}
+        with patch.dict(
+            os.environ, {"DISCORD_WEBHOOK_URL": "https://example.com/webhook"}
+        ):
+            platform = DiscordPlatform(config)
+            assert platform.validate_config(config) is False
+
+    def test_split_content_short(self):
+        with patch.dict(os.environ, {"DISCORD_WEBHOOK_URL": "https://test.com"}):
+            config = {"apiKeyName": "DISCORD_WEBHOOK_URL"}
+            platform = DiscordPlatform(config)
+            short_content = "Hello"
+            chunks = platform._split_content(short_content, limit=2000)
+            assert len(chunks) == 1
+            assert chunks[0] == "Hello"
+
+    def test_split_content_long_message(self):
+        with patch.dict(os.environ, {"DISCORD_WEBHOOK_URL": "https://test.com"}):
+            config = {"apiKeyName": "DISCORD_WEBHOOK_URL"}
+            platform = DiscordPlatform(config)
+            long_content = "A\n" * 2500
+            chunks = platform._split_content(long_content, limit=2000)
+            assert len(chunks) > 1
+            assert all(len(c) <= 2000 for c in chunks)
+
+    def test_split_content_exact_boundary(self):
+        with patch.dict(os.environ, {"DISCORD_WEBHOOK_URL": "https://test.com"}):
+            config = {"apiKeyName": "DISCORD_WEBHOOK_URL"}
+            platform = DiscordPlatform(config)
+            content = "A" * 2000
+            chunks = platform._split_content(content, limit=2000)
+            assert len(chunks) == 1
+
+    def test_split_content_unicode(self):
+        with patch.dict(os.environ, {"DISCORD_WEBHOOK_URL": "https://test.com"}):
+            config = {"apiKeyName": "DISCORD_WEBHOOK_URL"}
+            platform = DiscordPlatform(config)
+            content = "你好" * 500
+            chunks = platform._split_content(content, limit=100)
+            assert len(chunks) >= 1
+
+    @pytest.mark.asyncio
+    async def test_send_success(self):
+        with patch.dict(
+            os.environ,
+            {"DISCORD_WEBHOOK_URL": "https://discord.com/api/webhooks/test/abc"},
+        ):
+            config = {"apiKeyName": "DISCORD_WEBHOOK_URL"}
+            platform = DiscordPlatform(config)
+
+            with patch.object(platform, "send", new_callable=AsyncMock) as mock_send:
+                mock_send.return_value = True
+
+                result = await mock_send("Test message")
+
+            assert result is True
+
+    @pytest.mark.asyncio
+    async def test_send_failure(self):
+        with patch.dict(
+            os.environ,
+            {"DISCORD_WEBHOOK_URL": "https://discord.com/api/webhooks/test/abc"},
+        ):
+            config = {"apiKeyName": "DISCORD_WEBHOOK_URL"}
+            platform = DiscordPlatform(config)
+
+            with patch.object(platform, "send", new_callable=AsyncMock) as mock_send:
+                mock_send.return_value = False
+
+                result = await mock_send("Test message")
+
+            assert result is False
+
+
+class TestFeishuPlatform:
+    """测试飞书推送"""
+
+    def test_validate_config_valid(self):
+        config = {
+            "enabled": True,
+            "apiKeyName": "FEISHU_WEBHOOK_URL",
+        }
+        with patch.dict(os.environ, {"FEISHU_WEBHOOK_URL": "https://open.feishu.cn/open-apis/bot/v2/hook/test"}):
+            platform = FeishuPlatform(config)
+            assert platform.validate_config(config) is True
+
+    def test_validate_config_disabled(self):
+        config = {
+            "enabled": False,
+            "apiKeyName": "FEISHU_WEBHOOK_URL",
+        }
+        with patch.dict(os.environ, {"FEISHU_WEBHOOK_URL": "https://open.feishu.cn/open-apis/bot/v2/hook/test"}):
+            platform = FeishuPlatform(config)
+            assert platform.validate_config(config) is False
+
+    def test_validate_config_missing_key(self):
+        config = {"enabled": True, "apiKeyName": "FEISHU_WEBHOOK_URL"}
+        with patch.dict(os.environ, {"FEISHU_WEBHOOK_URL": ""}):
+            platform = FeishuPlatform(config)
+            assert platform.validate_config(config) is False
+
+    def test_validate_config_any_non_empty_webhook(self):
+        config = {"enabled": True, "apiKeyName": "FEISHU_WEBHOOK_URL"}
+        with patch.dict(os.environ, {"FEISHU_WEBHOOK_URL": "https://open.feishu.cn/open-apis/bot/v2/hook/test"}):
+            platform = FeishuPlatform(config)
+            assert platform.validate_config(config) is True
+
+
+class TestPushFactory:
+    """测试平台工厂"""
+
+    def test_create_enabled_platform(self):
+        config = {
+            "enabled": True,
+            "apiKeyName": "DISCORD_WEBHOOK_URL",
+        }
+        with patch.dict(
+            os.environ,
+            {"DISCORD_WEBHOOK_URL": "https://discord.com/api/webhooks/123/abc"},
+        ):
+            platform = create_platform("discord", config)
+            assert platform is not None
+
+    def test_create_disabled_platform_returns_none(self):
+        config = {"enabled": False, "apiKeyName": "DISCORD_WEBHOOK_URL"}
+        platform = create_platform("discord", config)
+        assert platform is None
+
+    def test_create_unknown_platform_raises(self):
+        with pytest.raises(ValueError):
+            create_platform("unknown", {})
+
+    def test_create_feishu_platform(self):
+        config = {
+            "enabled": True,
+            "apiKeyName": "FEISHU_WEBHOOK_URL",
+        }
+        with patch.dict(os.environ, {"FEISHU_WEBHOOK_URL": "https://open.feishu.cn/open-apis/bot/v2/hook/test"}):
+            platform = create_platform("feishu", config)
+            assert platform is not None
+            assert isinstance(platform, FeishuPlatform)
+
+    def test_create_discord_platform(self):
+        config = {
+            "enabled": True,
+            "apiKeyName": "DISCORD_WEBHOOK_URL",
+        }
+        with patch.dict(
+            os.environ,
+            {"DISCORD_WEBHOOK_URL": "https://discord.com/api/webhooks/test/abc"},
+        ):
+            platform = create_platform("discord", config)
+            assert platform is not None
+            assert isinstance(platform, DiscordPlatform)