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)