def test_flag_triggered_appointments(db_manager): for _ in range(ITERATIONS): appointments, locator_uuid_map = set_up_appointments( db_manager, MAX_ITEMS) triggered_appointments = random.sample(list(appointments.keys()), k=ITEMS) len_before_clean = len(appointments) Cleaner.flag_triggered_appointments(triggered_appointments, appointments, locator_uuid_map, db_manager) # ITEMS appointments should have been deleted from memory assert len(appointments) == len_before_clean - ITEMS # Make sure that all appointments are flagged as triggered in the db db_appointments = db_manager.load_all_triggered_flags() assert set(triggered_appointments).issubset(db_appointments)
def test_flag_triggered_appointments(db_manager): # Test that when an appointment is flagged and triggered it is deleted from memory and the flags are added to the db for _ in range(ITERATIONS): appointments, locator_uuid_map = set_up_appointments( db_manager, MAX_ITEMS) triggered_appointments = random.sample(list(appointments.keys()), k=ITEMS) # Flag the appointments Cleaner.flag_triggered_appointments(triggered_appointments, appointments, locator_uuid_map, db_manager) # Check that the flagged appointments are not in memory anymore assert not set(triggered_appointments).issubset(appointments) # Make sure that all appointments are flagged as triggered in the db db_appointments = db_manager.load_all_triggered_flags() assert set(triggered_appointments).issubset(db_appointments)