1 files changed, 48 insertions, 16 deletions
diff --git a/archaeological_warehouse/tests.py b/archaeological_warehouse/tests.py
index 938d1cbc0..850af8218 100644
--- a/archaeological_warehouse/tests.py
+++ b/archaeological_warehouse/tests.py
@@ -26,9 +26,7 @@ from ishtar_common.tests import WizardTest, WizardTestFormData as FormData
 
 from archaeological_warehouse import models, views, forms
 
-
-class ContainerWizardCreationTest(WizardTest, FindInit, TestCase):
-    fixtures = [settings.ROOT_PATH +
+warehouse_fixtures = [settings.ROOT_PATH +
                 '../fixtures/initial_data.json',
                 settings.ROOT_PATH +
                 '../ishtar_common/fixtures/initial_data.json',
@@ -41,6 +39,52 @@ class ContainerWizardCreationTest(WizardTest, FindInit, TestCase):
                 settings.ROOT_PATH +
                 '../archaeological_warehouse/fixtures/initial_data-fr.json',
                 ]
+
+
+class WarehouseWizardCreationTest(WizardTest, FindInit, TestCase):
+    fixtures = warehouse_fixtures
+    url_name = 'warehouse_creation'
+    wizard_name = 'warehouse_wizard'
+    steps = views.warehouse_creation_steps
+    form_datas = [
+        FormData(
+            'Warehouse creation',
+            form_datas={
+                'warehouse-warehouse_creation': {
+                    'name': 'warehouse-ref',
+                    'warehouse_type': None,
+                    'location': None,
+                    'responsible': None,
+                },
+                'divisions-warehouse_creation': [
+                    {
+                        'division': None,
+                        'order': 42
+                    }
+                ]
+            },
+        ),
+    ]
+
+    def pre_wizard(self):
+        main_data = self.form_datas[0].form_datas
+        main_data['warehouse-warehouse_creation']['warehouse_type'] = \
+            models.WarehouseType.objects.all()[0].pk
+        main_data['divisions-warehouse_creation'][0]['division'] = \
+            models.WarehouseDivision.create_default_for_test()[0].pk
+        self.warehouse_number = models.Warehouse.objects.count()
+        self.warehouse_div_link = models.WarehouseDivisionLink.objects.count()
+        super(WarehouseWizardCreationTest, self).pre_wizard()
+
+    def post_wizard(self):
+        self.assertEqual(models.Warehouse.objects.count(),
+                         self.warehouse_number + 1)
+        self.assertEqual(models.WarehouseDivisionLink.objects.count(),
+                         self.warehouse_div_link + 1)
+
+
+class ContainerWizardCreationTest(WizardTest, FindInit, TestCase):
+    fixtures = warehouse_fixtures
     url_name = 'container_creation'
     wizard_name = 'container_wizard'
     steps = views.container_creation_steps
@@ -122,19 +166,7 @@ class ContainerWizardCreationTest(WizardTest, FindInit, TestCase):
 
 
 class ContainerTest(FindInit, TestCase):
-    fixtures = [settings.ROOT_PATH +
-                '../fixtures/initial_data.json',
-                settings.ROOT_PATH +
-                '../ishtar_common/fixtures/initial_data.json',
-                settings.ROOT_PATH +
-                '../archaeological_files/fixtures/initial_data.json',
-                settings.ROOT_PATH +
-                '../archaeological_operations/fixtures/initial_data-fr.json',
-                settings.ROOT_PATH +
-                '../archaeological_finds/fixtures/initial_data-fr.json',
-                settings.ROOT_PATH +
-                '../archaeological_warehouse/fixtures/initial_data-fr.json',
-                ]
+    fixtures = warehouse_fixtures
 
     def testFormCreation(self):
         main_warehouse = models.Warehouse.objects.create(