Luís Murta
1cfc8dbe38
Move the token creation/renewal code into the NordigenClient and remove external access. The NordigenClient now also takes the DB client in the constructor. While creating the unit test, noticed that the nordigen mocks for downloaded transactions could match the simple transactions mock, thus helping in the online download command. Also created the nordigen.NordigenClient mock, with the used methods mocked and raising a NotImplement when a new endpoint is requested.
74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
from datetime import date
|
|
from decimal import Decimal
|
|
|
|
from pfbudget.db.model import (
|
|
BankTransaction,
|
|
CategorySelector,
|
|
MoneyTransaction,
|
|
Note,
|
|
SplitTransaction,
|
|
Transaction,
|
|
TransactionCategory,
|
|
TransactionTag,
|
|
)
|
|
|
|
# The simple and simple_transformed match the nordigen mocks
|
|
simple = [
|
|
BankTransaction(date(2023, 1, 14), "string", Decimal("328.18"), bank="bank"),
|
|
BankTransaction(date(2023, 2, 14), "string", Decimal("947.26"), bank="bank"),
|
|
]
|
|
|
|
simple_transformed = [
|
|
BankTransaction(
|
|
date(2023, 1, 14),
|
|
"",
|
|
Decimal("328.18"),
|
|
bank="bank",
|
|
category=TransactionCategory("category#1", CategorySelector.algorithm),
|
|
),
|
|
BankTransaction(
|
|
date(2023, 2, 14),
|
|
"",
|
|
Decimal("947.26"),
|
|
bank="bank",
|
|
category=TransactionCategory("category#2", CategorySelector.algorithm),
|
|
),
|
|
]
|
|
|
|
bank = [
|
|
BankTransaction(date(2023, 1, 1), "", Decimal("-10"), bank="bank#1"),
|
|
BankTransaction(date(2023, 1, 1), "", Decimal("-10"), bank="bank#2"),
|
|
]
|
|
|
|
money = [
|
|
MoneyTransaction(date(2023, 1, 1), "", Decimal("-10")),
|
|
MoneyTransaction(date(2023, 1, 1), "", Decimal("-10")),
|
|
]
|
|
|
|
__original = Transaction(date(2023, 1, 1), "", Decimal("-10"), split=True)
|
|
__original.id = 9000
|
|
|
|
split = [
|
|
__original,
|
|
SplitTransaction(date(2023, 1, 1), "", Decimal("-5"), original=__original.id),
|
|
SplitTransaction(date(2023, 1, 1), "", Decimal("-5"), original=__original.id),
|
|
]
|
|
|
|
tagged = [
|
|
Transaction(
|
|
date(2023, 1, 1),
|
|
"",
|
|
Decimal("-10"),
|
|
tags={TransactionTag("tag#1"), TransactionTag("tag#1")},
|
|
)
|
|
]
|
|
|
|
noted = [
|
|
Transaction(
|
|
date(2023, 1, 1),
|
|
"",
|
|
Decimal("-10"),
|
|
note=Note("note#1"),
|
|
)
|
|
]
|