@@ -422,16 +422,17 @@ def check_output(output, expect_patterns):
output.seek(0)
testutil.check_lines_match(iter(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
@pytest.mark.parametrize('invoice,expected', [
('rt:505/5050', "Zero balance outstanding since 2010-05-05"),
('rt:510/5100', "Zero balance outstanding since 2010-05-10"),
@@ -522,14 +523,14 @@ def run_aging_report(postings, today=None):
groups = {
key: group
for _, related in accrual.AccrualPostings.group_by_meta(postings, 'invoice')
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)
def test_aging_report(accrual_postings):
output = run_aging_report(accrual_postings)
check_aging_ods(output)