rdmsandClaude Opus 5.5
1cded22e90
Co-Authored-By: Claude Opus 5.5 (1M context) <noreply@anthropic.com>
89 lines
3.7 KiB
Python
89 lines
3.7 KiB
Python
from datetime import date
|
|
from decimal import Decimal
|
|
|
|
from app.engine.breakdown import Filters
|
|
from app.models.tables import (Category, Loan, PlannedItem, RecurringItem,
|
|
Scenario, ScenarioLoan, ScenarioModifier,
|
|
ScenarioPlannedItem)
|
|
from app.services.breakdown import PLAN_NAME, compute_breakdown
|
|
|
|
M = date(2030, 3, 1)
|
|
D = Decimal
|
|
|
|
|
|
def _basis(db):
|
|
wohnen = Category(name="Wohnen")
|
|
db.add(wohnen)
|
|
db.flush()
|
|
miete = RecurringItem(name="Miete", amount=D("-1000.00"), rhythm="monthly",
|
|
due_day=1, category_id=wohnen.id)
|
|
gehalt = RecurringItem(name="Gehalt", amount=D("3000.00"), rhythm="monthly", due_day=1)
|
|
reise = PlannedItem(name="Reise", amount=D("-600.00"), due=date(2030, 3, 20))
|
|
k1 = Loan(name="K1", principal=D("1200.00"), annual_rate_pct=D("0"), term_months=12,
|
|
payout_date=date(2030, 1, 15), repayment_type="annuity")
|
|
k2 = Loan(name="K2", principal=D("2400.00"), annual_rate_pct=D("0"), term_months=12,
|
|
payout_date=date(2030, 1, 15), repayment_type="annuity")
|
|
db.add_all([miete, gehalt, reise, k1, k2])
|
|
db.commit()
|
|
return wohnen, miete, k1, k2
|
|
|
|
|
|
def test_plan_name():
|
|
assert PLAN_NAME == "System-Plan"
|
|
|
|
|
|
def test_system_plan_alle_kredite_ohne_modifikatoren(db):
|
|
wohnen, miete, k1, k2 = _basis(db)
|
|
b = compute_breakdown(db, date(2030, 3, 17), None, Filters())
|
|
assert b.month == M
|
|
assert {r.name: r.amount for r in b.rows} == {
|
|
"Miete": D("-1000.00"), "Gehalt": D("3000.00"), "Reise": D("-600.00"),
|
|
"K1": D("-100.00"), "K2": D("-200.00")}
|
|
assert {r.name: r.category for r in b.rows}["Miete"] == "Wohnen"
|
|
assert {r.name: r.category for r in b.rows}["Reise"] == "unkategorisiert"
|
|
assert {r.name: r.category for r in b.rows}["K1"] == "System-Kreditraten"
|
|
assert b.summary.income == D("3000.00") and b.summary.costs == D("1900.00")
|
|
|
|
|
|
def test_szenario_nur_zugeordnete_kredite_und_flags(db):
|
|
wohnen, miete, k1, k2 = _basis(db)
|
|
sc = Scenario(name="S", description="", include_recurring=True, include_planned=False)
|
|
db.add(sc)
|
|
db.flush()
|
|
db.add_all([
|
|
ScenarioLoan(scenario_id=sc.id, loan_id=k1.id),
|
|
ScenarioModifier(scenario_id=sc.id, target_type="category", target_id=wohnen.id,
|
|
kind="percent", value=D("50")),
|
|
ScenarioPlannedItem(scenario_id=sc.id, name="Szenario-Kauf", amount=D("-90.00"),
|
|
due=date(2030, 3, 5)),
|
|
])
|
|
db.commit()
|
|
b = compute_breakdown(db, M, sc, Filters())
|
|
got = {r.name: (r.amount, r.category) for r in b.rows}
|
|
assert got == {"Miete": (D("-500.00"), "Wohnen"), "Gehalt": (D("3000.00"), "unkategorisiert"),
|
|
"Szenario-Kauf": (D("-90.00"), "unkategorisiert"),
|
|
"K1": (D("-100.00"), "System-Kreditraten")}
|
|
|
|
|
|
def test_szenario_ohne_fixposten(db):
|
|
_basis(db)
|
|
sc = Scenario(name="S", description="", include_recurring=False, include_planned=True)
|
|
db.add(sc)
|
|
db.commit()
|
|
assert [r.name for r in compute_breakdown(db, M, sc, Filters()).rows] == ["Reise"]
|
|
|
|
|
|
def test_konflikt_nur_wenn_beide_zeilen_enthalten(db):
|
|
_basis(db)
|
|
rate = RecurringItem(name="Autorate", amount=D("-100.00"), rhythm="monthly", due_day=15)
|
|
db.add(rate)
|
|
db.commit()
|
|
b = compute_breakdown(db, M, None, Filters())
|
|
assert [(k.posten_name, k.kredit_name) for k in b.conflicts] == [("Autorate", "K1")]
|
|
assert compute_breakdown(db, M, None, Filters(loans=False)).conflicts == []
|
|
assert compute_breakdown(db, M, None, Filters(recurring=False)).conflicts == []
|
|
sc = Scenario(name="Ohne K1", description="")
|
|
db.add(sc)
|
|
db.commit()
|
|
assert compute_breakdown(db, M, sc, Filters()).conflicts == []
|