@@ -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()
report = accrual.AgingReport(rt_client, output, today)
rt_wrapper = rtutil.RT(RTClient())
report = accrual.AgingReport(rt_wrapper, output, today)
report.run(groups)