diff --git a/tests/test_reports_accrual.py b/tests/test_reports_accrual.py index 6b7ab910cc2f1bffd3c2f9773fc096210b52f034..676d2ae4986627bb49f0ee4f0ccfa862707e0ca6 100644 --- a/tests/test_reports_accrual.py +++ b/tests/test_reports_accrual.py @@ -425,10 +425,11 @@ def check_output(output, expect_patterns): def run_outgoing(rt_id, postings, rt_client=None): if rt_client is None: rt_client = RTClient() + rt_wrapper = rtutil.RT(rt_client) if not isinstance(postings, core.RelatedPostings): postings = accruals_by_meta(postings, rt_id, 'rt-id', wrap_type=accrual.AccrualPostings) output = io.StringIO() - report = accrual.OutgoingReport(rt_client, output) + report = accrual.OutgoingReport(rt_wrapper, output) report.run({rt_id: postings}) return output @@ -525,8 +526,8 @@ def run_aging_report(postings, today=None): for key, group in related.make_consistent() } output = io.BytesIO() - rt_client = RTClient() - report = accrual.AgingReport(rt_client, output, today) + rt_wrapper = rtutil.RT(RTClient()) + report = accrual.AgingReport(rt_wrapper, output, today) report.run(groups) return output