def test_activity_instance_retrieval(self):
scheduling_manager = ActivityInstanceManager()
vendor1 = create_vendor('vendor1')
activity1 = create_activity('activity1', vendor1)
activity2 = create_activity('activity2', vendor1)
activity1_instance = ActivityInstance(activity=activity1,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Australia/Sydney')).replace(hour=3, minute=30, month=1).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('Australia/Sydney')).replace(hour=4, minute=0, month=1).isoformat('T')),
price=12.12, capacity=10, bookings=0)
activity1_instance.save()
activity2_instance = ActivityInstance(activity=activity2,
start_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('US/Eastern')).replace(hour=8, month=6).isoformat('T')),
end_time=util.iso8601_to_utc_timestamp(datetime.now(tz=pytz.timezone('US/Eastern')).replace(hour=15, month=6).isoformat('T')),
price=12.12, capacity=10, bookings=4)
activity2_instance.save()
retrieved_activity1_instance = scheduling_manager.retrieve_activity_instance(
activity1_instance.activity_instance_id)
self.assertIsNotNone(retrieved_activity1_instance)
retrieved_activity2_instance = scheduling_manager.retrieve_activity_instance(
activity2_instance.activity_instance_id)
self.assertIsNotNone(retrieved_activity2_instance)
self.assertIsNone(scheduling_manager.retrieve_activity_instance("0"))
self.assertIsNone(scheduling_manager.retrieve_activity_instance(""))
self.assertIsNone(scheduling_manager.retrieve_activity_instance(None))
class ActivityInstanceView(View):
def __init__(self):
self.activity_instance_manager = ActivityInstanceManager()
def get(self, request, activity_instance_id):
activity_instance = self.activity_instance_manager.retrieve_activity_instance(activity_instance_id)
if activity_instance is None:
raise Http404
else:
activity_instance = convert_activity_instance_timestamp_to_iso(activity_instance)
return HttpResponse(util.obj_to_json(activity_instance), content_type='application/json')
def put(self, request, activity_instance_id):
attributes_dictionary = ast.literal_eval(request.read().decode('utf-8'))
if attributes_dictionary.get('start_time') is not None:
attributes_dictionary['start_time'] = util.iso8601_to_utc_timestamp(attributes_dictionary['start_time'])
if attributes_dictionary.get('end_time') is not None:
attributes_dictionary['end_time'] = util.iso8601_to_utc_timestamp(attributes_dictionary['end_time'])
activity_instance = self.activity_instance_manager.update_activity_instance(activity_instance_id,
**attributes_dictionary)
if activity_instance is None:
raise Http404
else:
activity_instance = convert_activity_instance_timestamp_to_iso(activity_instance)
return HttpResponse(util.obj_to_json(activity_instance), content_type='application/json')
def post(self, request, activity_instance_id=None):
activity_id = request.POST.get('activity_id')
start_time = util.iso8601_to_utc_timestamp(request.POST.get('start_time'))
end_time = util.iso8601_to_utc_timestamp(request.POST.get('end_time'))
price = float(request.POST.get('price'))
capacity = int(request.POST.get('capacity'))
bookings = request.POST.get('bookings')
if bookings is None:
activity_instance = self.activity_instance_manager.create_activity_instance(activity_id=activity_id,
start_time=start_time,
end_time=end_time,
price=price,
capacity=capacity)
else:
activity_instance = self.activity_instance_manager.create_activity_instance(activity_id=activity_id,
start_time=start_time,
end_time=end_time,
price=price,
capacity=capacity,
bookings=int(bookings))
activity_instance = convert_activity_instance_timestamp_to_iso(activity_instance)
return HttpResponse(util.obj_to_json(activity_instance), content_type='application/json')
def delete(self, request, activity_instance_id):
if self.activity_instance_manager.delete_activity_instance(activity_instance_id):
return HttpResponse()
else:
raise Http404
