def test_fetch_events_as_ics(self, mock_extract_event, mock_get_event_ids): mock_extract_event.return_value = '1234' mock_get_event_ids.return_value = ['1', '2'] responses.add(responses.GET, "https://www.abdn.ac.uk/ims/seminars/ical/1/", body="raw ics 1", content_type="text/calendar") responses.add(responses.GET, "https://www.abdn.ac.uk/ims/seminars/ical/2/", body="raw ics 2", content_type="text/calendar") ics = "BEGIN:VCALENDAR\r\nclean ics 1\nclean ics 2\nEND:VCALENDAR\r\n" self.assertEqual(fetch_events_as_ics(), ics) mock_extract_event.assert_called_once('raw ics 1') mock_extract_event.assert_called_once('raw ics 2')
def update_calendar(): db = Database() db.insert_ics(fetch_events_as_ics()) db.commit() db.close()