Syndication - fix tests

1 files changed, 30 insertions, 22 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index cc0a8cb1b..371857337 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -87,10 +87,8 @@ from ishtar_common.models import (
     Document,
     ValueFormater,
     Regexp,
-    ApiUser,
-    ApiSearchModel,
 )
-from ishtar_common.models_imports import ImporterDefault, ImporterDefaultValues
+from ishtar_common.models_rest import ApiUser, ApiSearchModel, ApiSheetFilter
 from archaeological_files.models import File, FileType
 
 from ishtar_common import forms_common
@@ -4445,12 +4443,16 @@ class ApiTest(OperationInitTest, APITestCase):
         self.create_operation(self.user, self.orgas[0])
         self.create_operation(self.user, self.orgas[0])
 
-        keys = ("old-neolithic", "middle-neolithic", "recent-neolithic",
-                "final-neolithic")
-        old_neo = models.Period.objects.get(txt_idx='old-neolithic')
-        middle_neo = models.Period.objects.get(txt_idx='middle-neolithic')
-        recent_neo = models.Period.objects.get(txt_idx='recent-neolithic')
-        final_neo = models.Period.objects.get(txt_idx='final-neolithic')
+        keys = (
+            "old-neolithic",
+            "middle-neolithic",
+            "recent-neolithic",
+            "final-neolithic",
+        )
+        old_neo = models.Period.objects.get(txt_idx="old-neolithic")
+        middle_neo = models.Period.objects.get(txt_idx="middle-neolithic")
+        recent_neo = models.Period.objects.get(txt_idx="recent-neolithic")
+        final_neo = models.Period.objects.get(txt_idx="final-neolithic")
         self.operation_0 = models.Operation.objects.get(pk=self.operations[0].pk)
         self.operation_0.code_patriarche = "28123"
         self.operation_0.save()
@@ -4769,10 +4771,7 @@ class ApiTest(OperationInitTest, APITestCase):
 
     def _mock_search(self, mock_get, model, url):
         _get_item = get_item(
-            model,
-            "get_" + model.SLUG,
-            model.SLUG,
-            no_permission_check=True
+            model, "get_" + model.SLUG, model.SLUG, no_permission_check=True
         )
         mock_get.__call__ = _get_item
 
@@ -4824,7 +4823,9 @@ class ApiTest(OperationInitTest, APITestCase):
         url = base_url + params
         self._mock_search(mock_get, models.Operation, "/get-operation/" + params)
         response = self.client.get(
-            url, format="json", HTTP_AUTHORIZATION=self.auth_token,
+            url,
+            format="json",
+            HTTP_AUTHORIZATION=self.auth_token,
         )
         self.assertEqual(response.status_code, 200)
         j = json.loads(response.content.decode())
@@ -4839,7 +4840,9 @@ class ApiTest(OperationInitTest, APITestCase):
         url = base_url + params
         self._mock_search(mock_get, models.Operation, "/get-operation/" + params)
         response = self.client.get(
-            url, format="json", HTTP_AUTHORIZATION=self.auth_token,
+            url,
+            format="json",
+            HTTP_AUTHORIZATION=self.auth_token,
         )
         self.assertEqual(response.status_code, 200)
         j = json.loads(response.content.decode())
@@ -4852,18 +4855,23 @@ class ApiTest(OperationInitTest, APITestCase):
         # test timeout
         pass
 
-    def test_distant_sheet_display(self):
-        # POV: local
-        # test query limitation
-        pass
-
     def test_sheet_display(self):
         # POV: external
         api_search_model = self.create_api_search_model()
         url = reverse("api-get-operation", args=[self.operations[0].pk])
-        data = {"submited": 1, "search_vector": "28"}
         response = self.client.get(
-            url, format="json", HTTP_AUTHORIZATION=self.auth_token, data=data
+            url, format="json", HTTP_AUTHORIZATION=self.auth_token
+        )
+        self.assertEqual(response.status_code, 200)
+        j = json.loads(response.content.decode())
+        self.assertIsNotNone(j.get("cached_label", None))
+        # filter
+        ApiSheetFilter.objects.create(
+            api_search_model=api_search_model, key="cached_label"
+        )
+        response = self.client.get(
+            url, format="json", HTTP_AUTHORIZATION=self.auth_token
         )
         self.assertEqual(response.status_code, 200)
         j = json.loads(response.content.decode())
+        self.assertIsNone(j.get("cached_label", None))