diff --git a/tests/testutil.py b/tests/testutil.py index c85e901a15dcfab00b5d68230ff78b15b43a48c6..b50ee4e507f0bfc1bcfcfffc58ef23e437bbf966 100644 --- a/tests/testutil.py +++ b/tests/testutil.py @@ -23,6 +23,8 @@ import beancount.core.data as bc_data from decimal import Decimal from pathlib import Path +from conservancy_beancount import rtutil + EXTREME_FUTURE_DATE = datetime.date(datetime.MAXYEAR, 12, 30) FUTURE_DATE = datetime.date.today() + datetime.timedelta(days=365 * 99) FY_START_DATE = datetime.date(2020, 3, 1) @@ -108,12 +110,26 @@ class Transaction: class TestConfig: - def __init__(self, repo_path=None): + def __init__(self, + repo_path=None, + rt_client=None, + ): self.repo_path = test_path(repo_path) + self._rt_client = rt_client + if rt_client is None: + self._rt_wrapper = None + else: + self._rt_wrapper = rtutil.RT(rt_client) def repository_path(self): return self.repo_path + def rt_client(self): + return self._rt_client + + def rt_wrapper(self): + return self._rt_wrapper + class _TicketBuilder: MESSAGE_ATTACHMENTS = [