77 lines
2.4 KiB
Python
77 lines
2.4 KiB
Python
"""src.main.collect_entries_for_push patched at source (lazy import in section)"""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
from unittest.mock import patch, AsyncMock
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, str(Path(__file__).parent.parent.parent))
|
|
|
|
from src.sections.rss.section import run_rss_section
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_returns_markdown_when_entries_present(sample_config, tmp_path):
|
|
digest_raw = (
|
|
"---\n"
|
|
'title: "🌙 AI Daily 晚报 | 测试"\n'
|
|
'lead: "今日测试导读"\n'
|
|
"highlights:\n - 重点1\n"
|
|
"---\n\n"
|
|
"### 1️⃣ digest body"
|
|
)
|
|
with patch(
|
|
"src.main.collect_entries_for_push",
|
|
return_value=([{"link": "x", "title": "t", "score": 80}], []),
|
|
), patch(
|
|
"src.sections.rss.section.compose_digest",
|
|
new=AsyncMock(return_value=digest_raw),
|
|
), patch(
|
|
"src.sections.rss.section.load_recent_push_content", return_value=""
|
|
), patch(
|
|
"src.sections.rss.section.get_last_push_file", return_value=None
|
|
):
|
|
md, meta, err = await run_rss_section(sample_config, now=None)
|
|
|
|
assert md == "### 1️⃣ digest body"
|
|
assert err is None
|
|
assert meta["title"] == "🌙 AI Daily 晚报 | 测试"
|
|
assert meta["lead"] == "今日测试导读"
|
|
assert meta["highlights"] == ["重点1"]
|
|
assert meta["profile"] == "default"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_returns_empty_when_no_entries(sample_config):
|
|
with patch(
|
|
"src.main.collect_entries_for_push", return_value=([], [])
|
|
), patch(
|
|
"src.sections.rss.section.get_last_push_file", return_value=None
|
|
):
|
|
md, meta, err = await run_rss_section(sample_config, now=None)
|
|
|
|
assert md == ""
|
|
assert meta is None
|
|
assert err is None
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_returns_error_on_compose_failure(sample_config):
|
|
with patch(
|
|
"src.main.collect_entries_for_push",
|
|
return_value=([{"link": "x"}], []),
|
|
), patch(
|
|
"src.sections.rss.section.compose_digest",
|
|
new=AsyncMock(side_effect=RuntimeError("LLM down")),
|
|
), patch(
|
|
"src.sections.rss.section.load_recent_push_content", return_value=""
|
|
), patch(
|
|
"src.sections.rss.section.get_last_push_file", return_value=None
|
|
):
|
|
md, meta, err = await run_rss_section(sample_config, now=None)
|
|
|
|
assert md == ""
|
|
assert meta is None
|
|
assert "LLM down" in err
|