mirror of
https://github.com/Zproger/Aurum.git
synced 2026-10-13 11:06:40 +07:00
0134cdb6bf
Один чек часто смешивает несколько категорий трат (гипермаркет: часть — продукты, часть — бытовая химия). Раньше такую транзакцию приходилось огрублять до одной категории или дробить вручную на несколько записей. Теперь у транзакции есть второй режим: 2+ доли (TransactionSplit) с суммами по разным категориям, которые обязаны в точности складываться в сумму транзакции. - Новая модель TransactionSplit + миграция. - split_rule_violation() проверяет инварианты на создании и редактировании. - category_rollup.py — общий helper, которым теперь пользуются Dashboard, отчёт по категориям и бюджеты, учитывая доли сплитов наравне с обычными транзакциями. - Фильтр по категории в списке транзакций находит и сплит-транзакции. - Сплиты попадают в полный бэкап/восстановление БД. - 13 новых backend-тестов. - Фронтенд: переключатель «Разбить на несколько категорий» в форме транзакции с повторяемыми строками и индикатором остатка суммы; строка транзакции в списке показывает все категории сплита.
94 lines
3.8 KiB
Python
94 lines
3.8 KiB
Python
"""Full-database backup export/import (services/backup_service.py) — only
|
|
the parts this change touched: subcategories (self-referential parent_id)
|
|
and tags (many-to-many) surviving a round trip.
|
|
"""
|
|
from httpx import AsyncClient
|
|
|
|
from tests.helpers import txn_payload as _txn
|
|
|
|
|
|
async def test_backup_roundtrip_preserves_subcategories_and_tags(client: AsyncClient, account_id, categories):
|
|
parent = await client.post("/categories", json={"name": "Custom Parent", "kind": "expense", "color": "#e34948"})
|
|
parent_id = parent.json()["id"]
|
|
child = await client.post(
|
|
"/categories", json={"name": "Custom Child", "kind": "expense", "color": "#e34948", "parent_id": parent_id}
|
|
)
|
|
child_id = child.json()["id"]
|
|
|
|
tag = (await client.post("/tags", json={"name": "Roundtrip"})).json()["id"]
|
|
created = await client.post("/transactions", json=_txn(account_id, category_id=child_id, tag_ids=[tag]))
|
|
txn_id = created.json()["id"]
|
|
|
|
export_resp = await client.get("/backup/export")
|
|
assert export_resp.status_code == 200
|
|
payload = export_resp.json()
|
|
|
|
import_resp = await client.post("/backup/import", json=payload)
|
|
assert import_resp.status_code == 200, import_resp.text
|
|
|
|
refetched_categories = {c["id"]: c for c in (await client.get("/categories")).json()}
|
|
assert refetched_categories[child_id]["parent_id"] == parent_id
|
|
|
|
refetched_txn = next(t for t in (await client.get("/transactions")).json()["items"] if t["id"] == txn_id)
|
|
assert [t["id"] for t in refetched_txn["tags"]] == [tag]
|
|
|
|
|
|
async def test_backup_import_rejects_transaction_with_unknown_tag_id(client: AsyncClient, account_id, categories):
|
|
export_resp = await client.get("/backup/export")
|
|
payload = export_resp.json()
|
|
|
|
created = await client.post(
|
|
"/transactions", json=_txn(account_id, category_id=categories["Groceries"]["id"], date="2026-01-01")
|
|
)
|
|
payload["transactions"] = (await client.get("/transactions")).json()["items"]
|
|
# Reshape to the backup wire format (flat FKs, not nested account/category
|
|
# objects) and inject a tag_id that doesn't exist in payload["tags"].
|
|
payload["transactions"] = [
|
|
{
|
|
"id": t["id"],
|
|
"account_id": t["account_id"],
|
|
"category_id": t["category_id"],
|
|
"transfer_account_id": t["transfer_account_id"],
|
|
"type": t["type"],
|
|
"amount": t["amount"],
|
|
"description": t["description"],
|
|
"merchant": t["merchant"],
|
|
"notes": t["notes"],
|
|
"date": t["date"],
|
|
"tag_ids": [999999] if t["id"] == created.json()["id"] else [],
|
|
}
|
|
for t in payload["transactions"]
|
|
]
|
|
|
|
resp = await client.post("/backup/import", json=payload)
|
|
assert resp.status_code == 400
|
|
|
|
|
|
async def test_backup_roundtrip_preserves_transaction_splits(client: AsyncClient, account_id, categories):
|
|
groceries = categories["Groceries"]["id"]
|
|
shopping = categories["Shopping"]["id"]
|
|
created = await client.post(
|
|
"/transactions",
|
|
json=_txn(
|
|
account_id,
|
|
amount="100.00",
|
|
category_id=None,
|
|
splits=[
|
|
{"category_id": groceries, "amount": "70.00"},
|
|
{"category_id": shopping, "amount": "30.00", "note": "household chemicals"},
|
|
],
|
|
),
|
|
)
|
|
txn_id = created.json()["id"]
|
|
|
|
payload = (await client.get("/backup/export")).json()
|
|
assert len(payload["transaction_splits"]) == 2
|
|
|
|
import_resp = await client.post("/backup/import", json=payload)
|
|
assert import_resp.status_code == 200, import_resp.text
|
|
|
|
refetched = next(t for t in (await client.get("/transactions")).json()["items"] if t["id"] == txn_id)
|
|
splits_by_note = {s["note"]: s for s in refetched["splits"]}
|
|
assert splits_by_note["household chemicals"]["category_id"] == shopping
|
|
assert refetched["category"] is None
|