103 lines
6.8 KiB
Python
103 lines
6.8 KiB
Python
import sqlite3
|
|
|
|
from jarca_ledger_mcp import service
|
|
from jarca_ledger_mcp.db import init_db
|
|
|
|
|
|
def _account(db_path, currency="KRW"):
|
|
result = service.create_ledger_account("주계좌", currency=currency, db_path=db_path)
|
|
assert result["success"]
|
|
return result["account"]["id"]
|
|
|
|
|
|
def test_account_valuation_schema_is_v12_and_idempotent(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
init_db(path)
|
|
init_db(path)
|
|
with sqlite3.connect(path) as conn:
|
|
cols = {row[1] for row in conn.execute("PRAGMA table_info(account_valuations)")}
|
|
version = conn.execute("SELECT value FROM schema_meta WHERE key='schema_version'").fetchone()[0]
|
|
assert version == "12"
|
|
assert {"id", "account_id", "basis_date", "amount", "currency", "valuation_kind", "source_type", "source_ref", "verification_status", "recorded_at", "memo", "idempotency_key", "supersedes_id", "correction_reason", "as_of_at", "timezone", "cost_basis", "net_contributed", "reported_principal_kind", "reported_principal_label", "profit", "return_rate", "return_basis", "principal", "interest", "fee", "representative_liability_definition"} <= cols
|
|
|
|
|
|
def test_register_same_idempotency_payload_reuses_row_and_changed_payload_conflicts(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
account_id = _account(path)
|
|
first = service.register_account_valuation(account_id, "2026-06-30", 1000, idempotency_key="sync-1", verification_status="verified", source_type="import", source_ref="stmt-1", db_path=path)
|
|
same = service.register_account_valuation(account_id, "2026-06-30", 1000, idempotency_key="sync-1", verification_status="verified", source_type="import", source_ref="stmt-1", db_path=path)
|
|
conflict = service.register_account_valuation(account_id, "2026-06-30", 1200, idempotency_key="sync-1", verification_status="verified", source_type="import", source_ref="stmt-1", db_path=path)
|
|
assert first["success"] and same["success"]
|
|
assert same["reused"] is True and same["valuation"]["id"] == first["valuation"]["id"]
|
|
assert conflict["success"] is False and conflict["error_code"] == "idempotency_conflict"
|
|
|
|
|
|
def test_select_exact_and_as_of_excludes_unverified_and_reports_carry_forward(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
account_id = _account(path)
|
|
service.register_account_valuation(account_id, "2026-06-01", 1000, verification_status="verified", idempotency_key="v1", db_path=path)
|
|
service.register_account_valuation(account_id, "2026-06-15", 2000, verification_status="unverified", idempotency_key="v2", db_path=path)
|
|
exact = service.select_account_valuation(account_id, "2026-06-15", db_path=path)
|
|
as_of = service.select_account_valuation(account_id, "2026-06-20", as_of=True, db_path=path)
|
|
assert exact["success"] and exact["valuation"] is None
|
|
assert exact["warning"] == "unverified_candidate"
|
|
assert as_of["success"] and as_of["valuation"]["amount"] == 1000
|
|
assert as_of["carried_forward"] is True and as_of["valuation_age_days"] == 19
|
|
|
|
|
|
def test_correction_is_append_only_reasoned_and_single_child(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
account_id = _account(path)
|
|
original = service.register_account_valuation(account_id, "2026-06-30", 1000, verification_status="verified", idempotency_key="v1", db_path=path)
|
|
corrected = service.correct_account_valuation(original["valuation"]["id"], 1200, correction_reason="은행 원장 정정", idempotency_key="v2", db_path=path)
|
|
duplicate = service.correct_account_valuation(original["valuation"]["id"], 1300, correction_reason="두 번째 분기", idempotency_key="v3", db_path=path)
|
|
listed = service.list_account_valuations(account_id, db_path=path)
|
|
assert corrected["success"] and corrected["valuation"]["supersedes_id"] == original["valuation"]["id"]
|
|
assert duplicate["success"] is False and duplicate["error_code"] == "correction_conflict"
|
|
assert listed["count"] == 2
|
|
selected = service.select_account_valuation(account_id, "2026-06-30", db_path=path)
|
|
assert selected["valuation"]["amount"] == 1200
|
|
|
|
|
|
def test_account_balance_write_records_unverified_candidate_only_with_real_basis_date(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
created = service.create_ledger_account("동기화 계좌", last_balance=500, last_synced_at="2026-06-30", db_path=path, last_balance_basis_date="2026-06-30")
|
|
service.create_ledger_account("날짜 없는 계좌", last_balance=700, db_path=path)
|
|
values = service.list_account_valuations(created["account"]["id"], db_path=path)
|
|
assert values["count"] == 1
|
|
assert values["valuations"][0]["verification_status"] == "unverified"
|
|
assert values["valuations"][0]["basis_date"] == "2026-06-30"
|
|
|
|
|
|
def test_idempotency_scope_includes_basis_and_correction_metadata(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
account_id = _account(path)
|
|
first = service.register_account_valuation(account_id, "2026-06-30", 1000, idempotency_key="same", db_path=path)
|
|
changed_basis = service.register_account_valuation(account_id, "2026-07-01", 1000, idempotency_key="same", db_path=path)
|
|
changed_reason = service.register_account_valuation(account_id, "2026-06-30", 1000, idempotency_key="same", supersedes_id="other", correction_reason="정정", db_path=path)
|
|
assert first["success"]
|
|
assert changed_basis["error_code"] == "idempotency_conflict"
|
|
assert changed_reason["error_code"] == "idempotency_conflict"
|
|
|
|
|
|
def test_correction_preserves_optional_fields_unless_overridden(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
account_id = _account(path, currency="USD")
|
|
original = service.register_account_valuation(account_id, "2026-06-30", 1000, currency="USD", cost_basis=900, net_contributed=950, profit=50, return_rate=0.055, principal=800, interest=20, fee=5, as_of_at="2026-06-30T12:00:00+09:00", timezone="Asia/Seoul", db_path=path)
|
|
corrected = service.correct_account_valuation(original["valuation"]["id"], 1100, correction_reason="정정", idempotency_key="corr", profit=200, db_path=path)
|
|
assert corrected["success"]
|
|
value = corrected["valuation"]
|
|
assert value["cost_basis"] == 900 and value["net_contributed"] == 950
|
|
assert value["profit"] == 200 and value["return_rate"] == 0.055
|
|
assert value["principal"] == 800 and value["interest"] == 20 and value["fee"] == 5
|
|
assert value["as_of_at"] == "2026-06-30T12:00:00+09:00" and value["timezone"] == "Asia/Seoul"
|
|
|
|
|
|
def test_account_name_only_update_does_not_create_candidate(tmp_path):
|
|
path = str(tmp_path / "ledger.sqlite3")
|
|
created = service.create_ledger_account("계좌", last_balance=500, last_synced_at="2026-06-30", db_path=path, last_balance_basis_date="2026-06-30")
|
|
account_id = created["account"]["id"]
|
|
updated = service.update_ledger_account(account_id, name="새 이름", db_path=path)
|
|
values = service.list_account_valuations(account_id, db_path=path)
|
|
assert updated["success"] and values["count"] == 1
|