Fix related elements type filter

arkindex/documents/api/elements.py

@@ -217,7 +217,7 @@ class RelatedElementsList(ListAPIView):
         }
         type_param = self.request.query_params.get('type')
         if type_param is not None:
-            filtering['type'] = type_param
+            filtering['type__slug'] = type_param
         related = Element.objects.get_related(self.kwargs['pk'], **filtering)
         prefetch_related_objects(related, 'corpus', 'zone__image__server')
         return related

arkindex/documents/tests/test_elements_api.py

@@ -809,6 +809,21 @@ class TestElementsAPI(FixtureAPITestCase):
             }
         ])
 
+    def test_related_elements(self):
+        act = self.corpus.elements.get(name='Act 1')
+        response = self.client.get(reverse('api:related-elements', kwargs={'pk': str(act.id)}))
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertListEqual(
+            [r['name'] for r in response.json()['results']],
+            ['Volume 1', 'Surface A']
+        )
+        response = self.client.get(reverse('api:related-elements', kwargs={'pk': str(act.id)}) + '?type=volume')
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertListEqual(
+            [r['name'] for r in response.json()['results']],
+            ['Volume 1']
+        )
+
     def test_create_metadata_verified(self):
         response = self.client.post(reverse('api:element-metadata', kwargs={'pk': str(self.vol.id)}))
         self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)