def test_update_an_order_not_exists(self):
""" Update a non-existing Order """
order_item1 = OrderItem(product_id=1, quantity=1, price=5, status="PLACED")
order_items = [order_item1]
order = Order(id=1234567, customer_id=111, order_items=order_items)
order.update()
self.assertRaises(DataValidationError)
def test_update_an_order(self):
""" Update an existing Order """
order_item1 = OrderItem(product_id=1, quantity=1, price=5, status="PLACED")
order_items = [order_item1]
order = Order(customer_id=111, order_items=order_items)
order.create()
self.assertTrue(order.id is not None)
order.customer_id = 234
order.update()
new_order = Order.find(order.id)
self.assertEqual(new_order.id, order.id)
self.assertEqual(new_order.customer_id, 234)
