Merge branch 'master' into develop

3 files changed, 38 insertions, 4 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index 3aa0f6ea0..75f580633 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -26,7 +26,7 @@ from django.core.files.uploadedfile import SimpleUploadedFile
 from django.core.urlresolvers import reverse
 from django.test.client import Client
 from ishtar_common.models import ImporterType, IshtarUser, ImporterColumn,\
-    FormaterType, ImportTarget
+    FormaterType, ImportTarget, IshtarSiteProfile
 
 from ishtar_common.models import Person, get_current_profile
 from archaeological_context_records.models import Period, Dating
@@ -95,7 +95,6 @@ class FindInit(ContextRecordInit):
 
 
 class FindWizardCreationTest(WizardTest, FindInit, TestCase):
-    # TODO: first to be run because of strange init things...
     fixtures = [settings.ROOT_PATH +
                 '../fixtures/initial_data.json',
                 settings.ROOT_PATH +
@@ -154,6 +153,37 @@ class FindWizardCreationTest(WizardTest, FindInit, TestCase):
         self.assertEqual(find.datings.count(), 1)
 
 
+class FindWizardDeletionWithWarehouseModTest(WizardTest, FindInit, TestCase):
+    fixtures = FindWizardCreationTest.fixtures
+    url_name = 'find_deletion'
+    wizard_name = 'find_deletion_wizard'
+    steps = views.find_deletion_steps
+    form_datas = [
+        FormData(
+            'Find deletion',
+            form_datas={
+                'selecw': {},
+            },
+            ignored=['selec-find_deletion']
+        )
+    ]
+
+    def pre_wizard(self):
+        profile, created = IshtarSiteProfile.objects.get_or_create(
+            slug='default', active=True)
+        profile.warehouse = True
+        profile.save()
+
+        find, bf = self.get_default_find(force=True)
+        self.form_datas[0].set('selecw', 'pk', find.pk)
+        self.find_number = models.Find.objects.count()
+        super(FindWizardDeletionWithWarehouseModTest, self).pre_wizard()
+
+    def post_wizard(self):
+        self.assertEqual(models.Find.objects.count(),
+                         self.find_number - 1)
+
+
 class TreatmentWizardCreationTest(WizardTest, FindInit, TestCase):
     fixtures = [settings.ROOT_PATH +
                 '../fixtures/initial_data.json',
diff --git a/archaeological_finds/views.py b/archaeological_finds/views.py
index 681296b3c..4874abefb 100644
--- a/archaeological_finds/views.py
+++ b/archaeological_finds/views.py
@@ -181,10 +181,13 @@ find_deletion_condition_dict = {
     'selecw-find_deletion': check_warehouse_module,
 }
 
-find_deletion_wizard = FindDeletionWizard.as_view([
+find_deletion_steps = [
     ('selec-find_deletion', FindFormSelection),
     ('selecw-find_deletion', FindFormSelectionWarehouseModule),
-    ('final-find_deletion', FindDeletionForm)],
+    ('final-find_deletion', FindDeletionForm)]
+
+find_deletion_wizard = FindDeletionWizard.as_view(
+    find_deletion_steps,
     condition_dict=find_deletion_condition_dict,
     label=_(u"Find deletion"),
     url_name='find_deletion',)
diff --git a/archaeological_finds/wizards.py b/archaeological_finds/wizards.py
index 43fe6262f..b4471a80b 100644
--- a/archaeological_finds/wizards.py
+++ b/archaeological_finds/wizards.py
@@ -83,6 +83,7 @@ class FindModificationWizard(FindWizard):
 
 class FindDeletionWizard(DeletionWizard):
     model = models.Find
+    main_item_select_keys = ('selec-', 'selecw-')
     fields = ['label', 'material_types', 'datings', 'find_number',
               'object_types', 'description', 'conservatory_state', 'mark',
               'preservation_to_considers', 'integrities', 'remarkabilities',