feat(reconcile): safely seed canonical cost basis

This commit is contained in:
Hermes Agent
2026-07-17 09:57:08 +09:00
parent 18bd8d9f0b
commit 81a853a314
25 changed files with 1689 additions and 66 deletions
+185 -24
View File
@@ -17,7 +17,6 @@ TDD coverage:
from __future__ import annotations
import logging
import uuid
from uuid import uuid4
import pytest
@@ -69,15 +68,41 @@ class FakeCostBasisPort:
def __init__(self, values: dict[str, float | None]):
self._v = values
def get_cost_basis_with_diagnostics(self, portfolio_id: str):
value = self._v.get(portfolio_id)
if value is None:
return None, "none", "missing"
return value, "legacy", None
# Backward compatibility for non-diagnostic callers
def get_cost_basis(self, portfolio_id: str) -> float | None:
return self._v.get(portfolio_id)
value = self._v.get(portfolio_id)
return value
class RaisingCostBasisPort:
def get_cost_basis(self, portfolio_id: str) -> float | None:
def get_cost_basis_with_diagnostics(self, portfolio_id: str):
raise RuntimeError("simulated network error")
class DiagnosticCostBasisPort:
def __init__(self, result):
self._result = result
def get_cost_basis_with_diagnostics(self, portfolio_id: str):
return self._result
class LegacyOnlyCostBasisPort:
"""Compatibility seam with no diagnostic method."""
def __init__(self, value):
self._value = value
def get_cost_basis(self, portfolio_id: str):
return self._value
def _setup(slot: Slot, broker_balances: dict[str, int],
cost_basis_port=None, dry_run=False,
warn_store=None):
@@ -104,19 +129,23 @@ def _setup(slot: Slot, broker_balances: dict[str, int],
# ---------------------------------------------------------------------------
def test_sync_inactive_to_holding_on_cold_start():
"""AC-1: INACTIVE slot with broker qty → INACTIVE→HOLDING, qty set, warning cleared."""
"""AC-1: INACTIVE slot with broker qty + seeded cost-basis transitions to HOLDING."""
slot = _make_slot(SlotState.INACTIVE, qty=None)
ws = InMemoryReconcileWarningStore()
ws.set_warned(str(slot.slot_id), portfolio_id="pf-005930", case_code="B")
cb_port = FakeCostBasisPort({"pf-005930": 50000.0})
result, repo, ws_out = _setup(slot, {"005930": 10}, warn_store=ws)
result, repo, ws_out = _setup(slot, {"005930": 10}, cost_basis_port=cb_port, warn_store=ws)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed.status is SlotState.HOLDING
assert refreshed.qty == 10.0
assert refreshed.avg_price == 50000.0
assert result["warning_cleared"] is True
assert not ws_out.is_warned(str(slot.slot_id))
assert result["skipped"] is False
assert result["seeded_avg_price"] is True
assert result["cost_basis_source"] == "legacy"
assert result["before"]["status"] == "inactive"
assert result["after"]["status"] == "holding"
assert result["after"]["qty"] == 10.0
@@ -130,6 +159,7 @@ def test_sync_inactive_to_holding_sets_transition_log():
svc = SlotService(repo, INSTANCE)
ws = InMemoryReconcileWarningStore()
broker = FakeBroker({"005930": 5})
cb_port = FakeCostBasisPort({"pf-005930": 50000.0})
sync_slot_to_broker(
slot_id=str(slot.slot_id),
@@ -138,6 +168,7 @@ def test_sync_inactive_to_holding_sets_transition_log():
broker=broker,
instance_key=INSTANCE,
warn_store=ws,
cost_basis_port=cb_port,
)
transitions = repo._transitions
@@ -199,20 +230,25 @@ def test_sync_seeds_avg_price_holding_with_cost_basis():
# AC-5: graceful degrade — no cost_basis_port or port fails
# ---------------------------------------------------------------------------
def test_sync_degrades_gracefully_no_cost_basis_port():
"""AC-5: When cost_basis_port is None, avg_price is not seeded (just qty updated)."""
def test_sync_inactive_requires_cost_basis_live():
"""AC-5: INACTIVE live sync requires avg_price seed; missing cost_basis_port fails closed."""
slot = _make_slot(SlotState.INACTIVE, qty=None, pid="pf-005930")
result, repo, _ = _setup(slot, {"005930": 10}, cost_basis_port=None)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed.qty == 10.0
assert refreshed.avg_price is None # no seed
assert result.get("seeded_avg_price") is not True
assert refreshed.status is SlotState.INACTIVE
assert refreshed.qty is None
assert refreshed.avg_price is None
assert result["skipped"] is True
assert result["error"] == "cost_basis_unavailable"
assert result["dry_run"] is False
assert result["skip_reason"] == "cost_basis_unavailable"
assert result.get("seeded_avg_price") is False
def test_sync_degrades_gracefully_cost_basis_port_raises(caplog):
"""AC-5: When cost_basis_port raises, qty is updated and WARN is logged."""
def test_sync_inactive_cost_basis_port_raises_fail_closed(caplog):
"""AC-5: INACTIVE live sync fails closed if cost_basis_port diagnostics raises."""
slot = _make_slot(SlotState.INACTIVE, qty=None, pid="pf-005930")
cb_port = RaisingCostBasisPort()
@@ -220,7 +256,11 @@ def test_sync_degrades_gracefully_cost_basis_port_raises(caplog):
result, repo, _ = _setup(slot, {"005930": 10}, cost_basis_port=cb_port)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed.qty == 10.0 # qty still synced
assert refreshed.status is SlotState.INACTIVE
assert refreshed.qty is None
assert result["skipped"] is True
assert result["error"] == "cost_basis_unavailable"
assert "cost_basis_unavailable" in result.get("skip_reason", "")
assert "graceful degrade" in caplog.text or "cost_basis_port" in caplog.text
@@ -236,10 +276,132 @@ def test_sync_degrades_gracefully_cost_basis_returns_none():
assert refreshed.avg_price == 60000.0 # unchanged
@pytest.mark.parametrize("invalid_wac", (0.0, -1.0, float("nan"), float("inf")))
@pytest.mark.parametrize("dry_run", (True, False))
def test_sync_legacy_only_invalid_wac_fails_closed_without_writes(invalid_wac, dry_run):
"""QC-6: legacy-only invalid WAC never bypasses inactive fail-closed barrier."""
slot = _make_slot(SlotState.INACTIVE, qty=None, pid="pf-005930")
ws = InMemoryReconcileWarningStore()
ws.set_warned(str(slot.slot_id), portfolio_id="pf-005930", case_code="B")
result, repo, ws_out = _setup(
slot,
{"005930": 10},
cost_basis_port=LegacyOnlyCostBasisPort(invalid_wac),
dry_run=dry_run,
warn_store=ws,
)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed is not None
assert refreshed.status is SlotState.INACTIVE
assert refreshed.qty is None
assert refreshed.avg_price is None
assert ws_out.is_warned(str(slot.slot_id))
assert result["after"] == result["before"]
assert result["warning_cleared"] is False
assert result["seeded_avg_price"] is False
assert result["cost_basis_source"] == "none"
assert result["cost_basis_reason"] == "invalid"
assert result["skipped"] is True
assert result["skip_reason"] == "cost_basis_unavailable"
if dry_run:
assert result["dry_run"] is True
assert "error" not in result
else:
assert result["dry_run"] is False
assert result["error"] == "cost_basis_unavailable"
def test_sync_normalizes_diagnostic_source_aliases_and_rejects_unknown_sources():
"""REQ-123: diagnostic seam never exposes raw provider field names."""
for raw_source in ("avg_cost", "wac", "cost_basis"):
slot = _make_slot(SlotState.INACTIVE, qty=None)
result, repo, _ = _setup(
slot,
{"005930": 10},
cost_basis_port=DiagnosticCostBasisPort((50000.0, raw_source, None)),
)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed is not None
assert refreshed.status is SlotState.HOLDING
assert result["cost_basis_source"] == "legacy"
assert result["cost_basis_reason"] is None
for raw_source in ("provider_private", 7, None):
slot = _make_slot(SlotState.INACTIVE, qty=None)
result, repo, _ = _setup(
slot,
{"005930": 10},
cost_basis_port=DiagnosticCostBasisPort((50000.0, raw_source, "raw.detail")),
)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed is not None
assert refreshed.status is SlotState.INACTIVE
assert result["cost_basis_source"] == "none"
assert result["cost_basis_reason"] == "diagnostic_source_invalid"
def test_sync_preserves_only_sanitized_adapter_failure_reasons():
"""REQ-123: public result reasons are stable and do not expose parser detail."""
cases = (
(None, "canonical", "invalid", "invalid"),
(None, "legacy", "missing", "missing"),
(None, "none", "transport", "transport"),
(None, "none", "http_status_503", "http_status_503"),
(None, "none", "invalid_json", "invalid_json"),
(None, "none", "invalid_payload", "invalid_payload"),
)
for value, source, reason, expected_reason in cases:
slot = _make_slot(SlotState.INACTIVE, qty=None)
result, repo, _ = _setup(
slot,
{"005930": 10},
cost_basis_port=DiagnosticCostBasisPort((value, source, reason)),
dry_run=True,
)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed is not None
assert refreshed.status is SlotState.INACTIVE
assert result["cost_basis_source"] in {"canonical", "legacy", "none"}
assert result["cost_basis_reason"] == expected_reason
# ---------------------------------------------------------------------------
# AC-6: dry-run — no writes, shows before/after
# ---------------------------------------------------------------------------
def test_sync_dry_run_inactive_valid_canonical_seed_previews_without_writes():
"""QC-5: canonical seed dry-run previews holding state without mutations."""
slot = _make_slot(SlotState.INACTIVE, qty=None)
ws = InMemoryReconcileWarningStore()
ws.set_warned(str(slot.slot_id), portfolio_id="pf-005930", case_code="B")
cb_port = DiagnosticCostBasisPort((75000.0, "canonical", None))
result, repo, ws_out = _setup(
slot,
{"005930": 5},
cost_basis_port=cb_port,
dry_run=True,
warn_store=ws,
)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed is not None
assert refreshed.status is SlotState.INACTIVE
assert refreshed.qty is None
assert refreshed.avg_price is None
assert ws_out.is_warned(str(slot.slot_id))
assert result["dry_run"] is True
assert result["warning_cleared"] is False
assert result["after"]["status"] == "holding"
assert result["after"]["qty"] == 5.0
assert result["after"]["avg_price"] == 75000.0
assert result["seeded_avg_price"] is True
assert result["cost_basis_source"] == "canonical"
assert result["cost_basis_reason"] is None
def test_sync_dry_run_no_write():
"""AC-6: dry-run does NOT modify the slot, returns before/after preview."""
slot = _make_slot(SlotState.INACTIVE, qty=None)
@@ -254,11 +416,12 @@ def test_sync_dry_run_no_write():
assert refreshed.qty is None
# warning must NOT be cleared
assert ws_out.is_warned(str(slot.slot_id))
# result has dry_run flag and after preview
# result has dry_run flag and no-write preview
assert result["dry_run"] is True
assert result["after"]["status"] == "holding"
assert result["after"]["qty"] == 10.0
assert result["after"]["status"] == "inactive"
assert result["after"]["qty"] is None
assert result["warning_cleared"] is False
assert result["skip_reason"] == "cost_basis_unavailable"
# ---------------------------------------------------------------------------
@@ -359,20 +522,18 @@ def test_sync_missing_slot_returns_not_found():
# ---------------------------------------------------------------------------
def test_sync_unwarned_slot_still_syncs_qty():
"""sync_slot_to_broker works on slots without an active warning.
warning_cleared=True means clear_warned() was called and succeeded
(idempotent: even if there was no warning, clear is a no-op so True is fine).
The important thing is that qty was updated.
"""
slot = _make_slot(SlotState.INACTIVE, qty=None)
"""sync_slot_to_broker works on slots without an active warning,
but INACTIVE still requires avg_price for live write."""
slot = _make_slot(SlotState.INACTIVE, qty=None, pid="pf-005930")
ws = InMemoryReconcileWarningStore() # no warning set
cb_port = FakeCostBasisPort({"pf-005930": 50000.0})
result, repo, _ = _setup(slot, {"005930": 7}, warn_store=ws)
result, repo, _ = _setup(slot, {"005930": 7}, cost_basis_port=cb_port, warn_store=ws)
refreshed = repo.fetch_slot(slot.slot_id)
assert refreshed.status is SlotState.HOLDING
assert refreshed.qty == 7.0
assert refreshed.avg_price == 50000.0
# warning_cleared is True because clear_warned() was called successfully
# (idempotent no-op when there was no warning)
assert result["warning_cleared"] is True