feat(PORT-003): add historical ltv charts
This commit is contained in:
171
tests/test_ltv_history.py
Normal file
171
tests/test_ltv_history.py
Normal file
@@ -0,0 +1,171 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import csv
|
||||
import json
|
||||
from decimal import Decimal
|
||||
from io import StringIO
|
||||
from uuid import uuid4
|
||||
|
||||
from app.models.ltv_history import LtvHistoryRepository
|
||||
from app.services.ltv_history import LtvHistoryService
|
||||
|
||||
|
||||
def test_ltv_history_repository_persists_structured_workspace_snapshots(tmp_path) -> None:
|
||||
workspace_id = str(uuid4())
|
||||
service = LtvHistoryService(repository=LtvHistoryRepository(base_path=tmp_path / "workspaces"))
|
||||
|
||||
service.record_workspace_snapshot(
|
||||
workspace_id,
|
||||
{
|
||||
"ltv_ratio": Decimal("0.74"),
|
||||
"margin_call_ltv": Decimal("0.80"),
|
||||
"loan_amount": Decimal("222000"),
|
||||
"gold_value": Decimal("300000"),
|
||||
"spot_price": Decimal("4041.9"),
|
||||
"quote_source": "yfinance",
|
||||
"quote_updated_at": "2026-03-20T00:00:00+00:00",
|
||||
},
|
||||
)
|
||||
|
||||
payload = json.loads((tmp_path / "workspaces" / workspace_id / "ltv_history.json").read_text())
|
||||
|
||||
assert payload[0]["ltv_ratio"] == {"value": "0.74", "unit": "ratio"}
|
||||
assert payload[0]["margin_threshold"] == {"value": "0.80", "unit": "ratio"}
|
||||
assert payload[0]["spot_price"] == {"value": "4041.9", "currency": "USD", "per_weight_unit": "ozt"}
|
||||
assert payload[0]["loan_amount"] == {"value": "222000", "currency": "USD"}
|
||||
assert payload[0]["collateral_value"] == {"value": "300000", "currency": "USD"}
|
||||
|
||||
|
||||
def test_ltv_history_service_replaces_same_day_snapshot_and_builds_range_models(tmp_path) -> None:
|
||||
workspace_id = str(uuid4())
|
||||
service = LtvHistoryService(repository=LtvHistoryRepository(base_path=tmp_path / "workspaces"))
|
||||
|
||||
service.record_workspace_snapshot(
|
||||
workspace_id,
|
||||
{
|
||||
"ltv_ratio": Decimal("0.70"),
|
||||
"margin_call_ltv": Decimal("0.80"),
|
||||
"loan_amount": Decimal("210000"),
|
||||
"gold_value": Decimal("300000"),
|
||||
"spot_price": Decimal("4100"),
|
||||
"quote_source": "seed",
|
||||
"quote_updated_at": "2026-01-01T00:00:00+00:00",
|
||||
},
|
||||
)
|
||||
service.record_workspace_snapshot(
|
||||
workspace_id,
|
||||
{
|
||||
"ltv_ratio": Decimal("0.75"),
|
||||
"margin_call_ltv": Decimal("0.80"),
|
||||
"loan_amount": Decimal("225000"),
|
||||
"gold_value": Decimal("300000"),
|
||||
"spot_price": Decimal("4000"),
|
||||
"quote_source": "seed",
|
||||
"quote_updated_at": "2026-03-15T00:00:00+00:00",
|
||||
},
|
||||
)
|
||||
service.record_workspace_snapshot(
|
||||
workspace_id,
|
||||
{
|
||||
"ltv_ratio": Decimal("0.76"),
|
||||
"margin_call_ltv": Decimal("0.80"),
|
||||
"loan_amount": Decimal("228000"),
|
||||
"gold_value": Decimal("300000"),
|
||||
"spot_price": Decimal("3990"),
|
||||
"quote_source": "seed",
|
||||
"quote_updated_at": "2026-03-15T12:00:00+00:00",
|
||||
},
|
||||
)
|
||||
snapshots = service.record_workspace_snapshot(
|
||||
workspace_id,
|
||||
{
|
||||
"ltv_ratio": Decimal("0.78"),
|
||||
"margin_call_ltv": Decimal("0.80"),
|
||||
"loan_amount": Decimal("234000"),
|
||||
"gold_value": Decimal("300000"),
|
||||
"spot_price": Decimal("3950"),
|
||||
"quote_source": "seed",
|
||||
"quote_updated_at": "2026-03-20T00:00:00+00:00",
|
||||
},
|
||||
)
|
||||
|
||||
assert [snapshot.snapshot_date for snapshot in snapshots] == ["2026-01-01", "2026-03-15", "2026-03-20"]
|
||||
assert str(snapshots[1].ltv_ratio) == "0.76"
|
||||
|
||||
chart_7 = service.chart_model(snapshots, days=7, current_margin_threshold=Decimal("0.80"))
|
||||
chart_30 = service.chart_model(snapshots, days=30, current_margin_threshold=Decimal("0.80"))
|
||||
chart_90 = service.chart_model(snapshots, days=90, current_margin_threshold=Decimal("0.80"))
|
||||
|
||||
assert chart_7.title == "7 Day"
|
||||
assert chart_7.labels == ("2026-03-15", "2026-03-20")
|
||||
assert chart_7.ltv_values == (76.0, 78.0)
|
||||
assert chart_7.threshold_values == (80.0, 80.0)
|
||||
assert chart_30.labels == ("2026-03-15", "2026-03-20")
|
||||
assert chart_30.threshold_values == (80.0, 80.0)
|
||||
assert chart_90.labels == ("2026-01-01", "2026-03-15", "2026-03-20")
|
||||
|
||||
|
||||
def test_ltv_history_repository_rejects_invalid_numeric_and_date_payloads(tmp_path) -> None:
|
||||
workspace_id = str(uuid4())
|
||||
repo = LtvHistoryRepository(base_path=tmp_path / "workspaces")
|
||||
history_path = tmp_path / "workspaces" / workspace_id / "ltv_history.json"
|
||||
history_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
history_path.write_text(
|
||||
json.dumps(
|
||||
[
|
||||
{
|
||||
"snapshot_date": "not-a-date",
|
||||
"captured_at": "2026-03-20T00:00:00+00:00",
|
||||
"ltv_ratio": {"value": "bad", "unit": "ratio"},
|
||||
"margin_threshold": {"value": "0.80", "unit": "ratio"},
|
||||
"loan_amount": {"value": "234000", "currency": "USD"},
|
||||
"collateral_value": {"value": "300000", "currency": "USD"},
|
||||
"spot_price": {"value": "3950", "currency": "USD", "per_weight_unit": "ozt"},
|
||||
"source": "seed",
|
||||
}
|
||||
]
|
||||
)
|
||||
)
|
||||
|
||||
from app.models.ltv_history import LtvHistoryLoadError
|
||||
|
||||
try:
|
||||
repo.load(workspace_id)
|
||||
except LtvHistoryLoadError as exc:
|
||||
assert "invalid" in str(exc)
|
||||
else:
|
||||
raise AssertionError("Expected invalid LTV history payload to raise LtvHistoryLoadError")
|
||||
|
||||
|
||||
def test_ltv_history_service_exports_csv(tmp_path) -> None:
|
||||
workspace_id = str(uuid4())
|
||||
service = LtvHistoryService(repository=LtvHistoryRepository(base_path=tmp_path / "workspaces"))
|
||||
|
||||
snapshots = service.record_workspace_snapshot(
|
||||
workspace_id,
|
||||
{
|
||||
"ltv_ratio": Decimal("0.78"),
|
||||
"margin_call_ltv": Decimal("0.80"),
|
||||
"loan_amount": Decimal("234000"),
|
||||
"gold_value": Decimal("300000"),
|
||||
"spot_price": Decimal("3950"),
|
||||
"quote_source": "seed",
|
||||
"quote_updated_at": "2026-03-20T00:00:00+00:00",
|
||||
},
|
||||
)
|
||||
|
||||
csv_content = service.export_csv(snapshots)
|
||||
rows = list(csv.DictReader(StringIO(csv_content)))
|
||||
|
||||
assert rows == [
|
||||
{
|
||||
"snapshot_date": "2026-03-20",
|
||||
"captured_at": "2026-03-20T00:00:00+00:00",
|
||||
"ltv_ratio_pct": "78.0",
|
||||
"margin_threshold_pct": "80.0",
|
||||
"loan_amount_usd": "234000",
|
||||
"collateral_value_usd": "300000",
|
||||
"spot_price_usd_per_ozt": "3950",
|
||||
"source": "seed",
|
||||
}
|
||||
]
|
||||
57
tests/test_overview_ltv_history_playwright.py
Normal file
57
tests/test_overview_ltv_history_playwright.py
Normal file
@@ -0,0 +1,57 @@
|
||||
from __future__ import annotations
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
from playwright.sync_api import expect, sync_playwright
|
||||
|
||||
BASE_URL = "http://127.0.0.1:8000"
|
||||
ARTIFACTS = Path("tests/artifacts")
|
||||
ARTIFACTS.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
|
||||
def test_overview_shows_ltv_history_and_exports_csv() -> None:
|
||||
with sync_playwright() as p:
|
||||
browser = p.chromium.launch(headless=True)
|
||||
page = browser.new_page(viewport={"width": 1440, "height": 1000})
|
||||
|
||||
page.goto(BASE_URL, wait_until="domcontentloaded", timeout=30000)
|
||||
expect(page.locator("text=Create a private workspace URL").first).to_be_visible(timeout=15000)
|
||||
page.get_by_role("button", name="Get started").click()
|
||||
page.wait_for_url(f"{BASE_URL}/*", timeout=15000)
|
||||
|
||||
expect(page.locator("text=Overview").first).to_be_visible(timeout=15000)
|
||||
expect(page.locator("text=Historical LTV").first).to_be_visible(timeout=15000)
|
||||
expect(page.locator("text=7 Day").first).to_be_visible(timeout=15000)
|
||||
expect(page.locator("text=30 Day").first).to_be_visible(timeout=15000)
|
||||
expect(page.locator("text=90 Day").first).to_be_visible(timeout=15000)
|
||||
expect(page.get_by_role("button", name="Export CSV")).to_be_visible(timeout=15000)
|
||||
|
||||
series_names = page.evaluate("""
|
||||
async () => {
|
||||
const importMap = JSON.parse(document.querySelector('script[type="importmap"]').textContent).imports;
|
||||
const mod = await import(importMap['nicegui-echart']);
|
||||
const chart = mod.echarts.getInstanceByDom(document.querySelector('.nicegui-echart'));
|
||||
return chart ? chart.getOption().series.map(series => series.name) : [];
|
||||
}
|
||||
""")
|
||||
assert series_names == ["LTV", "Margin threshold"]
|
||||
|
||||
with page.expect_download() as download_info:
|
||||
page.get_by_role("button", name="Export CSV").click()
|
||||
download = download_info.value
|
||||
assert download.suggested_filename.endswith("-ltv-history.csv")
|
||||
download_path = ARTIFACTS / "ltv-history-export.csv"
|
||||
download.save_as(str(download_path))
|
||||
csv_content = download_path.read_text()
|
||||
assert (
|
||||
"snapshot_date,captured_at,ltv_ratio_pct,margin_threshold_pct,loan_amount_usd,collateral_value_usd,spot_price_usd_per_ozt,source"
|
||||
in csv_content
|
||||
)
|
||||
|
||||
body = page.locator("body").inner_text(timeout=15000)
|
||||
assert "RuntimeError" not in body
|
||||
assert "Server error" not in body
|
||||
assert "Traceback" not in body
|
||||
page.screenshot(path=str(ARTIFACTS / "overview-ltv-history.png"), full_page=True)
|
||||
|
||||
browser.close()
|
||||
Reference in New Issue
Block a user