44 lines
1.5 KiB
Python
44 lines
1.5 KiB
Python
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from decimal import Decimal
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
|
||
|
|
from quartermaster import service
|
||
|
|
from quartermaster.models import Section
|
||
|
|
|
||
|
|
|
||
|
|
def test_add_and_total(db):
|
||
|
|
service.add_entry(db, Section.income, "Paycheck", Decimal("2500.00"))
|
||
|
|
service.add_entry(db, Section.income, "Side gig", Decimal("250.50"))
|
||
|
|
entries = service.list_entries(db, Section.income)
|
||
|
|
assert [e.name for e in entries] == ["Paycheck", "Side gig"]
|
||
|
|
assert service.section_total(entries) == Decimal("2750.50")
|
||
|
|
|
||
|
|
|
||
|
|
def test_delete_entry(db):
|
||
|
|
entry = service.add_entry(db, Section.other, "One-off", Decimal("10.00"))
|
||
|
|
service.delete_entry(db, entry.id)
|
||
|
|
assert service.list_entries(db, Section.other) == []
|
||
|
|
|
||
|
|
|
||
|
|
def test_set_debt_target(db):
|
||
|
|
dm = service.add_entry(db, Section.debt_minimum, "Card A", Decimal("50.00"))
|
||
|
|
target = service.set_debt_target(db, dm.id)
|
||
|
|
assert target.debt_minimum_id == dm.id
|
||
|
|
assert target.entry is not None and target.entry.name == "Card A"
|
||
|
|
|
||
|
|
|
||
|
|
def test_debt_target_rejects_non_debt_minimum(db):
|
||
|
|
income = service.add_entry(db, Section.income, "Paycheck", Decimal("1.00"))
|
||
|
|
with pytest.raises(ValueError):
|
||
|
|
service.set_debt_target(db, income.id)
|
||
|
|
|
||
|
|
|
||
|
|
def test_debt_target_cleared_on_delete(db):
|
||
|
|
dm = service.add_entry(db, Section.debt_minimum, "Card B", Decimal("75.00"))
|
||
|
|
service.set_debt_target(db, dm.id)
|
||
|
|
service.delete_entry(db, dm.id)
|
||
|
|
target = service.get_debt_target(db)
|
||
|
|
assert target.debt_minimum_id is None
|