def test_view_order_main(self): from datetime import datetime from objects import Shift, Delivery, Order, Tip from processes.view import view_order from utility.utility import now, To_Datetime # no data entered order = Order(Delivery(Shift(now().date())), 123) test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tTotal tip: $0.00\n' self.assertEqual(test, expected) # card order.tip = Tip(3.97) test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tCard tip: $3.97\n' self.assertEqual(test, expected) # cash order.tip = Tip(cash=3.97) test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tCash tip: $3.97\n' self.assertEqual(test, expected) # card and cash order.tip = Tip(3, 2) test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tTotal tip: $5.00\n'\ '\tCard tip: $3.00\n'\ '\tCash tip: $2.00\n' self.assertEqual(test, expected) # unknown order.tip = Tip(unknown=3.97) test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tUnknown tip: $3.97\n' self.assertEqual(test, expected) # distance order.miles_traveled = 3.9 test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tUnknown tip: $3.97\n'\ '\tDistance to address: 3.9 miles\n' self.assertEqual(test, expected) # end time order.end_time = datetime.strptime('15:27:42', '%H:%M:%S') test = view_order(order) expected = 'Order I.D. #: 123\n'\ '\tUnknown tip: $3.97\n'\ '\tDistance to address: 3.9 miles\n'\ '\tCompleted at: 03:27:42 PM\n' self.assertEqual(test, expected)