From 03c3d7f40728b2ac1c329ff9c6581d06cfb737d5 Mon Sep 17 00:00:00 2001
From: Helge Jung <hej@c3pb.de>
Date: Mon, 25 Dec 2023 19:19:12 +0100
Subject: [PATCH] schedule import: adjust tests to count speaker mappings, too

---
 src/core/tests/schedules.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/core/tests/schedules.py b/src/core/tests/schedules.py
index 94caccff0..a53bfff0e 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)
-- 
GitLab