Serializations: operations, sites, parcels

1 files changed, 37 insertions, 7 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 647f0f86e..d0726fbdc 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -38,7 +38,7 @@ from django.utils.translation import ugettext_lazy as _, pgettext, pgettext_lazy
 from . import models
 
 from ishtar_common.views import document_deletion_steps
-from archaeological_operations import views
+from archaeological_operations import views, serializers
 
 from ishtar_common.models import OrganizationType, Organization, ItemKey, \
     ImporterType, IshtarUser, TargetKey, ImporterModel, IshtarSiteProfile, \
@@ -54,7 +54,7 @@ from ishtar_common import forms_common
 from ishtar_common.tests import WizardTest, WizardTestFormData as FormData, \
     create_superuser, create_user, TestCase, OPERATION_FIXTURES, \
     AutocompleteTestBase, AcItem, OPERATION_TOWNS_FIXTURES, FILE_FIXTURES, \
-    FILE_TOWNS_FIXTURES
+    COMMON_FIXTURES, GenericSerializationTest, WAREHOUSE_FIXTURES
 
 
 class FileInit(object):
@@ -743,6 +743,35 @@ class ImportStepByStepTest(ImportTest, TestCase):
         self.assertIn(error, response.content.decode('utf-8'))
 
 
+class SerializationTest(GenericSerializationTest, TestCase):
+    fixtures = COMMON_FIXTURES + WAREHOUSE_FIXTURES
+
+    def setUp(self):
+        self.username, self.password, self.user = create_superuser()
+        operation = create_operation(self.user,
+                                     values={"code_patriarche": "66666"})
+        site = models.ArchaeologicalSite.objects.create(
+            reference="ref-site"
+        )
+        operation.top_sites.add(site)
+        default = {"town": Town.objects.create(numero_insee="66666"),
+                   "section": 'A', 'parcel_number': '1', "operation": operation}
+        parcel = models.Parcel.objects.create(**default)
+        models.ParcelOwner.objects.create(
+            parcel=parcel, owner=Person.objects.create(),
+            start_date=datetime.date.today(), end_date=datetime.date.today())
+
+    def test_serialization(self):
+        self.generic_serialization_test(serializers.operation_serialization)
+
+    def test_conf_restore(self):
+        current_number, zip_filename = self.generic_restore_test_genzip(
+            serializers.OPERATION_MODEL_LIST,
+            serializers.operation_serialization)
+        self.generic_restore_test(zip_filename, current_number,
+                                  serializers.OPERATION_MODEL_LIST)
+
+
 class ParcelTest(ImportTest, TestCase):
     fixtures = OPERATION_TOWNS_FIXTURES
 
@@ -1029,13 +1058,14 @@ class OperationInitTest(object):
             self.create_towns()
         return self.towns[0]
 
-    def create_parcel(self, data={}):
+    def create_parcel(self, data=None):
         default = {'town': self.get_default_town(),
                    'section': 'A', 'parcel_number': '1'}
         if not hasattr(self, 'operations'):
             self.create_operation()
         default['operation'] = self.operations[0]
-        default.update(data)
+        if data:
+            default.update(data)
         if not getattr(self, 'parcels', None):
             self.parcels = []
         self.parcels.append(models.Parcel.objects.create(**default))
@@ -2721,7 +2751,7 @@ class SiteTest(TestCase, OperationInitTest):
             codename='view_own_operation'))
         self.orgas = self.create_orgas(self.user)
 
-    def test_create_or_update_top_operation(self):
+    def test_create_or_update_top_operations(self):
         operation_0 = self.create_operation(self.user, self.orgas[0])[0]
         operation_1 = self.create_operation(self.alt_user, self.orgas[0])[1]
         site = models.ArchaeologicalSite.objects.create(
@@ -2731,12 +2761,12 @@ class SiteTest(TestCase, OperationInitTest):
         q = models.ArchaeologicalSite.objects.filter(reference='ref-site')
         site = q.all()[0]
         # creation not forced - no creation
-        self.assertIsNone(site.top_operation)
+        self.assertEqual(site.top_operations.count(), 0)
 
         site.create_or_update_top_operation(create=True)
         site = q.all()[0]
         # a default operation has been created
-        self.assertIsNotNone(site.top_operation)
+        self.assertEqual(site.top_operations.count(), 1)
         self.assertTrue(site.top_operation.virtual_operation)
         self.assertEqual(site.top_operation.right_relations.count(), 0)