84 lines
3.8 KiB
Python
84 lines
3.8 KiB
Python
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
|
|
|
|
def test_single_category_rule_preview_and_apply_only_uncategorized(tmp_path):
|
|
db_path = str(tmp_path / "ledger.sqlite3")
|
|
seeded = service.seed_default_category_rules(db_path=db_path)
|
|
assert seeded["success"] is True
|
|
rules = service.list_category_rules(db_path=db_path)["rules"]
|
|
mart_rule = next(r for r in rules if r["pattern"] == "이마트")
|
|
|
|
uncategorized = service.create_ledger_transaction(
|
|
"2026-06-01", "expense", 10_000, merchant="이마트에브리데이", raw_summary="체크카드", db_path=db_path
|
|
)["transaction"]
|
|
categorized = service.create_ledger_transaction(
|
|
"2026-06-02", "expense", 20_000, category="생활", subcategory="생필품", merchant="이마트24", raw_summary="체크카드", db_path=db_path
|
|
)["transaction"]
|
|
service.create_ledger_transaction(
|
|
"2026-06-03", "expense", 30_000, merchant="알수없음", raw_summary="체크카드", db_path=db_path
|
|
)
|
|
|
|
preview = service.preview_category_rule_matches(mart_rule["id"], db_path=db_path)
|
|
assert preview["success"] is True
|
|
assert preview["matched_count"] == 2
|
|
assert preview["uncategorized_matched_count"] == 1
|
|
assert len(preview["sample"]) == 2
|
|
|
|
applied = service.apply_category_rule(mart_rule["id"], db_path=db_path)
|
|
assert applied["success"] is True
|
|
assert applied["applied_count"] == 1
|
|
|
|
all_tx = service.list_ledger_transactions(limit=10, db_path=db_path)["transactions"]
|
|
by_id = {tx["id"]: tx for tx in all_tx}
|
|
assert by_id[uncategorized["id"]]["category"] == "식비"
|
|
assert by_id[categorized["id"]]["category"] == "생활"
|