diff options
| -rw-r--r-- | archaeological_operations/tests.py | 41 |
1 files changed, 39 insertions, 2 deletions
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py index 3e172fc88..67f4484e2 100644 --- a/archaeological_operations/tests.py +++ b/archaeological_operations/tests.py @@ -2692,7 +2692,7 @@ class OperationWizardCreationTest(WizardTest, OperationInitTest, TestCase): 'town': town.pk, 'year': 2017, 'section': 'S', 'parcel_number': '42'} - for idx in range(0, 2): + for idx in range(2): self.form_datas[idx].append('townsgeneral', town_data) self.form_datas[idx].append('parcelsgeneral', parcel_data) @@ -2805,6 +2805,18 @@ class OperationWizardModifTest(WizardTest, OperationInitTest, TestCase): ignored=base_ignored_steps, error_expected='general' ), + FormData( + "Operation: add a parcel", + form_datas={ + 'selec': {}, + 'general': { + 'operation_type': 2, + 'year': 2017}, + 'townsgeneral': [], + 'parcelsgeneral': [], + }, + ignored=base_ignored_steps + ), ] def pre_wizard(self): @@ -2834,16 +2846,19 @@ class OperationWizardModifTest(WizardTest, OperationInitTest, TestCase): self.form_datas[1].set('general', 'operation_type', self.ope_type.pk) self.form_datas[2].set('general', 'operation_type', self.ope_type.pk) self.form_datas[3].set('general', 'operation_type', self.ope_type.pk) + self.form_datas[4].set('general', 'operation_type', self.ope_type.pk) data = self.form_datas[0].form_datas data2 = self.form_datas[1].form_datas data3 = self.form_datas[2].form_datas data4 = self.form_datas[3].form_datas + data5 = self.form_datas[4].form_datas data['selec']['pk'] = operation.pk data2['selec']['pk'] = operation.pk data3['selec']['pk'] = operation.pk data4['selec']['pk'] = operation.pk + data5['selec']['pk'] = operation.pk town = self.create_towns( datas={'numero_insee': '67890', 'name': 'Twin Peaks'})[-1] @@ -2852,11 +2867,15 @@ class OperationWizardModifTest(WizardTest, OperationInitTest, TestCase): 'town': town.pk, 'year': 2017, 'section': 'S', 'parcel_number': '42'} - for idx in range(0, 4): + for idx in range(0, 5): for t in towns: self.form_datas[idx].append('townsgeneral', t) self.form_datas[idx].append('parcelsgeneral', parcel_data) + p = parcel_data.copy() + p['parcel_number'] = '43' + self.form_datas[4].append('parcelsgeneral', p) + parcel_data_2 = { 'town': init_parcel.town.pk, 'year': init_parcel.year or '', 'section': init_parcel.section, @@ -2920,10 +2939,28 @@ class OperationWizardModifTest(WizardTest, OperationInitTest, TestCase): test_object.assertEqual(models.Parcel.objects.count(), test_object.parcel_number + 1) + def pre_fifth_wizard(test_object): + test_object.parcel_number = models.Parcel.objects.count() + operation = models.Operation.objects.get( + pk=test_object.operations[0].pk) + test_object.operation_parcel_number = operation.parcels.count() + + def post_fifth_wizard(test_object, final_step_response): + test_object.assertEqual(models.Operation.objects.count(), + test_object.operation_number) + operation = models.Operation.objects.get( + pk=test_object.operations[0].pk) + test_object.assertEqual(models.Parcel.objects.count(), + test_object.parcel_number + 1) + test_object.assertEqual(operation.parcels.count(), + test_object.operation_parcel_number + 1) + self.form_datas[0].extra_tests = [post_first_wizard] self.form_datas[1].extra_tests = [post_second_wizard] self.form_datas[2].pre_tests = [pre_third_wizard] self.form_datas[2].extra_tests = [post_third_wizard] + self.form_datas[4].pre_tests = [pre_fifth_wizard] + self.form_datas[4].extra_tests = [post_fifth_wizard] super(OperationWizardModifTest, self).pre_wizard() |