def test_remove_all(self):
""" Test dropping and recreating all tables in the database """
self._add_two_test_payments()
self.assertEqual(len(Payment.all()), 2)
Payment.disconnect()
Payment.remove_all()
self.assertEqual(len(Payment.all()), 0)
def test_delete_a_payment(self):
""" Delete a Payment """
payment = Payment(order_id="1",
customer_id="1",
available=True,
payments_type="credit card")
payment.save()
self.assertEqual(len(Payment.all()), 1)
# delete the pet and make sure it isn't in the database
payment.delete()
self.assertEqual(len(Payment.all()), 0)
def test_add_a_payment(self):
""" Create a payment and add it to the database """
payments = Payment.all()
self.assertEqual(payments, [])
payment = Payment(order_id="1",
customer_id="1",
available=True,
payments_type="credit card")
self.assertTrue(payment is not None)
self.assertEqual(payment.id, None)
payment.save()
# Asert that it was assigned an id and shows up in the database
self.assertEqual(payment.id, 1)
payments = Payment.all()
self.assertEqual(len(payments), 1)
def list_payments():
""" Returns all of the Payments """
app.logger.info('Request for payments list')
payments = []
customer_id = request.args.get('customer_id')
order_id = request.args.get('order_id')
if customer_id:
app.logger.info('Request for payments list with customer_id : %s', customer_id)
payments = Payment.find_by_customer(customer_id)
elif order_id:
app.logger.info('Request for payments list with order_id : %s', order_id)
payments = Payment.find_by_order(order_id)
else:
payments = Payment.all()
results = [payment.serialize() for payment in payments]
return make_response(jsonify(results), status.HTTP_200_OK)
