def test_get_invoice(client):
with db.connection() as DB:
invoice = Invoice(amount=5, description='Test invoice')
DB.add(invoice)
response = client.get('/invoices/{0}/'.format(invoice.id))
data = response.json
data['amount'] = Decimal(data['amount'])
data['amount_remaining'] = Decimal(data['amount_remaining'])
invoice_details = invoice.to_json()
invoice_details['created_at'] = invoice_details['created_at'].strftime(DATE_FORMAT) # noqa
invoice_details['updated_at'] = invoice_details['updated_at'].strftime(DATE_FORMAT) # noqa
assert data == invoice_details
def test_list_invoices(client):
with db.connection() as DB:
invoice = Invoice(amount=5, description='Test invoice')
DB.add(invoice)
response = client.get('/invoices/')
data = response.json
for n in range(len(data)):
data[n]['amount'] = Decimal(data[n]['amount'])
data[n]['amount_remaining'] = Decimal(data[n]['amount_remaining'])
invoice_details = invoice.to_json()
invoice_details['created_at'] = invoice_details['created_at'].strftime(DATE_FORMAT) # noqa
invoice_details['updated_at'] = invoice_details['updated_at'].strftime(DATE_FORMAT) # noqa
assert data == [invoice_details]
