test: add unit tests for partitions and processors module

Author: BryanttV

openedx_user_groups/partitions/user_group_partition_scheme.py

@@ -3,16 +3,46 @@
 """
 
 import logging
+from unittest.mock import Mock
 
 from django.utils.translation import gettext_lazy as _
-from lms.djangoapps.courseware.masquerade import (
-    get_course_masquerade,
-    get_masquerading_user_group,
-    is_masquerading_as_specific_student,
-)
+
+try:
+    from lms.djangoapps.courseware.masquerade import (
+        get_course_masquerade,
+        get_masquerading_user_group,
+        is_masquerading_as_specific_student,
+    )
+    from openedx.core import types
+    from xmodule.partitions.partitions import Group, UserPartition, UserPartitionError
+except ImportError:
+    get_course_masquerade = Mock()
+    get_masquerading_user_group = Mock()
+    is_masquerading_as_specific_student = Mock()
+    types = Mock()
+    Group = Mock()
+
+    class UserPartitionError(Exception):
+        """Mock UserPartitionError class for testing."""
+
+    class UserPartition:
+        """Mock UserPartition class for testing."""
+
+        # pylint: disable=redefined-builtin, too-many-positional-arguments, unused-argument
+        def __init__(self, id, name, description, groups, scheme, parameters, active=True):
+            self.parameters = parameters
+            self.scheme = scheme
+            self.id = id
+            self.name = name
+            self.description = description
+            self.active = active
+
+        @property
+        def groups(self):
+            return self.groups
+
+
 from opaque_keys.edx.keys import CourseKey
-from openedx.core import types
-from xmodule.partitions.partitions import Group, UserPartition, UserPartitionError
 
 from openedx_user_groups.models import UserGroup, UserGroupMembership
 from openedx_user_groups.toggles import is_user_groups_enabled
@@ -31,7 +61,7 @@ class UserGroupPartition(UserPartition):
     """
 
     @property
-    def groups(self):
+    def groups(self) -> list[Group]:
         """
         Dynamically generate groups (based on user groups) for this partition.
         """
@@ -98,7 +128,7 @@ def get_group_for_user(
 
         return user_groups
 
-    # pylint: disable=redefined-builtin, invalid-name
+    # pylint: disable=redefined-builtin, invalid-name, too-many-positional-arguments
     @classmethod
     def create_user_partition(
         cls,
@@ -152,7 +182,7 @@ def create_user_partition(
         return user_group_partition
 
 
-def create_user_group_partition_with_course_id(course_id):
+def create_user_group_partition_with_course_id(course_id: CourseKey) -> UserPartition | None:
     """
     Create and return the user group partition based only on course_id.
     If it cannot be created, None is returned.

openedx_user_groups/processors/user_group_partition_groups.py

@@ -3,13 +3,30 @@
 """
 
 from datetime import datetime
-from typing import Dict, Set
+from typing import Dict, List, Set
+from unittest.mock import Mock
 
 from opaque_keys.edx.keys import CourseKey
-from openedx.core import types
-from openedx.core.djangoapps.content.learning_sequences.api.processors.base import OutlineProcessor
-from xmodule.partitions.partitions import Group
-from xmodule.partitions.partitions_service import get_user_partition_groups
+
+try:
+    from openedx.core import types
+    from openedx.core.djangoapps.content.learning_sequences.api.processors.base import OutlineProcessor
+    from xmodule.partitions.partitions import Group
+    from xmodule.partitions.partitions_service import get_user_partition_groups
+except ImportError:
+    types = Mock()
+    Group = Mock()
+    get_user_partition_groups = Mock()
+
+    class OutlineProcessor:
+        """Mock OutlineProcessor class."""
+
+        def __init__(self, course_key, user, at_time):
+            """Initialize the OutlineProcessor."""
+            self.course_key = course_key
+            self.user = user
+            self.at_time = at_time
+
 
 from openedx_user_groups.partitions.user_group_partition_scheme import (
     USER_GROUP_PARTITION_ID,
@@ -37,7 +54,7 @@ def __init__(self, course_key: CourseKey, user: types.User, at_time: datetime):
             at_time (datetime): The time at which the data is loaded.
         """
         super().__init__(course_key, user, at_time)
-        self.user_groups: Dict[str, Group] = {}
+        self.user_groups: List[Group] = []
 
     def load_data(self, _) -> None:
         """

openedx_user_groups/toggles.py

@@ -4,12 +4,15 @@
 within the Open edX platform. These toggles allow for dynamic control of features without requiring code changes.
 """
 
+from unittest.mock import Mock
+
 from opaque_keys.edx.keys import CourseKey
 
 try:
     from openedx.core.djangoapps.waffle_utils import CourseWaffleFlag
 except ImportError:
-    CourseWaffleFlag = None
+    CourseWaffleFlag = Mock()
+
 
 # Namespace for all user group related waffle flags
 WAFFLE_FLAG_NAMESPACE = "user_groups"