def test_finish_reminds_to_save_orders(self):
messages = (
"You have not saved your order.\n"
"If you wish to continue, type finish again.\n"
"If you want to save your order, type save"
)
self.assertEqual(messages, pizza.finish("load", {}, self.orders))
def test_finish_exits_when_confirmed(self):
message = "Finishing order. Goodbye!"
self.assertEqual(message, pizza.finish("finish", {}, self.orders))
def test_finish_exits_when_all_orders_are_saved(self):
message = "Finishing order. Goodbye!"
self.assertEqual(message, pizza.finish("load", {}, {}))
