def testGenerateEvent(self):
startDatetime = generateStartDatetime("2018-12-01T08:00:00.000")
timestampTupleList = generateTimestamps(startDatetime, 2)
eventList = generate_airport_events(timestampTupleList)
expectedDay2EventDict = {'eventCategory': 'airport', 'participants': [], 'airportCode': 'LAX',
'isClosed': False}
self.assertDictContainsSubset(expectedDay2EventDict, eventList[1].to_dict())
def populate_events(start_string="2018-12-07T08:00:00.000", num_days=35, event_category="airport"):
start_datetime = generateStartDatetime(start_string)
timestamp_tuple_list = generateTimestamps(start_datetime, num_days)
assert event_category == "airport" or event_category == "campus"
event_list = generate_airport_events(timestamp_tuple_list) if event_category == "airport" else generate_uc_events(
timestamp_tuple_list)
for event in event_list:
event_ref = EventDao().create(event)
print(event_ref)
def testGenerateTimestamps(self):
startDatetime = generateStartDatetime("2018-12-17T08:00:00.000")
timestampTupleList = generateTimestamps(startDatetime, 2)
expectedValue = [[1545033600.0, 1545119999.0, "2018-12-17"], [1545120000.0, 1545206399.0, "2018-12-18"]]
count = 0
for startTimestamp, endTimestamp, dateStr in timestampTupleList:
self.assertEqual(startTimestamp, expectedValue[count][0])
self.assertEqual(endTimestamp, expectedValue[count][1])
self.assertEqual(dateStr, expectedValue[count][2])
count += 1
def execute(self) -> list:
"""
:return: a list of DocumentReference for documents just created
"""
refs = list()
start_datetime = generateStartDatetime(self.start_string)
timestamp_tuple_list = generateTimestamps(start_datetime, self.num_days)
event_list = generate_airport_events(timestamp_tuple_list) \
if self.event_category == "airport" \
else generate_uc_events(timestamp_tuple_list)
for event in event_list:
event_ref = EventDao().create(event)
refs.append(event_ref)
print(event_ref)
return refs