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