def test_receipt_saved_from_new(session):
"""Assert that the receipt is saved to the table."""
payment_account = factory_payment_account()
payment = factory_payment()
payment_account.save()
payment.save()
i = factory_invoice(payment_id=payment.id, account_id=payment_account.id)
i.save()
factory_invoice_reference(i.id).save()
receipt_service = ReceiptService()
receipt_service.receipt_number = '1234567890'
receipt_service.invoice_id = i.id
receipt_service.receipt_date = datetime.now()
receipt_service.receipt_amount = 100
receipt_service = receipt_service.save()
receipt_service = ReceiptService.find_by_id(receipt_service.id)
assert receipt_service is not None
assert receipt_service.id is not None
assert receipt_service.receipt_date is not None
assert receipt_service.invoice_id is not None
receipt_service = ReceiptService.find_by_invoice_id_and_receipt_number(
i.id, receipt_service.receipt_number)
assert receipt_service is not None
assert receipt_service.id is not None
def test_receipt_invalid_lookup(session):
"""Test invalid lookup."""
receipt = ReceiptService.find_by_id(999)
assert receipt is not None
assert receipt.id is None
receipt = ReceiptService.find_by_invoice_id_and_receipt_number(999, '1234567890')
assert receipt is not None
assert receipt.id is None