wehub-resource-sync
226 lines
8.6 KiB
Python
226 lines
8.6 KiB
Python
"""
|
|
tests/test_social_parser.py
|
|
pytest 测试套件 · social_parser.py
|
|
"""
|
|
|
|
import pytest
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from tools.social_parser import (
|
|
detect_platform,
|
|
platform_display,
|
|
scan_directory,
|
|
read_text_file,
|
|
parse_json_export,
|
|
scan_signals,
|
|
generate_report,
|
|
)
|
|
|
|
|
|
# ─────────────────────────────────────────────
|
|
# 纯函数
|
|
# ─────────────────────────────────────────────
|
|
|
|
class TestDetectPlatform:
|
|
def test_weibo(self) -> None:
|
|
assert detect_platform("微博_2024.txt") == "weibo"
|
|
assert detect_platform("WB_001.jpg") == "weibo"
|
|
|
|
def test_xiaohongshu(self) -> None:
|
|
assert detect_platform("小红书截图.png") == "xiaohongshu"
|
|
assert detect_platform("xhs_01.jpg") == "xiaohongshu"
|
|
assert detect_platform("rednote_post.txt") == "xiaohongshu"
|
|
|
|
def test_moments(self) -> None:
|
|
assert detect_platform("朋友圈截图.jpg") == "moments"
|
|
assert detect_platform("wechat_moments.png") == "moments"
|
|
|
|
def test_douyin(self) -> None:
|
|
assert detect_platform("抖音视频.jpg") == "douyin"
|
|
assert detect_platform("dy_001.png") == "douyin"
|
|
|
|
def test_unknown(self) -> None:
|
|
assert detect_platform("random_file.jpg") == "未知平台"
|
|
assert detect_platform("untitled.txt") == "未知平台"
|
|
|
|
|
|
class TestPlatformDisplay:
|
|
def test_known_platforms(self) -> None:
|
|
assert platform_display("weibo") == "微博"
|
|
assert platform_display("xiaohongshu") == "小红书"
|
|
assert platform_display("moments") == "微信朋友圈"
|
|
assert platform_display("douyin") == "抖音"
|
|
assert platform_display("instagram") == "Instagram"
|
|
|
|
def test_unknown(self) -> None:
|
|
assert platform_display("未知平台") == "未知来源"
|
|
assert platform_display("random") == "random"
|
|
|
|
|
|
class TestScanSignals:
|
|
def test_positive_signals(self) -> None:
|
|
result = scan_signals("今天和你在一起很开心，期待下次见面！")
|
|
assert "积极信号" in result
|
|
assert "开心" in result["积极信号"] or "期待" in result["积极信号"]
|
|
|
|
def test_low_mood_signals(self) -> None:
|
|
result = scan_signals("今天心情不好，失眠了，好难过")
|
|
assert "情绪低落" in result
|
|
assert "难过" in result["情绪低落"]
|
|
|
|
def test_love_related(self) -> None:
|
|
result = scan_signals("暗恋一个人很久了，想表白")
|
|
assert "感情相关" in result
|
|
assert "暗恋" in result["感情相关"]
|
|
|
|
def test_no_signal(self) -> None:
|
|
result = scan_signals("今天吃了火锅")
|
|
assert result == {}
|
|
|
|
def test_catches_multiple_categories(self) -> None:
|
|
result = scan_signals("想你啊，好难过")
|
|
assert len(result) >= 2 # "积极信号" or "情绪低落"
|
|
|
|
|
|
# ─────────────────────────────────────────────
|
|
# 文件操作
|
|
# ─────────────────────────────────────────────
|
|
|
|
class TestReadTextFile:
|
|
def test_read_normal_file(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "note.txt"
|
|
f.write_text("这是内容", encoding="utf-8")
|
|
assert read_text_file(str(f)) == "这是内容"
|
|
|
|
def test_truncate_long_file(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "long.txt"
|
|
f.write_text("x" * 10000, encoding="utf-8")
|
|
result = read_text_file(str(f), max_chars=100)
|
|
assert len(result) <= 100 + 50 # allow for truncation msg
|
|
assert "[... 内容过长" in result
|
|
|
|
def test_read_nonexistent(self) -> None:
|
|
result = read_text_file("/no/such/file.txt")
|
|
assert "[读取失败" in result
|
|
|
|
def test_encoding_errors_ignored(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "weird.txt"
|
|
f.write_bytes(b"\x00\xff\xfe bad content")
|
|
result = read_text_file(str(f))
|
|
assert isinstance(result, str)
|
|
|
|
|
|
class TestParseJsonExport:
|
|
def test_list_format(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "export.json"
|
|
f.write_text(json.dumps([
|
|
{"text": "微博内容1", "created_at": "2024-01-01"},
|
|
{"text": "微博内容2", "time": "2024-01-02"},
|
|
]), encoding="utf-8")
|
|
posts = parse_json_export(str(f))
|
|
assert len(posts) == 2
|
|
assert posts[0]["text"] == "微博内容1"
|
|
|
|
def test_dict_with_data_key(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "export.json"
|
|
f.write_text(json.dumps({"data": [
|
|
{"content": "小红书笔记", "timestamp": "2024-01-01"},
|
|
]}), encoding="utf-8")
|
|
posts = parse_json_export(str(f))
|
|
assert len(posts) == 1
|
|
assert posts[0]["text"] == "小红书笔记"
|
|
|
|
def test_empty_json(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "empty.json"
|
|
f.write_text("{}", encoding="utf-8")
|
|
assert parse_json_export(str(f)) == []
|
|
|
|
def test_max_50_items(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "many.json"
|
|
items = [{"text": f"post_{i}"} for i in range(100)]
|
|
f.write_text(json.dumps(items), encoding="utf-8")
|
|
posts = parse_json_export(str(f))
|
|
assert len(posts) == 50
|
|
|
|
def test_invalid_json(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "bad.json"
|
|
f.write_text("not json", encoding="utf-8")
|
|
assert parse_json_export(str(f)) == []
|
|
|
|
def test_handles_other_keys(self, tmp_path: Path) -> None:
|
|
f = tmp_path / "other.json"
|
|
f.write_text(json.dumps([
|
|
{"description": "微博描述", "date": "2024-01-01"},
|
|
{"body": "帖子正文", "timestamp": "2024-01-02"},
|
|
]), encoding="utf-8")
|
|
posts = parse_json_export(str(f))
|
|
assert len(posts) == 2
|
|
|
|
|
|
class TestScanDirectory:
|
|
def test_empty_dir(self, tmp_path: Path) -> None:
|
|
result = scan_directory(str(tmp_path))
|
|
assert result["images"] == []
|
|
assert result["texts"] == []
|
|
|
|
def test_finds_images_and_texts(self, tmp_path: Path) -> None:
|
|
(tmp_path / "a.jpg").write_text("x")
|
|
(tmp_path / "b.png").write_text("y")
|
|
(tmp_path / "note.txt").write_text("note")
|
|
(tmp_path / "readme.md").write_text("md")
|
|
(tmp_path / "data.json").write_text("{}")
|
|
result = scan_directory(str(tmp_path))
|
|
assert len(result["images"]) == 2
|
|
assert len(result["texts"]) == 3
|
|
|
|
def test_report_md_excluded(self, tmp_path: Path) -> None:
|
|
(tmp_path / "report.md").write_text("x")
|
|
result = scan_directory(str(tmp_path))
|
|
names = {p["name"] for p in result["others"]}
|
|
assert "report.md" not in names
|
|
|
|
def test_nonexistent_dir(self) -> None:
|
|
result = scan_directory("/no/such/path")
|
|
assert result["images"] == []
|
|
assert result["texts"] == []
|
|
|
|
def test_subdirectory_scan(self, tmp_path: Path) -> None:
|
|
sub = tmp_path / "sub"
|
|
sub.mkdir()
|
|
(sub / "deep.jpg").write_text("x")
|
|
result = scan_directory(str(tmp_path))
|
|
assert any(p["rel"].startswith("sub") for p in result["images"])
|
|
|
|
def test_others_category(self, tmp_path: Path) -> None:
|
|
(tmp_path / "video.mp4").write_text("x")
|
|
result = scan_directory(str(tmp_path))
|
|
assert len(result["others"]) == 1
|
|
|
|
|
|
class TestGenerateReport:
|
|
def test_empty_directory(self, tmp_path: Path) -> None:
|
|
result = generate_report(str(tmp_path), "小美")
|
|
assert "社交媒体内容报告" in result
|
|
assert "⚠️" in result
|
|
|
|
def test_returns_string(self, tmp_path: Path) -> None:
|
|
result = generate_report(str(tmp_path), "小美")
|
|
assert isinstance(result, str)
|
|
|
|
def test_with_image_files(self, tmp_path: Path) -> None:
|
|
(tmp_path / "weibo_01.jpg").write_text("x")
|
|
(tmp_path / "小红书截图.png").write_text("x")
|
|
result = generate_report(str(tmp_path), "小美")
|
|
assert "图片清单" in result
|
|
|
|
def test_with_text_files(self, tmp_path: Path) -> None:
|
|
(tmp_path / "note.txt").write_text("今天很开心，想你！", encoding="utf-8")
|
|
result = generate_report(str(tmp_path), "小美")
|
|
assert "文字内容" in result
|
|
assert "积极信号" in result
|
|
|
|
def test_output_path_writes_file(self, tmp_path: Path) -> None:
|
|
out = tmp_path / "report.md"
|
|
generate_report(str(tmp_path), "小美", str(out))
|
|
assert out.exists() |