Serializers: manage operation filter for warehouse

1 files changed, 69 insertions, 13 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index 2431dc9a5..a14c0fb8c 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -43,8 +43,9 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase):
         cr2 = self.create_context_record(
             data={"label": u"CR 2", "operation": ope2}
         )[1]
-        self.create_finds(data_base={"context_record": cr})[0]
-        self.create_finds(data_base={"context_record": cr2})[1]
+        self.create_finds(data_base={"context_record": cr})
+        self.create_finds(data_base={"context_record": cr2})
+        self.create_finds(data_base={"context_record": cr2})
         w1 = models.Warehouse.objects.create(
             name="Test1",
             external_id="test",
@@ -52,7 +53,12 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase):
         )
         w2 = models.Warehouse.objects.create(
             name="Test2",
-            external_id="test",
+            external_id="test2",
+            warehouse_type=models.WarehouseType.objects.all()[0],
+        )
+        w3 = models.Warehouse.objects.create(
+            name="Test3",
+            external_id="test3",
             warehouse_type=models.WarehouseType.objects.all()[0],
         )
         c1 = models.Container.objects.create(
@@ -95,6 +101,18 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase):
         f1.container = c3
         f1.container_ref = c4
         f1.save()
+        c5 = models.Container.objects.create(
+            location=w3,
+            responsible=w3,
+            container_type=models.ContainerType.objects.all()[0],
+            reference="Réf5",
+            index=1,
+            external_id="ref5-2"
+        )
+        f2 = self.finds[2]
+        f2.container = c5
+        f2.container_ref = c5
+        f2.save()
         wd1 = models.WarehouseDivision.objects.create(
             label="Étagère", txt_idx="etagere"
         )
@@ -109,6 +127,14 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase):
             warehouse=w1,
             division=wd2
         )
+        wl3 = models.WarehouseDivisionLink.objects.create(
+            warehouse=w2,
+            division=wd2
+        )
+        wl4 = models.WarehouseDivisionLink.objects.create(
+            warehouse=w3,
+            division=wd1
+        )
         models.ContainerLocalisation.objects.create(
             container=c1,
             division=wl1,
@@ -119,6 +145,16 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase):
             division=wl2,
             reference="A2"
         )
+        models.ContainerLocalisation.objects.create(
+            container=c2,
+            division=wl3,
+            reference="A4"
+        )
+        models.ContainerLocalisation.objects.create(
+            container=c3,
+            division=wl4,
+            reference="A5"
+        )
 
     def test_serialization(self):
         res = self.generic_serialization_test(
@@ -126,26 +162,46 @@ class SerializationTest(GenericSerializationTest, FindInit, TestCase):
         warehouse_json = json.loads(
             res[('warehouse', "archaeological_warehouse__Warehouse")]
         )
-        self.assertEqual(len(warehouse_json), 2)
+        self.assertEqual(len(warehouse_json), 3)
         container_json = json.loads(
             res[('warehouse', "archaeological_warehouse__Container")]
         )
-        self.assertEqual(len(container_json), 4)
+        self.assertEqual(len(container_json), 5)
+        div_json = json.loads(
+            res[('warehouse',
+                 "archaeological_warehouse__WarehouseDivisionLink")]
+        )
+        self.assertEqual(len(div_json), 4)
+        loca_json = json.loads(
+            res[('warehouse',
+                 "archaeological_warehouse__ContainerLocalisation")]
+        )
+        self.assertEqual(len(loca_json), 4)
 
-        """
         result_queryset = Operation.objects.filter(uuid=self.operations[0].uuid)
         res = self.generic_serialization_test(
-            serializers.find_serialization, no_test=True,
+            serializers.warehouse_serialization, no_test=True,
             kwargs={"operation_queryset": result_queryset}
         )
-        find_json = json.loads(
-            res[('finds', 'archaeological_finds__Find')]
+        warehouse_json = json.loads(
+            res[('warehouse', "archaeological_warehouse__Warehouse")]
         )
-        self.assertEqual(len(find_json), 1)
-        bfind_json = json.loads(
-            res[('finds', 'archaeological_finds__BaseFind')]
+        self.assertEqual(len(warehouse_json), 2)
+        container_json = json.loads(
+            res[('warehouse', "archaeological_warehouse__Container")]
         )
-        self.assertEqual(len(bfind_json), 1)
+        self.assertEqual(len(container_json), 2)
+        div_json = json.loads(
+            res[('warehouse',
+                 "archaeological_warehouse__WarehouseDivisionLink")]
+        )
+        self.assertEqual(len(div_json), 3)
+        loca_json = json.loads(
+            res[('warehouse',
+                 "archaeological_warehouse__ContainerLocalisation")]
+        )
+        self.assertEqual(len(loca_json), 3)
+        """
 
         result_queryset = ContextRecord.objects.filter(
             uuid=self.context_records[0].uuid)