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