From 4d92ad9457a001cbd300b927ddc2129e91efa663 Mon Sep 17 00:00:00 2001
From: Erwan Rouchet <rouchet@teklia.com>
Date: Wed, 23 Nov 2022 17:44:04 +0100
Subject: [PATCH] Add missing feature flag overrides on Transkribus import
 tests

---
 arkindex/process/tests/test_transkribus_import.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/arkindex/process/tests/test_transkribus_import.py b/arkindex/process/tests/test_transkribus_import.py
index e54b17c66e..9788bc61d7 100644
--- a/arkindex/process/tests/test_transkribus_import.py
+++ b/arkindex/process/tests/test_transkribus_import.py
@@ -43,6 +43,7 @@ class TestTranskribusImport(FixtureAPITestCase):
             configuration={}
         )
 
+    @override_settings(ARKINDEX_FEATURES={"transkribus": True})
     def test_requires_login(self):
         response = self.client.post(reverse("api:import-transkribus"))
         self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
@@ -56,6 +57,7 @@ class TestTranskribusImport(FixtureAPITestCase):
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertEqual(response.json(), ["Transkribus import is unavailable due to the transkribus feature being disabled."])
 
+    @override_settings(ARKINDEX_FEATURES={"transkribus": True})
     def test_requires_transkribus_email(self):
         User.objects.update(transkribus_email=None)
         self.client.force_login(self.user)
@@ -65,6 +67,7 @@ class TestTranskribusImport(FixtureAPITestCase):
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertEqual(response.json(), {"__all__": ["You have not registered your transkribus email"]})
 
+    @override_settings(ARKINDEX_FEATURES={"transkribus": True})
     @patch("transkribus.TranskribusAPI.list_user_collection")
     def test_arkindex_has_not_access(self, mock_transkribus):
         # Not a mistake: Transkribus client raises `Exception` directly when it runs out of retries
@@ -76,6 +79,7 @@ class TestTranskribusImport(FixtureAPITestCase):
         self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
         self.assertEqual(response.json(), {"collection_id" : ["User arkindex@teklia.com is not a member of the collection 12345"]})
 
+    @override_settings(ARKINDEX_FEATURES={"transkribus": True})
     @patch("transkribus.TranskribusAPI.list_user_collection")
     def test_user_has_not_access(self, mock_transkribus):
         mock_transkribus.return_value = [{"email": "arkindex@teklia.com"}]
@@ -88,7 +92,8 @@ class TestTranskribusImport(FixtureAPITestCase):
         self.assertEqual(response.json(), {"collection_id": ["User nope@nope.fr is not a member of the collection 12345"]})
 
     @override_settings(
-        PONOS_RECIPE={}
+        PONOS_RECIPE={},
+        ARKINDEX_FEATURES={"transkribus": True},
     )
     @patch("transkribus.TranskribusAPI.list_user_collection")
     def test_create_import(self, mock_transkribus):
@@ -154,7 +159,8 @@ class TestTranskribusImport(FixtureAPITestCase):
         })
 
     @override_settings(
-        PONOS_RECIPE={}
+        PONOS_RECIPE={},
+        ARKINDEX_FEATURES={'transkribus': True},
     )
     @patch("transkribus.TranskribusAPI.list_user_collection")
     def test_corpus_worker_version_list_transkribus(self, mock_transkribus):
-- 
GitLab