[Cherry-Pick-Enterprise] Adding pagination to label relationship endp…

…oint
sensiml · Jan 16, 2025 · b2b044b · b2b044b
1 parent 5dad6dd
commit b2b044b
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/src/server/datamanager/tests/views/test_labelrelationship.py b/src/server/datamanager/tests/views/test_labelrelationship.py
@@ -974,6 +974,49 @@ def test_project_label_relationship(self, client):
             ]
         )
 
+        label_url = reverse(
+            "project-label-relationship-list",
+            kwargs={"project_uuid": self.project.uuid},
+        )
+
+        response = client.get(
+            f"{label_url}?page_size=1",
+        )
+
+        assert len(response.data["results"]) == 1
+        assert sorted(list(response.data["results"][0].keys())) == sorted(
+            [
+                "uuid",
+                "capture",
+                "label",
+                "label_value",
+                "capture_sample_sequence_start",
+                "capture_sample_sequence_end",
+                "label_info",
+                "segmenter",
+                "created_at",
+                "last_modified",
+            ]
+        )
+
+        response = client.get(response.data["next"].replace("http://testserver", ""))
+
+        assert len(response.data["results"]) == 1
+        assert sorted(list(response.data["results"][0].keys())) == sorted(
+            [
+                "uuid",
+                "capture",
+                "label",
+                "label_value",
+                "capture_sample_sequence_start",
+                "capture_sample_sequence_end",
+                "label_info",
+                "segmenter",
+                "created_at",
+                "last_modified",
+            ]
+        )
+
     def test_project_label_relationship_delete_multi(self, client):
         self.make_label_labelvalue(client)
         clv = []

diff --git a/src/server/datamanager/views/label_relationship.py b/src/server/datamanager/views/label_relationship.py
@@ -164,6 +164,7 @@ def get_queryset(self):
 )
 class ProjectLabelRelationshipListView(generics.ListAPIView):
     serializer_class = V2CaptureLabelValueSerializer
+    pagination_class = QueryParamCursorPagination
 
     permission_classes = (
         permissions.IsAuthenticated,