Luís Murta
420a6cdfaa
Adds a new table, nordigen, with the access and refresh token, along with their validity. The Export/Import test would raise an integrety with the use of a real DB and the export of the transaction IDs, so add a try-except block to the database session to catch the error and re-raise an ImportError.
23 lines
597 B
Python
23 lines
597 B
Python
import datetime as dt
|
|
|
|
from pfbudget.db.client import Client
|
|
from pfbudget.db.model import Base, Nordigen
|
|
|
|
|
|
class MockClient(Client):
|
|
now = dt.datetime.now()
|
|
|
|
def __init__(self):
|
|
url = "sqlite://"
|
|
super().__init__(
|
|
url, execution_options={"schema_translate_map": {"pfbudget": None}}
|
|
)
|
|
Base.metadata.create_all(self.engine)
|
|
|
|
self.insert(
|
|
[
|
|
Nordigen("access", "token#1", self.now + dt.timedelta(days=1)),
|
|
Nordigen("refresh", "token#2", self.now + dt.timedelta(days=30)),
|
|
]
|
|
)
|