diff options
Diffstat (limited to 'archaeological_operations/tests.py')
| -rw-r--r-- | archaeological_operations/tests.py | 45 |
1 files changed, 41 insertions, 4 deletions
diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py index 33199dd04..9a61d8bbb 100644 --- a/archaeological_operations/tests.py +++ b/archaeological_operations/tests.py @@ -705,22 +705,59 @@ class OperationWizardCreationTest(WizardTest, OperationInitTest, TestCase): form_datas={ 'general-operation_creation': { 'operation_type': 1, # preventive diag - 'year': 2016} + 'year': 2016}, + 'townsgeneral-operation_creation': [], + 'parcelsgeneral-operation_creation': [], }, ignored=('filechoice-operation_creation', - 'preventive-operation_creation', 'towns-operation_creation', - 'parcels-operation_creation', ) + 'parcels-operation_creation', + 'preventive-operation_creation',) + ), + FormData( + "Create another preventive diag with same parcel name", + form_datas={ + 'general-operation_creation': { + 'operation_type': 1, # preventive diag + 'year': 2016}, + 'townsgeneral-operation_creation': [], + 'parcelsgeneral-operation_creation': [], + }, + ignored=('filechoice-operation_creation', + 'towns-operation_creation', + 'parcels-operation_creation', + 'preventive-operation_creation',) ) ] def pre_wizard(self): + if 'townsgeneral-operation_creation' not in \ + self.form_datas[0].form_datas: + return super(OperationWizardCreationTest, self).pre_wizard() + town = self.create_towns()[0] + town_data = {'town': town.pk} + self.form_datas[0].form_datas['townsgeneral-operation_creation'].append( + town_data + ) + self.form_datas[1].form_datas['townsgeneral-operation_creation'].append( + town_data + ) + parcel_data = { + 'town': town.pk, 'year': 2017, 'section': 'S', + 'parcel_number': '42'} + self.form_datas[0].form_datas[ + 'parcelsgeneral-operation_creation'].append(parcel_data) + self.form_datas[1].form_datas[ + 'parcelsgeneral-operation_creation'].append(parcel_data) self.operation_number = models.Operation.objects.count() + self.parcel_number = models.Parcel.objects.count() super(OperationWizardCreationTest, self).pre_wizard() def post_wizard(self): self.assertEqual(models.Operation.objects.count(), - self.operation_number + 1) + self.operation_number + 2) + self.assertEqual(models.Parcel.objects.count(), + self.parcel_number + 2) class OperationWizardDeleteTest(OperationWizardCreationTest): |