fix: use PublishableEntity.key instead of IDs

openedx · Aug 28, 2024 · 0e15dd0 · 0e15dd0
1 parent 7ec9ba5
commit 0e15dd0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/openedx_learning/apps/authoring/collections/api.py b/openedx_learning/apps/authoring/collections/api.py
@@ -161,13 +161,13 @@ def remove_from_collections(
     return total_deleted
 
 
-def get_object_collections(object_id: int) -> QuerySet[Collection]:
+def get_object_collections(object_key: str) -> QuerySet[Collection]:
     """
-    Get all collections associated with a given PublishableEntity.
+    Get all collections associated with a given PublishableEntity.key.
 
     Only enabled collections are returned.
     """
-    entity = PublishableEntity.objects.get(pk=object_id)
+    entity = PublishableEntity.objects.get(key=object_key)
     return entity.collections.filter(enabled=True).order_by("pk")
 
 

diff --git a/tests/openedx_learning/apps/authoring/collections/test_api.py b/tests/openedx_learning/apps/authoring/collections/test_api.py
@@ -388,7 +388,7 @@ def test_get_object_collections(self):
         Tests fetching the enabled collections which contain a given object.
         """
         collections = collection_api.get_object_collections(
-            self.published_entity.id,
+            self.published_entity.key,
         )
         assert list(collections) == [
             self.collection1,