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)
def test_customer_add_reservation(self): customer = Customer('999999', 'Tester', '2015-12-01', '2015-12-30') rid = '1' sid = '201' name = 'Swedish Massage' RMT = '{:%Y-%m-%d %H:%M}'.format(datetime.datetime.now()) r_date = '2015-12-25' start_time = '12:00' end_time = '13:00' time_length = 60 price = 3.0 customer.add_reservation(rid, sid, name, RMT, r_date, start_time, end_time, time_length, price)