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))