def test_create(self):
"""Ensures the create method works by creating an invoice and checking the database matches"""
invoice = Invoice.create(datetime.date(1999, 1, 1), 'test_payer',
'test_payee', ['XDSA', 'SDWF'])
id = invoice.id
with database() as db:
results = db.query(
'SELECT * FROM invoices WHERE invoice_number = %s', id)
self.assertEqual(
results,
[(id, datetime.date(1999, 1, 1), 'test_payer', 'test_payee')])
results = db.query(
'SELECT * FROM invoice_items WHERE invoice_number = %s', id)
self.assertEqual(results, [('XDSA', id), ('SDWF', id)])
invoice.delete()
def api_log_invoice():
invoice = Invoice.create(request.form.get('date'),
request.form.get('payer'),
request.form.get('payee'),
request.form.getlist('items'))
return redirect('/invoice/{}'.format(invoice.id))
