from __future__ import annotations

from jarca_ledger_mcp import service


def test_category_rules_preview_apply(tmp_path):
    db_path = str(tmp_path / "ledger.sqlite3")
    service.create_ledger_transaction("2026-06-01", "expense", 10_000, merchant="이마트에브리데이", raw_summary="체크카드", db_path=db_path)
    service.create_ledger_transaction("2026-06-02", "expense", 5_000, merchant="알수없음", raw_summary="체크카드", db_path=db_path)
    seeded = service.seed_default_category_rules(db_path=db_path)
    assert seeded["success"] is True
    rules = service.list_category_rules(db_path=db_path)
    assert rules["count"] > 0

    preview = service.preview_categorization(db_path=db_path)
    assert preview["matched_count"] == 1
    assert preview["unmatched_count"] == 1
    assert preview["matches"][0]["category"] == "식비"
    assert preview["matches"][0]["subcategory"] == "마트"

    applied = service.apply_categorization_rules(db_path=db_path)
    assert applied["success"] is True
    assert applied["applied_count"] == 1
    categorized = service.list_ledger_transactions(category="식비", db_path=db_path)
    assert categorized["count"] == 1
    uncategorized = service.list_uncategorized_transactions(db_path=db_path)
    assert uncategorized["count"] == 1


def test_apply_categorization_rules_applies_all_matches_not_just_preview_sample(tmp_path):
    db_path = str(tmp_path / "ledger.sqlite3")
    service.seed_default_category_rules(db_path=db_path)
    for idx in range(150):
        service.create_ledger_transaction(
            transaction_date=f"2026-06-{(idx % 28) + 1:02d}",
            type="expense",
            amount=10_000 + idx,
            merchant=f"이마트에브리데이 {idx}",
            raw_summary="체크카드",
            db_path=db_path,
        )

    preview = service.preview_categorization(limit=5000, db_path=db_path)
    assert preview["matched_count"] == 150
    assert len(preview["matches"]) == 100

    applied = service.apply_categorization_rules(limit=5000, db_path=db_path)
    assert applied["success"] is True
    assert applied["applied_count"] == 150
    uncategorized = service.list_uncategorized_transactions(limit=200, db_path=db_path)
    assert uncategorized["count"] == 0