added tests for office availability class

smash/web/tests/test_office_availability.py

+import logging
+
+from django.test import TestCase
+from django.utils import timezone
+import datetime
+import pandas as pd
+from datetime import timedelta
+from functions import create_availability, create_visit, create_appointment, create_appointment_type
+from web.utils import get_weekdays_in_period
+from web.officeAvailability import OfficeAvailability
+from web.models.holiday import Holiday
+from web.models.availability import Availability
+from web.models.appointment import Appointment
+from web.models.appointment_type_link import AppointmentTypeLink
+from web.models.constants import AVAILABILITY_HOLIDAY, AVAILABILITY_EXTRA
+from web.tests.functions import create_worker
+
+logger = logging.getLogger(__name__)
+
+class OfficeAvailabilityTest(TestCase):
+	def test_availability_cases(self):
+		#
+		today = timezone.now()
+		start_date = datetime.datetime(today.year, today.month, today.day, tzinfo=today.tzinfo) #today midnight
+		end_date = start_date + datetime.timedelta(days=1)
+
+		office_availability = OfficeAvailability('FirstName LastName', 
+			start=start_date, end=end_date, office_start='8:00', office_end='18:00')
+		
+		#no availabilties added yet
+		self.assertEqual(office_availability.is_available(), False)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), False)
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 0.0)
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 0.0)
+
+		#add availabilty from 8:00 to 18:00
+		weekday = list(get_weekdays_in_period(start_date, end_date))[0]
+		availability = create_availability(available_from='8:00', available_till='18:00', day_number=weekday)
+		office_availability.consider_this(availability)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False) #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 100.0)
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 41.70714781401804) # ((10*(60)+1)/(24*60.0+1))*100 That +1 is the zero minute
+
+		#add holiday from 16:00 to 18:00 # 2 hours less availability
+		start_date = datetime.datetime(today.year, today.month, today.day, 16, 00, tzinfo=today.tzinfo)
+		end_date = start_date + datetime.timedelta(hours=2)
+		holiday = Holiday(person=create_worker(), datetime_start=start_date, datetime_end=end_date, kind=AVAILABILITY_HOLIDAY)
+		office_availability.consider_this(holiday)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False)  #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 79.86688851913478) # ((8*60)/(10*60.0+1))*100  # the bordwer minute is 0, then no +1
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 33.31020124913255) # ((8*60)/(24*60.0+1))*100
+		
+		#add extra availability from 17:00 to 18:00 # 1 hour more availability 
+		start_date = datetime.datetime(today.year, today.month, today.day, 17, 00, tzinfo=today.tzinfo)
+		end_date = start_date + datetime.timedelta(hours=1)
+		extra_availability = Holiday(person=create_worker(), datetime_start=start_date, datetime_end=end_date, kind=AVAILABILITY_EXTRA)
+		office_availability.consider_this(extra_availability)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False)  #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 90.01663893510815) # ((9*60+1)/(10*60.0+1))*100 # the border minute is now 1 then +1
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 37.543372657876475)
+		
+		#add appointment from 9:00 to 10:00 # 1 hour less availability
+			#create visit
+		visit_start_date = start_date - datetime.timedelta(days=31)
+		visit_end_date   = end_date + datetime.timedelta(days=31)
+		visit = create_visit(subject=None, datetime_begin=visit_start_date, datetime_end=visit_end_date)
+			#create appointment
+		appointment_typeA = create_appointment_type(code='A', default_duration=40, description='test1')
+		appointment_typeB = create_appointment_type(code='B', default_duration=20, description='test2')
+		appointment_when = datetime.datetime(today.year, today.month, today.day, 9, 00, tzinfo=today.tzinfo)
+		appointment = create_appointment(visit=visit, when=appointment_when, length=60)
+		worker = create_worker()
+		app_type_linkA = AppointmentTypeLink(appointment=appointment, date_when=appointment_when, appointment_type=appointment_typeA, worker=worker)
+		app_type_linkB = AppointmentTypeLink(appointment=appointment, date_when=appointment_when+datetime.timedelta(minutes=20), appointment_type=appointment_typeB, worker=worker)
+		appointment.save()
+		#the availability percentage should be the same as before adding the extra availability
+		office_availability.consider_this(appointment)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False)  #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 79.86688851913478) # ((8*60)/(10*60.0+1))*100  # the bordwer minute is 0, then no +1
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 33.31020124913255) # ((8*60)/(24*60.0+1))*100
+		#consider the 2 AppointmentTypeLinks. The availability percentage shouldn't change
+		office_availability.consider_this(app_type_linkA)
+		office_availability.consider_this(app_type_linkB)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False)  #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 79.86688851913478) # ((8*60)/(10*60.0+1))*100  # the bordwer minute is 0, then no +1
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 33.31020124913255) # ((8*60)/(24*60.0+1))*100
+
+		#force add availability from 9:00 to 10:00 # 1 hour more availability
+		start_date = datetime.datetime(today.year, today.month, today.day, 9, 00, tzinfo=today.tzinfo).replace(second=0, microsecond=0)
+		end_date = start_date + datetime.timedelta(hours=1)
+		office_availability.add_availability(pd.date_range(start=start_date, end=end_date, freq='1T'), only_working_hours=False)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False)  #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 90.01663893510815) # ((9*60+1)/(10*60.0+1))*100 # the border minute is now 1 then +1
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 37.543372657876475)
+
+		#force remove availability from 9:00 to 12:00 # 3 hour less availability
+		start_date = datetime.datetime(today.year, today.month, today.day, 9, 00, tzinfo=today.tzinfo).replace(second=0, microsecond=0)
+		end_date = start_date + datetime.timedelta(hours=3)
+		office_availability.remove_availability(pd.date_range(start=start_date, end=end_date, freq='1T'), only_working_hours=True)
+		self.assertEqual(office_availability.is_available(only_working_hours=True), True)
+		self.assertEqual(office_availability.is_available(), False)  #less than 50%
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=True), 59.900166389351085) # ((6*60)/(10*60.0+1))*100 # the border minute is 0 then no +1
+		self.assertEqual(office_availability.get_availability_percentage(only_working_hours=False), 24.98265093684941)
+
+
+
+