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)