def test_load_order_completed(self):
from objects import Order
from os import mkdir, remove, rmdir, path
from processes.load import load_order
order = Order(self.shift.deliveries[0])
file_list = order.file_list()
# create file and directory
mkdir(self.shift.deliveries[0].file_list()['directory'])
with open(file_list['info'], 'w') as info_file:
info_file.write(self.shift.deliveries[0].orders[0].csv())
# check that file was created and baseline
self.assertTrue(path.exists(file_list['info']))
self.assertIsNot(order.tip, self.shift.deliveries[0].orders[0].tip)
self.assertNotEqual(order.miles_traveled,
self.shift.deliveries[0].orders[0].miles_traveled)
self.assertNotEqual(order.end_time,
self.shift.deliveries[0].orders[0].end_time)
# run function
order = load_order(order)
# check that data was loaded correctly
self.assertEqual(order.tip.csv(),
self.shift.deliveries[0].orders[0].tip.csv())
self.assertEqual(order.miles_traveled,
self.shift.deliveries[0].orders[0].miles_traveled)
self.assertEqual(order.end_time,
self.shift.deliveries[0].orders[0].end_time)
# delete file and directory
remove(file_list['info'])
rmdir(self.shift.deliveries[0].file_list()['directory'])
def result(self):
# order
if self.user_selection.lower() == 'o':
from objects import Order
import os
from processes.consolidate import consolidate_order
from processes.track import track_order
order = Order(self.delivery)
if os.path.exists(order.file_list()['directory']):
order.load_current()
order = track_order(order)
consolidate_order(order)
self.delivery.add_order(order)
# extra stop
elif self.user_selection.lower() == 'e':
from objects import Extra_Stop
import os
from processes.consolidate import consolidate_extra_stop
from processes.track import track_extra_stop
extra_stop = Extra_Stop(self.delivery).load_current()
extra_stop = track_extra_stop(extra_stop)
consolidate_extra_stop(extra_stop)
self.delivery.add_extra_stop(extra_stop)
# complete
elif self.user_selection.lower() == 'c':
from processes.consolidate import consolidate_delivery
from processes.track import end_delivery
self.delivery = end_delivery(self.delivery)
consolidate_delivery(self.delivery)
self.loop_condition = False
self.complete = True
# view
elif self.user_selection.lower() == 'v':
View_Delivery_Menu(self.delivery)
# revise
elif self.user_selection.lower() == 'r':
from processes.revise import Revise_Delivery
revise_delivery = Revise_Delivery(self.delivery)
self.delivery = revise_delivery.delivery
# back
elif self.user_selection.lower() == 'b':
self.loop_condition = False
