def test_generate(): usages_before = datetime.date.today() + datetime.timedelta(30) invoices = invoicelib.invoice_collection.generate(test_data.bizplace_id, usages_before=usages_before) for invoice in invoices: usaged_ids = invoice.usages usages = (usagelib.usage_resource.info(id) for id in usaged_ids) member_ids = set(usage.member for usage in usages) expected_members = set(dbaccess.get_billfrom_members(invoice.member)) assert member_ids.issubset(expected_members) env.context.pgcursor.connection.commit()
def uninvoiced(self, member_id, res_owner_id, start, end): """ find uninvoiced usages of a member """ member_ids = dbaccess.get_billfrom_members(member_id) return self.find(start=start, end=end, res_owner_ids=[res_owner_id], member_ids=member_ids, uninvoiced=True)