imacat/mia-accounting
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 41 Wiki Activity

Simplified the logic in the add_txn function in testlib_txn.py.

Browse Source
This commit is contained in:
依瑪貓 2023-03-01 00:27:39 +08:00
parent f049b5d7ee
commit 1402a12f04
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/testlib_txn.py
View File

@ -393,10 +393,11 @@ def add_txn(client: httpx.Client, form: dict[str, str]) -> int:
:return: The newly-added transaction ID.
"""
prefix: str = "/accounting/transactions"
txn_type: str \
= "income" if len({x for x in form if "-debit-" in x}) == 0 else\
("expense" if len({x for x in form if "-credit-" in x}) == 0 else
"transfer")
txn_type: str = "transfer"
if len({x for x in form if "-debit-" in x}) == 0:
txn_type = "income"
elif len({x for x in form if "-credit-" in x}) == 0:
txn_type = "expense"
store_uri = f"{prefix}/store/{txn_type}"
response: httpx.Response = client.post(store_uri, data=form)
assert response.status_code == 302