diff options
Diffstat (limited to 'archaeological_operations/tests.py')
| -rw-r--r-- | archaeological_operations/tests.py | 39 |
1 files changed, 38 insertions, 1 deletions
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py index d0726fbdc..2c06f5a35 100644 --- a/archaeological_operations/tests.py +++ b/archaeological_operations/tests.py @@ -750,9 +750,11 @@ class SerializationTest(GenericSerializationTest, TestCase): self.username, self.password, self.user = create_superuser() operation = create_operation(self.user, values={"code_patriarche": "66666"}) + create_operation(self.user, values={"code_patriarche": "66667"}) site = models.ArchaeologicalSite.objects.create( reference="ref-site" ) + operation.archaeological_sites.add(site) operation.top_sites.add(site) default = {"town": Town.objects.create(numero_insee="66666"), "section": 'A', 'parcel_number': '1', "operation": operation} @@ -762,7 +764,42 @@ class SerializationTest(GenericSerializationTest, TestCase): start_date=datetime.date.today(), end_date=datetime.date.today()) def test_serialization(self): - self.generic_serialization_test(serializers.operation_serialization) + res = self.generic_serialization_test( + serializers.operation_serialization) + ope_json = json.loads( + res[('operations', 'archaeological_operations__Operation')] + ) + self.assertEqual(len(ope_json), 2) + + result_queryset = models.Operation.objects.filter( + code_patriarche="66666") + res = self.generic_serialization_test( + serializers.operation_serialization, no_test=True, + kwargs={"operation_queryset": result_queryset} + ) + ope_json = json.loads( + res[('operations', 'archaeological_operations__Operation')] + ) + self.assertEqual(len(ope_json), 1) + site_json = json.loads( + res[('operations', 'archaeological_operations__ArchaeologicalSite')] + ) + self.assertEqual(len(site_json), 1) + + result_queryset = models.Operation.objects.filter( + code_patriarche="66667") + res = self.generic_serialization_test( + serializers.operation_serialization, no_test=True, + kwargs={"operation_queryset": result_queryset} + ) + ope_json = json.loads( + res[('operations', 'archaeological_operations__Operation')] + ) + self.assertEqual(len(ope_json), 1) + site_json = json.loads( + res[('operations', 'archaeological_operations__ArchaeologicalSite')] + ) + self.assertEqual(len(site_json), 0) def test_conf_restore(self): current_number, zip_filename = self.generic_restore_test_genzip( |