raccoongang · monteri · Mar 13, 2024 · Mar 13, 2024 · Mar 13, 2024 · monteri
diff --git a/lms/djangoapps/course_home_api/outline/serializers.py b/lms/djangoapps/course_home_api/outline/serializers.py
@@ -36,6 +36,9 @@ def get_blocks(self, block):  # pylint: disable=missing-function-docstring
         if graded and scored:
             icon = 'fa-pencil-square-o'
 
+        if block_type == 'vertical':
+            icon = self.get_vertical_icon_class(block)
+
         if 'special_exam_info' in block:
             description = block['special_exam_info'].get('short_description')
             icon = block['special_exam_info'].get('suggested_icon', 'fa-pencil-square-o')
@@ -62,6 +65,20 @@ def get_blocks(self, block):  # pylint: disable=missing-function-docstring
             serialized.update(self.get_blocks(child))
         return serialized
 
+    @staticmethod
+    def get_vertical_icon_class(block):
+        """
+        Get the icon class for a vertical block based on its children.
+        """
+        children = block.get('children', [])
+        child_classes = {child.get('type') for child in children}
+        new_class = 'other'
+        icon_call_priority = ['video', 'problem']
+        for higher_class in icon_call_priority:
+            if higher_class in child_classes:
+                new_class = higher_class
+        return new_class
-    @staticmethod
-    def get_vertical_icon_class(block):
-        """
-        Get the icon class for a vertical block based on its children.
-        """
-        children = block.get('children', [])
-        child_classes = {child.get('type') for child in children}
-        new_class = 'other'
-        icon_call_priority = ['video', 'problem']
-        for higher_class in icon_call_priority:
-            if higher_class in child_classes:
-                new_class = higher_class
-        return new_class
+@staticmethod
+def get_vertical_icon_class(block, icon_mapping=None):
+    """
+    Returns the icon class for a vertical block based on its children.
+
+    Args:
+        block (object): xblock object
+
+    Returns:
+        str: The icon class name to be used, or 'other' if no match is found.
+    """
+
+    if icon_mapping is None:
+        icon_mapping = {
+            'video': 'video',
+            'problem': 'problem',
+        }
+
+    for child in block.get('children', []):
+        if child.get('type') in icon_mapping:
+            return icon_mapping[child.get('type')]
+
+    return 'other' 
-    @staticmethod
-    def get_vertical_icon_class(block):
-        """
-        Get the icon class for a vertical block based on its children.
-        """
-        children = block.get('children', [])
-        child_classes = {child.get('type') for child in children}
-        new_class = 'other'
-        icon_call_priority = ['video', 'problem']
-        for higher_class in icon_call_priority:
-            if higher_class in child_classes:
-                new_class = higher_class
-        return new_class
+@staticmethod
+def get_vertical_icon_class(block, icon_mapping=None):
+    """
+    Returns the icon class for a vertical block based on its children.
+
+    Args:
+        block (object): xblock object
+
+    Returns:
+        str: The icon class name to be used, or 'other' if no match is found.
+    """
+
+    if icon_mapping is None:
+        icon_mapping = {
+            'video': 'video',
+            'problem': 'problem',
+        }
+
+    for child in block.get('children', []):
+        if child.get('type') in icon_mapping:
+            return icon_mapping[child.get('type')]
+
+    return 'other' 
+
 
 class CourseGoalsSerializer(serializers.Serializer):
     """