50 lines
2.1 KiB
Python
50 lines
2.1 KiB
Python
import json
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
import zipfile
|
|
from pathlib import Path
|
|
from xml.etree import ElementTree as ET
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
class ScriptTests(unittest.TestCase):
|
|
def test_csv_analysis_separates_maturity_dates(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
output = Path(tmp) / "analysis.json"
|
|
subprocess.run([
|
|
sys.executable, str(ROOT / "scripts/analyze_report.py"),
|
|
str(ROOT / "tests/fixtures/sample-report.csv"),
|
|
"--analysis-date", "2026-08-25", "--retention-lag-days", "1",
|
|
"--output", str(output),
|
|
], check=True, capture_output=True, text=True)
|
|
data = json.loads(output.read_text(encoding="utf-8"))
|
|
self.assertEqual(data["dates"]["analysis_date"], "2026-08-25")
|
|
self.assertEqual(data["dates"]["retention_mature_date"], "2026-08-24")
|
|
self.assertEqual(len(data["ads"]), 2)
|
|
self.assertNotIn("tier", data["ads"][0])
|
|
self.assertAlmostEqual(data["portfolio"]["analysis_day"]["spend"], 72)
|
|
|
|
def test_docx_contract_and_forbidden_headings(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
output = Path(tmp) / "brief.docx"
|
|
subprocess.run([
|
|
sys.executable, str(ROOT / "scripts/build_brief.py"),
|
|
str(ROOT / "examples/brief.example.json"), "--output", str(output),
|
|
], check=True, capture_output=True, text=True)
|
|
self.assertTrue(zipfile.is_zipfile(output))
|
|
with zipfile.ZipFile(output) as archive:
|
|
root = ET.fromstring(archive.read("word/document.xml"))
|
|
text = "".join(root.itertext())
|
|
for heading in ("1. 是否达标", "2. 问题分析", "3. 建议动作", "4. 近7天表现好的 Ads"):
|
|
self.assertIn(heading, text)
|
|
for forbidden in ("核心判断", "规则一", "规则二", "Campaign层级", "层级判断"):
|
|
self.assertNotIn(forbidden, text)
|
|
self.assertIn("决定", text)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|