diff --git a/src/core/tests/schedules.py b/src/core/tests/schedules.py index 94caccff0cb9226bb58ed7ef15d9b65fe1712379..a53bfff0e9c44514d82a0dfc01b219ab957c332f 100644 --- a/src/core/tests/schedules.py +++ b/src/core/tests/schedules.py @@ -310,7 +310,7 @@ class ScheduleTests(TestCase): import_type=ScheduleJSONSupport.identifier, import_url=f"file://{Path(__file__).parent.joinpath('import_data', 'schedule-2020.json')}", ) - self.check_json(src, rooms=1, events=1, room_name='Yellow Room') + self.check_json(src, rooms=1, events=1, speakers=2, room_name='Yellow Room', event_id='12345') @override_settings(SCHEDULES_SUPPORT_FILE_PROTOCOL=True) def test_json2021(self): @@ -320,7 +320,7 @@ class ScheduleTests(TestCase): import_type=ScheduleJSONSupport.identifier, import_url=f"file://{Path(__file__).parent.joinpath('import_data', 'schedule-2021.json')}", ) - self.check_json(src, rooms=2, events=1, room_name='Gray Room') + self.check_json(src, rooms=2, events=1, speakers=2, room_name='Gray Room', event_id='12345') @override_settings(SCHEDULES_SUPPORT_FILE_PROTOCOL=True) def test_json_frab(self): @@ -340,9 +340,9 @@ class ScheduleTests(TestCase): import_type=ScheduleJSONFrabSupport.identifier, import_url=f"file://{Path(__file__).parent.joinpath('import_data', 'frab.schedule.json')}", ) - self.check_json(src, rooms=3, events=1, room_name='Saal 1', event_id='12338') + self.check_json(src, rooms=3, events=1, speakers=3, room_name='Saal 1', event_id='12338') - def check_json(self, src, rooms, events, room_name='Gray Room', event_id='12345'): + def check_json(self, src: ScheduleSource, rooms: int, events: int, speakers: int, room_name: str, event_id: str): self.assertFalse(src.has_running_import) job = src.imports.create(state=ScheduleSourceImport.State.PREPARED) @@ -371,7 +371,7 @@ class ScheduleTests(TestCase): self.assertIsNotNone(e1) # check that mappings exist - self.assertEqual(rooms + events, src.mappings.count()) # 3 = 2 rooms + 1 event + self.assertEqual(rooms + events + speakers, src.mappings.count()) r1_m = src.mappings.get(mapping_type=ScheduleSourceMapping.MappingType.ROOM, source_id=room_name, local_id=r1.id) e1_m = src.mappings.get(mapping_type=ScheduleSourceMapping.MappingType.EVENT, source_id=event_id, local_id=e1.id) self.assertIsNotNone(r1_m)