add tests for new create_modal option

flask_admin/tests/test_model.py

@@ -457,22 +457,41 @@ def test_modal_edit():
     app_bs2 = Flask(__name__)
     admin_bs2 = Admin(app_bs2, template_mode="bootstrap2")
 
-    modal_view = MockModelView(Model, edit_modal=True, endpoint="modal_on")
-    no_modal_view = MockModelView(Model, edit_modal=False, endpoint="modal_off")
-
-    admin_bs2.add_view(modal_view)
-    admin_bs2.add_view(no_modal_view)
+    edit_modal_on = MockModelView(Model, edit_modal=True,
+                                  endpoint="edit_modal_on")
+    edit_modal_off = MockModelView(Model, edit_modal=False,
+                                   endpoint="edit_modal_off")
+    create_modal_on = MockModelView(Model, create_modal=True,
+                                    endpoint="create_modal_on")
+    create_modal_off = MockModelView(Model, create_modal=False,
+                                     endpoint="create_modal_off")
+    admin_bs2.add_view(edit_modal_on)
+    admin_bs2.add_view(edit_modal_off)
+    admin_bs2.add_view(create_modal_on)
+    admin_bs2.add_view(create_modal_off)
 
     client_bs2 = app_bs2.test_client()
 
     # bootstrap 2 - ensure modal window is added when edit_modal is enabled
-    rv = client_bs2.get('/admin/modal_on/')
+    rv = client_bs2.get('/admin/edit_modal_on/')
+    eq_(rv.status_code, 200)
+    data = rv.data.decode('utf-8')
+    ok_('fa_modal_window' in data)
+
+    # bootstrap 2 - test edit modal disabled
+    rv = client_bs2.get('/admin/edit_modal_off/')
+    eq_(rv.status_code, 200)
+    data = rv.data.decode('utf-8')
+    ok_('fa_modal_window' not in data)
+
+    # bootstrap 2 - ensure modal window is added when create_modal is enabled
+    rv = client_bs2.get('/admin/create_modal_on/')
     eq_(rv.status_code, 200)
     data = rv.data.decode('utf-8')
     ok_('fa_modal_window' in data)
 
-    # bootstrap 2 - test modal disabled
-    rv = client_bs2.get('/admin/modal_off/')
+    # bootstrap 2 - test create modal disabled
+    rv = client_bs2.get('/admin/create_modal_off/')
     eq_(rv.status_code, 200)
     data = rv.data.decode('utf-8')
     ok_('fa_modal_window' not in data)
@@ -481,19 +500,33 @@ def test_modal_edit():
     app_bs3 = Flask(__name__)
     admin_bs3 = Admin(app_bs3, template_mode="bootstrap3")
 
-    admin_bs3.add_view(modal_view)
-    admin_bs3.add_view(no_modal_view)
+    admin_bs3.add_view(edit_modal_on)
+    admin_bs3.add_view(edit_modal_off)
+    admin_bs3.add_view(create_modal_on)
+    admin_bs3.add_view(create_modal_off)
 
     client_bs3 = app_bs3.test_client()
 
     # bootstrap 3 - ensure modal window is added when edit_modal is enabled
-    rv = client_bs3.get('/admin/modal_on/')
+    rv = client_bs3.get('/admin/edit_modal_on/')
+    eq_(rv.status_code, 200)
+    data = rv.data.decode('utf-8')
+    ok_('fa_modal_window' in data)
+
+    # bootstrap 3 - test modal disabled
+    rv = client_bs3.get('/admin/edit_modal_off/')
+    eq_(rv.status_code, 200)
+    data = rv.data.decode('utf-8')
+    ok_('fa_modal_window' not in data)
+
+    # bootstrap 3 - ensure modal window is added when edit_modal is enabled
+    rv = client_bs3.get('/admin/create_modal_on/')
     eq_(rv.status_code, 200)
     data = rv.data.decode('utf-8')
     ok_('fa_modal_window' in data)
 
     # bootstrap 3 - test modal disabled
-    rv = client_bs3.get('/admin/modal_off/')
+    rv = client_bs3.get('/admin/create_modal_off/')
     eq_(rv.status_code, 200)
     data = rv.data.decode('utf-8')
     ok_('fa_modal_window' not in data)