def test_delete_an_order(self):
""" Delete an Order """
date = datetime.now()
order = Order(customer_id=1, date=date, shipped=True)
order.save()
self.assertEqual(len(Order.all()), 1)
order.delete()
self.assertEqual(len(Order.all()), 0)
def test_fetch_all_orders(self):
""" Test fetching all Orders """
date = datetime.now()
order = Order(customer_id=1, date=date, shipped=True)
order.save()
order2 = Order(customer_id=2, date=date, shipped=False)
order2.save()
Order.all()
self.assertEqual(len(Order.all()), 2)
def test_add_an_order(self):
""" Create an Order and add it to the database """
date = datetime.now()
orders = Order.all()
self.assertEqual(orders, [])
order = Order(customer_id=1, date=date, shipped=True)
self.assertEqual(order.id, None)
order.save()
self.assertEqual(order.id, 1)
orders = Order.all()
self.assertEqual(len(orders), 1)
def get_order():
token = request.args.get('token')
if token is None:
return json_response({'error': 'Token is empty'})
order = Order.all().filter('token = ', token).fetch(1)
if len(order) is 1:
return json_response(order[0]._entity)
else: return json_response({'error': 'Access denied'})
def delete_order():
token = request.args.get('token')
if token is None:
return json_response({'error': 'Token is empty'})
order = Order.all().filter('token = ', token).fetch(1)
if len(order) is 1:
order = order[0]
order.delete()
return json_response({'success': True})
else: return json_response({'error': 'Access denied'})
def test_update_an_order(self):
""" Update an Order """
date = datetime.now()
order = Order(customer_id=1, date=date, shipped=True)
order.save()
self.assertEqual(order.id, 1)
order.shipped = False
order.save()
orders = Order.all()
self.assertEqual(len(orders), 1)
self.assertEqual(orders[0].shipped, False)
def list_orders():
"""
Retrieve a list of Orders
This endpoint will return all Orders unless a query parameter is specificed
---
tags:
- Orders
description: The Orders endpoint allows you to query Orders
parameters:
- name: customer_name
in: query
description: the name of the customer who placed the Order
required: false
type: string
responses:
200:
description: An array of Orders
schema:
type: array
items:
schema:
id: Order
properties:
id:
type: integer
description: unique id assigned internally by service
customer_name:
type: string
description: the name of the person who placed the order
amount_paid:
type: integer
description: the amount the order came out to
"""
orders = []
customer_name = request.args.get('customer_name')
if customer_name:
orders = Order.find_by_customer_name(customer_name)
else:
orders = Order.all()
results = [order.serialize() for order in orders]
return make_response(jsonify(results), status.HTTP_200_OK)
def list_orders():
""" Returns all of the Orders """
orders = Order.all()
results = [order.serialize() for order in orders]
return make_response(jsonify(results), status.HTTP_200_OK)
