Make worker_version required

arkindex/documents/serializers/ml.py

@@ -380,7 +380,7 @@ class TranscriptionBulkItemSerializer(serializers.Serializer):
 
 
 class TranscriptionBulkSerializer(serializers.Serializer):
-    worker_version = serializers.PrimaryKeyRelatedField(queryset=WorkerVersion.objects.all(), default=None)
+    worker_version = serializers.PrimaryKeyRelatedField(queryset=WorkerVersion.objects.all())
     transcriptions = TranscriptionBulkItemSerializer(many=True)
 
     def validate(self, data):
@@ -414,7 +414,9 @@ class TranscriptionBulkSerializer(serializers.Serializer):
             for transcription in validated_data['transcriptions']
         ]
         Transcription.objects.bulk_create(transcriptions)
-        return {"transcriptions": transcriptions}
+
+        validated_data['transcriptions'] = transcriptions
+        return validated_data
 
 
 class ClassificationBulkSerializer(serializers.Serializer):

arkindex/documents/tests/test_bulk_transcriptions.py

@@ -2,10 +2,16 @@ from django.urls import reverse
 from rest_framework import status
 from arkindex_common.enums import TranscriptionType
 from arkindex.project.tests import FixtureAPITestCase
+from arkindex.dataimport.models import WorkerVersion
 
 
 class TestBulkTranscriptions(FixtureAPITestCase):
 
+    @classmethod
+    def setUpTestData(cls):
+        super().setUpTestData()
+        cls.worker_version = WorkerVersion.objects.get(worker__slug='reco')
+
     def test_bulk_transcriptions_requires_login(self):
         with self.assertNumQueries(0):
             response = self.client.post(reverse('api:transcription-bulk'))
@@ -15,8 +21,9 @@ class TestBulkTranscriptions(FixtureAPITestCase):
         self.client.force_login(self.user)
         self.user.corpus_right.all().delete()
         forbidden_element = self.corpus.elements.get(name='Volume 1, page 1r')
-        with self.assertNumQueries(3):
+        with self.assertNumQueries(4):
             response = self.client.post(reverse('api:transcription-bulk'), {
+                "worker_version": str(self.worker_version.id),
                 "transcriptions": [
                     {
                         "element_id": str(forbidden_element.id),
@@ -48,8 +55,9 @@ class TestBulkTranscriptions(FixtureAPITestCase):
         self.assertFalse(element1.transcriptions.exists())
         self.assertFalse(element2.transcriptions.exists())
 
-        with self.assertNumQueries(4):
+        with self.assertNumQueries(5):
             response = self.client.post(reverse('api:transcription-bulk'), {
+                "worker_version": str(self.worker_version.id),
                 "transcriptions": [
                     {
                         "element_id": str(element1.id),