def test_service_make_reservation(self): #New customer with new reservation customer = Customer('999999', 'Tester', '2015-12-01', '2015-12-30') service = Service('Other', '510', 'test_service', 3.0, ['30', '60']) my_datetime = datetime.datetime.strptime('2015-12-25 12:00', '%Y-%m-%d %H:%M') service.reserve(customer, my_datetime, 60) customer_folder = '../Customer/' pickle_file = customer_folder + '999999.pkl' os.remove(pickle_file)
def test_service_change_schedule(self): customer = Customer('999999', 'Tester', '2015-12-01', '2015-12-30') service = Service('Other', '510', 'test_service', 3.0, ['30', '60']) my_datetime = datetime.datetime.strptime('2015-12-25 12:00', '%Y-%m-%d %H:%M') service.reserve(customer, my_datetime, 60) reservation = customer.find_reservation('2015-12-25', '12:00') service.modify_schedule(reservation, my_datetime, 60) customer_folder = '../Customer/' pickle_file = customer_folder + '999999.pkl' os.remove(pickle_file)