Test administrativ act creation - improve test wizard

3 files changed, 52 insertions, 6 deletions

diff --git a/archaeological_operations/tests.py b/archaeological_operations/tests.py
index 29a7fdb00..5b7f71e1b 100644
--- a/archaeological_operations/tests.py
+++ b/archaeological_operations/tests.py
@@ -648,3 +648,45 @@ class OperationWizardCreationTest(WizardTest, OperationInitTest, TestCase):
     def post_wizard(self):
         self.assertEqual(models.Operation.objects.count(),
                          self.operation_number + 1)
+
+
+class OperationAdminActWizardCreationTest(WizardTest, OperationInitTest,
+                                          TestCase):
+    fixtures = [settings.ROOT_PATH +
+                '../fixtures/initial_data-auth-fr.json',
+                settings.ROOT_PATH +
+                '../ishtar_common/fixtures/initial_data-fr.json',
+                settings.ROOT_PATH +
+                '../archaeological_files/fixtures/initial_data.json',
+                settings.ROOT_PATH +
+                '../archaeological_operations/fixtures/initial_data-fr.json']
+    url_name = 'operation_administrativeactop'
+    wizard_name = 'operation_administrative_act_wizard'
+    steps = views.administrativeactop_steps
+    form_datas = [[
+        # data
+        {
+            'selec-operation_administrativeactop': {
+            },
+            'administrativeact-operation_administrativeactop': {
+                'signature_date': str(datetime.date.today())
+            }
+        },
+        # ignored
+        []
+    ]]
+
+    def pre_wizard(self):
+        ope = self.get_default_operation()
+        self.operation_number = models.Operation.objects.count()
+        data = self.form_datas[0][0]
+        data['selec-operation_administrativeactop']['pk'] = ope.pk
+        act = models.ActType.objects.filter(intented_to='O').all()[0].pk
+
+        data['administrativeact-operation_administrativeactop'][
+            'act_type'] = act
+        super(OperationAdminActWizardCreationTest, self).pre_wizard()
+
+    def post_wizard(self):
+        self.assertEqual(models.Operation.objects.count(),
+                         self.operation_number + 1)
diff --git a/archaeological_operations/views.py b/archaeological_operations/views.py
index 90f71abfe..d9baa4b7a 100644
--- a/archaeological_operations/views.py
+++ b/archaeological_operations/views.py
@@ -384,12 +384,16 @@ operation_administrativeactop_search_wizard = SearchWizard.as_view([
     label=_(u"Administrative act search"),
     url_name='operation_administrativeactop_search',)
 
+administrativeactop_steps = [
+    ('selec-operation_administrativeactop', OperationFormSelection),
+    ('administrativeact-operation_administrativeactop',
+     AdministrativeActOpeForm),
+    ('final-operation_administrativeactop', FinalForm)]
+
+
 operation_administrativeactop_wizard = \
-    OperationAdministrativeActWizard.as_view([
-        ('selec-operation_administrativeactop', OperationFormSelection),
-        ('administrativeact-operation_administrativeactop',
-         AdministrativeActOpeForm),
-        ('final-operation_administrativeactop', FinalForm)],
+    OperationAdministrativeActWizard.as_view(
+        administrativeactop_steps,
         label=_(u"Operation: new administrative act"),
         url_name='operation_administrativeactop',)
 
diff --git a/archaeological_operations/wizards.py b/archaeological_operations/wizards.py
index c51b88d0b..0cc336cc8 100644
--- a/archaeological_operations/wizards.py
+++ b/archaeological_operations/wizards.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
-# Copyright (C) 2012-2015  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
+# Copyright (C) 2012-2016  Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>
 
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as