From 3e732c0e39399402643e44fadc681936e04a3354 Mon Sep 17 00:00:00 2001
From: Iacopo Spalletti <i.spalletti@nephila.it>
Date: Sun, 25 Oct 2015 18:14:19 +0100
Subject: [PATCH] More tests

---
 djangocms_blog/cms_wizards.py |  6 ++++--
 tests/test_wizards.py         | 26 ++++++++++++++++++++++++++
 2 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/djangocms_blog/cms_wizards.py b/djangocms_blog/cms_wizards.py
index 0eded57..3909075 100644
--- a/djangocms_blog/cms_wizards.py
+++ b/djangocms_blog/cms_wizards.py
@@ -15,8 +15,9 @@ try:
         default_appconfig = None
 
         def __init__(self, *args, **kwargs):
-            kwargs['initial']['app_config'] = self.default_appconfig
-            if 'data' in kwargs:
+            if kwargs.get('initial', False):
+                kwargs['initial']['app_config'] = self.default_appconfig
+            if kwargs.get('data', False):
                 kwargs['data']['1-app_config'] = self.default_appconfig
             super(PostWizardForm, self).__init__(*args, **kwargs)
             self.fields['app_config'].widget.attrs['disabled'] = True
@@ -37,6 +38,7 @@ try:
             title=_('New {0}').format(config.object_name),
             weight=200,
             form=new_form,
+            model=Post,
             description=_('Create a new {0} in {1}').format(config.object_name, config.app_title),
         )
         wizard_pool.register(post_wizard)
diff --git a/tests/test_wizards.py b/tests/test_wizards.py
index 5be4e25..4937b1c 100644
--- a/tests/test_wizards.py
+++ b/tests/test_wizards.py
@@ -28,3 +28,29 @@ class WizardTest(BaseTest):
         titles = [entry.title for entry in wizard_pool.get_entries()]
         self.assertTrue('New Blog' in titles)
         self.assertTrue('New Article' in titles)
+
+    @skipIf(LooseVersion(cms.__version__) < LooseVersion('3.2'),
+            reason='Wizards not available for django CMS < 3.2')
+    def test_wizard_init(self):
+        from cms.wizards.wizard_pool import wizard_pool
+        from djangocms_blog.models import Post
+        self.get_pages()
+        wizard_pool._discover()
+
+        wizs = [entry for entry in wizard_pool.get_entries() if entry.model == Post]
+        for index, wiz in enumerate(wizs):
+            app_config = self.app_config_1.pk if index == 0 else self.app_config_2.pk
+            form = wiz.form(initial={'app_config': app_config})
+            self.assertTrue(form.fields['app_config'].widget.attrs['disabled'])
+
+            form = wiz.form(data={
+                '1-title': 'title',
+                '1-abstract': 'abstract',
+                '1-categories': [self.category_1.pk],
+            }, prefix=1)
+            self.assertEqual(form.default_appconfig, app_config)
+            self.assertTrue(form.is_valid())
+
+    def test_wizard_import(self):
+        # The following import should to fail in any django CMS version
+        from djangocms_blog import cms_wizards  # NOQA