#!/usr/bin/env python3
"""rewrite-selection expectedRevision 纯函数离线测试（不连库）。"""
import sys
from pathlib import Path

PROJECT_ROOT = Path(__file__).resolve().parents[3]
SCRIPTS = PROJECT_ROOT / ".agent" / "skills" / "rewrite-selection" / "scripts"
sys.path.insert(0, str(SCRIPTS))

import click  # noqa: E402

from assert_expected_revision import (  # noqa: E402
    RevisionConflict,
    assert_expected_revision,
)


def expect_conflict(fn) -> None:
    try:
        fn()
    except RevisionConflict:
        return
    raise AssertionError("应抛 RevisionConflict")


def main() -> None:
    snap = {
        "work_id": 8,
        "chapter_order": 12,
        "block_id": 99,
        "revision": 4,
    }
    assert assert_expected_revision(snap, 4)["revision"] == 4
    expect_conflict(lambda: assert_expected_revision(snap, 3))
    empty = {**snap, "block_id": None, "revision": 0}
    assert assert_expected_revision(empty, 0)["revision"] == 0
    expect_conflict(lambda: assert_expected_revision(empty, 1))
    try:
        raise RevisionConflict("x")
    except click.ClickException:
        pass
    print("PASS：assert_expected_revision 纯函数全部通过")


if __name__ == "__main__":
    main()