Basic MongoEngine integration tests

.travis.yml

@@ -5,4 +5,6 @@ python:
 
 install: "pip install -r requirements.txt --use-mirrors"
 
+services: mongodb
+
 script: nosetests

flask_admin/tests/mongoengine/__init__.py

+from flask import Flask
+from flask.ext.admin import Admin
+from flask.ext.mongoengine import MongoEngine
+
+
+def setup():
+    app = Flask(__name__)
+    app.config['SECRET_KEY'] = '1'
+    app.config['CSRF_ENABLED'] = False
+    app.config['MONGODB_SETTINGS'] = {'DB': 'tests'}
+
+    db = MongoEngine()
+    db.init_app(app)
+
+    admin = Admin(app)
+
+    return app, db, admin

flask_admin/tests/mongoengine/test_basic.py

+from nose.tools import eq_, ok_
+
+from flask.ext import wtf
+from flask.ext.admin.contrib.mongoengine import ModelView
+
+from . import setup
+
+
+class CustomModelView(ModelView):
+    def __init__(self, model,
+                 name=None, category=None, endpoint=None, url=None,
+                 **kwargs):
+        for k, v in kwargs.iteritems():
+            setattr(self, k, v)
+
+        super(CustomModelView, self).__init__(model,
+                                              name, category,
+                                              endpoint, url)
+
+
+def create_models(db):
+    class Model1(db.Document):
+        test1 = db.StringField(max_length=20)
+        test2 = db.StringField(max_length=20)
+        test3 = db.StringField()
+        test4 = db.StringField()
+
+    class Model2(db.Document):
+        int_field = db.IntField()
+        bool_field = db.BooleanField()
+
+    Model1.objects.delete()
+    Model2.objects.delete()
+
+    return Model1, Model2
+
+
+def test_model():
+    app, db, admin = setup()
+
+    Model1, Model2 = create_models(db)
+
+    view = CustomModelView(Model1)
+    admin.add_view(view)
+
+    eq_(view.model, Model1)
+    eq_(view.name, 'Model1')
+    eq_(view.endpoint, 'model1view')
+
+    eq_(view._primary_key, 'id')
+
+    ok_('test1' in view._sortable_columns)
+    ok_('test2' in view._sortable_columns)
+    ok_('test3' in view._sortable_columns)
+    ok_('test4' in view._sortable_columns)
+
+    ok_(view._create_form_class is not None)
+    ok_(view._edit_form_class is not None)
+    eq_(view._search_supported, False)
+    eq_(view._filters, None)
+
+    # Verify form
+    eq_(view._create_form_class.test1.field_class, wtf.StringField)
+    eq_(view._create_form_class.test2.field_class, wtf.StringField)
+    eq_(view._create_form_class.test3.field_class, wtf.TextAreaField)
+    eq_(view._create_form_class.test4.field_class, wtf.TextAreaField)
+
+    # Make some test clients
+    client = app.test_client()
+
+    rv = client.get('/admin/model1view/')
+    eq_(rv.status_code, 200)
+
+    rv = client.get('/admin/model1view/new/')
+    eq_(rv.status_code, 200)
+
+    rv = client.post('/admin/model1view/new/',
+                     data=dict(test1='test1large', test2='test2'))
+    eq_(rv.status_code, 302)
+
+    model = Model1.objects.first()
+    eq_(model.test1, 'test1large')
+    eq_(model.test2, 'test2')
+    eq_(model.test3, '')
+    eq_(model.test4, '')
+
+    rv = client.get('/admin/model1view/')
+    eq_(rv.status_code, 200)
+    ok_('test1large' in rv.data)
+
+    url = '/admin/model1view/edit/?id=%s' % model.id
+    rv = client.get(url)
+    eq_(rv.status_code, 200)
+
+    rv = client.post(url,
+                     data=dict(test1='test1small', test2='test2large'))
+    eq_(rv.status_code, 302)
+
+    model = Model1.objects.first()
+    eq_(model.test1, 'test1small')
+    eq_(model.test2, 'test2large')
+    eq_(model.test3, '')
+    eq_(model.test4, '')
+
+    url = '/admin/model1view/delete/?id=%s' % model.id
+    rv = client.post(url)
+    eq_(rv.status_code, 302)
+    eq_(Model1.objects.count(), 0)