def test_line_saved_from_new(session):
"""Assert that the payment is saved to the table."""
payment_account = factory_payment_account()
payment = factory_payment()
payment_account.save()
payment.save()
invoice = factory_invoice(payment.id, payment_account.id)
invoice.save()
factory_invoice_reference(invoice.id).save()
fee_schedule = FeeSchedule.find_by_filing_type_and_corp_type('CP', 'OTANN')
line = factory_payment_line_item(invoice.id, fee_schedule_id=fee_schedule.fee_schedule_id)
line.save()
line = factory_payment_line_item(invoice.id, fee_schedule_id=fee_schedule.fee_schedule_id, status='DELETED')
line.save()
p = PaymentLineService.find_by_id(line.id)
assert p is not None
assert p.id is not None
assert p.invoice_id is not None
assert p.filing_fees is not None
assert p.fee_schedule_id is not None
assert p.gst is None
assert p.pst is None
assert p.line_item_status_code is not None
assert p.priority_fees is None
assert p.future_effective_fees is None
invoice = Invoice.find_by_id(invoice.id)
schema = InvoiceSchema()
d = schema.dump(invoice)
assert d.get('id') == invoice.id
assert len(d.get('line_items')) == 1
def test_invoice_find_by_id(session):
"""Assert a invoice is stored.
Start with a blank database.
"""
payment_account = factory_payment_account()
payment = factory_payment()
payment_account.save()
payment.save()
invoice = factory_invoice(payment=payment, payment_account=payment_account)
invoice.save()
assert invoice.find_by_id(invoice.id) is not None
schema = InvoiceSchema()
d = schema.dump(invoice)
assert d.get('id') == invoice.id
def asdict(self):
"""Return the invoice as a python dict."""
invoice_schema = InvoiceSchema()
d = invoice_schema.dump(self._dao)
return d
def asdict(self, include_dynamic_fields: bool = False):
"""Return the invoice as a python dict."""
invoice_schema = InvoiceSchema()
d = invoice_schema.dump(self._dao)
self._add_dynamic_fields(d, include_dynamic_fields)
return d
