Serializers: warehouse module

1 files changed, 206 insertions, 2 deletions

diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index 07171ebb4..2431dc9a5 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -17,13 +17,217 @@
 
 # See the file COPYING for details.
 
+import json
+
 from archaeological_finds.tests import FindInit
 
 from ishtar_common.tests import WizardTest, WizardTestFormData as FormData, \
-    TestCase, WAREHOUSE_FIXTURES
+    TestCase, WAREHOUSE_FIXTURES, GenericSerializationTest, COMMON_FIXTURES
 
 from ishtar_common.models import IshtarSiteProfile, SpatialReferenceSystem
-from archaeological_warehouse import models, views, forms
+from archaeological_operations.models import Operation
+from archaeological_context_records.models import ContextRecord
+
+from archaeological_warehouse import models, views, forms, serializers
+
+
+class SerializationTest(GenericSerializationTest, FindInit, TestCase):
+    fixtures = COMMON_FIXTURES + WAREHOUSE_FIXTURES
+
+    def setUp(self):
+        ope1 = self.create_operation()[0]
+        ope2 = self.create_operation()[1]
+        cr = self.create_context_record(
+            data={"label": u"CR 1", "operation": ope1}
+        )[0]
+        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]
+        w1 = models.Warehouse.objects.create(
+            name="Test1",
+            external_id="test",
+            warehouse_type=models.WarehouseType.objects.all()[0],
+        )
+        w2 = models.Warehouse.objects.create(
+            name="Test2",
+            external_id="test",
+            warehouse_type=models.WarehouseType.objects.all()[0],
+        )
+        c1 = models.Container.objects.create(
+            location=w1,
+            responsible=w1,
+            container_type=models.ContainerType.objects.all()[0],
+            reference="Réf1",
+            index=1,
+            external_id="ref1-1"
+        )
+        c2 = models.Container.objects.create(
+            location=w2,
+            responsible=w1,
+            container_type=models.ContainerType.objects.all()[0],
+            reference="Réf2",
+            index=2,
+            external_id="ref2-2"
+        )
+        f0 = self.finds[0]
+        f0.container = c1
+        f0.container_ref = c2
+        f0.save()
+        c3 = models.Container.objects.create(
+            location=w1,
+            responsible=w2,
+            container_type=models.ContainerType.objects.all()[0],
+            reference="Réf3",
+            index=1,
+            external_id="ref3-1"
+        )
+        c4 = models.Container.objects.create(
+            location=w2,
+            responsible=w2,
+            container_type=models.ContainerType.objects.all()[0],
+            reference="Réf4",
+            index=2,
+            external_id="ref4-2"
+        )
+        f1 = self.finds[1]
+        f1.container = c3
+        f1.container_ref = c4
+        f1.save()
+        wd1 = models.WarehouseDivision.objects.create(
+            label="Étagère", txt_idx="etagere"
+        )
+        wd2 = models.WarehouseDivision.objects.create(
+            label="Allée", txt_idx="allee"
+        )
+        wl1 = models.WarehouseDivisionLink.objects.create(
+            warehouse=w1,
+            division=wd1
+        )
+        wl2 = models.WarehouseDivisionLink.objects.create(
+            warehouse=w1,
+            division=wd2
+        )
+        models.ContainerLocalisation.objects.create(
+            container=c1,
+            division=wl1,
+            reference="A1"
+        )
+        models.ContainerLocalisation.objects.create(
+            container=c1,
+            division=wl2,
+            reference="A2"
+        )
+
+    def test_serialization(self):
+        res = self.generic_serialization_test(
+            serializers.warehouse_serialization)
+        warehouse_json = json.loads(
+            res[('warehouse', "archaeological_warehouse__Warehouse")]
+        )
+        self.assertEqual(len(warehouse_json), 2)
+        container_json = json.loads(
+            res[('warehouse', "archaeological_warehouse__Container")]
+        )
+        self.assertEqual(len(container_json), 4)
+
+        """
+        result_queryset = Operation.objects.filter(uuid=self.operations[0].uuid)
+        res = self.generic_serialization_test(
+            serializers.find_serialization, no_test=True,
+            kwargs={"operation_queryset": result_queryset}
+        )
+        find_json = json.loads(
+            res[('finds', 'archaeological_finds__Find')]
+        )
+        self.assertEqual(len(find_json), 1)
+        bfind_json = json.loads(
+            res[('finds', 'archaeological_finds__BaseFind')]
+        )
+        self.assertEqual(len(bfind_json), 1)
+
+        result_queryset = ContextRecord.objects.filter(
+            uuid=self.context_records[0].uuid)
+        res = self.generic_serialization_test(
+            serializers.find_serialization, no_test=True,
+            kwargs={"cr_queryset": result_queryset}
+        )
+        find_json = json.loads(
+            res[('finds', 'archaeological_finds__Find')]
+        )
+        self.assertEqual(len(find_json), 1)
+        bfind_json = json.loads(
+            res[('finds', 'archaeological_finds__BaseFind')]
+        )
+        self.assertEqual(len(bfind_json), 1)
+
+        result_queryset = models.Find.objects.filter(
+            uuid=self.finds[0].uuid)
+        res = self.generic_serialization_test(
+            serializers.find_serialization, no_test=True,
+            kwargs={"find_queryset": result_queryset}
+        )
+        find_json = json.loads(
+            res[('finds', 'archaeological_finds__Find')]
+        )
+        self.assertEqual(len(find_json), 1)
+        bfind_json = json.loads(
+            res[('finds', 'archaeological_finds__BaseFind')]
+        )
+        self.assertEqual(len(bfind_json), 1)
+        """
+
+    """
+    def test_ope_serialization_with_find_filter(self):
+        res = self.generic_serialization_test(
+            operation_serialization, no_test=True,
+        )
+        ope_json = json.loads(
+            res[('operations', 'archaeological_operations__Operation')]
+        )
+        self.assertEqual(len(ope_json), 2)
+
+        result_queryset = models.Find.objects.filter(
+            uuid=self.finds[0].uuid)
+        res = self.generic_serialization_test(
+            operation_serialization, no_test=True,
+            kwargs={"find_queryset": result_queryset}
+        )
+        ope_json = json.loads(
+            res[('operations', 'archaeological_operations__Operation')]
+        )
+        self.assertEqual(len(ope_json), 1)
+
+    def test_cr_serialization_with_find_filter(self):
+        res = self.generic_serialization_test(
+            cr_serialization, no_test=True,
+        )
+        cr_json = json.loads(
+            res[('context_records',
+                 'archaeological_context_records__ContextRecord')]
+        )
+        self.assertEqual(len(cr_json), 2)
+
+        result_queryset = models.Find.objects.filter(
+            uuid=self.finds[0].uuid)
+        res = self.generic_serialization_test(
+            cr_serialization, no_test=True,
+            kwargs={"find_queryset": result_queryset}
+        )
+        cr_json = json.loads(
+            res[('context_records',
+                 'archaeological_context_records__ContextRecord')]
+        )
+        self.assertEqual(len(cr_json), 1)
+    """
+
+    def test_restore(self):
+        current_number, zip_filename = self.generic_restore_test_genzip(
+            serializers.WAREHOUSE_MODEL_LIST,
+            serializers.warehouse_serialization)
+        self.generic_restore_test(zip_filename, current_number,
+                                  serializers.WAREHOUSE_MODEL_LIST)
 
 
 class WarehouseWizardCreationTest(WizardTest, FindInit, TestCase):