"""不联网的冒烟测试：验证 crypto 正确性、链接解析、文件名清洗、去重记录。"""

from __future__ import annotations

# 移到 scripts/ tests/ 后，直接运行时脚本目录成了 sys.path[0]，
# 把项目根加回来，import douyin / pipeline 才找得到。
import sys as _sys
from pathlib import Path as _Path
_sys.path.insert(0, str(_Path(__file__).resolve().parent.parent))

import tempfile
from pathlib import Path

from douyin import Registry, extract_aweme_id, safe_filename
from douyin.crypto import self_test
from douyin.urls import ParseError


def test_crypto():
    self_test()


def test_extract_id():
    cases = {
        "7123456789012345678": "7123456789012345678",
        "https://www.douyin.com/video/7123456789012345678": "7123456789012345678",
        "https://www.douyin.com/note/7123456789012345678": "7123456789012345678",
        "https://www.douyin.com/?modal_id=7123456789012345678": "7123456789012345678",
        "https://www.iesdouyin.com/share/video/7123456789012345678/": "7123456789012345678",
        "看看这个 https://www.douyin.com/video/7123456789012345678?a=1 好玩":
            "7123456789012345678",
    }
    for raw, want in cases.items():
        got = extract_aweme_id(raw)
        assert got == want, f"{raw!r} -> {got}, 期望 {want}"

    for bad in ["随便一段没有链接的文字", "https://example.com/foo"]:
        try:
            extract_aweme_id(bad)
        except ParseError:
            pass
        else:
            raise AssertionError(f"{bad!r} 本应报 ParseError")
    print("extract_aweme_id OK")


def test_safe_filename():
    assert safe_filename('a/b\\c:d*e?f"g<h>i|j') == "a_b_c_d_e_f_g_h_i_j"
    assert safe_filename("   ") == "untitled"
    assert len(safe_filename("x" * 200)) == 80
    assert safe_filename("多个   空格") == "多个 空格"
    print("safe_filename OK")


def test_registry():
    with tempfile.TemporaryDirectory() as d:
        root = Path(d)
        reg = Registry(root / ".downloaded.json")
        f = root / "v.mp4"
        f.write_bytes(b"x" * 10)
        assert "111" not in reg
        reg.add("111", f, 10, "标题")
        assert "111" in reg
        # 重新加载后仍在
        assert "111" in Registry(root / ".downloaded.json")
        # 文件被删则视为未下载
        f.unlink()
        assert "111" not in Registry(root / ".downloaded.json")
    print("Registry OK")


if __name__ == "__main__":
    test_crypto()
    test_extract_id()
    test_safe_filename()
    test_registry()
    print("\n全部冒烟测试通过")