| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081 |
- from nose.tools import eq_, ok_
- from wtforms import form, fields
- from flask_admin.contrib.pymongo import ModelView
- from . import setup
- class TestForm(form.Form):
- test1 = fields.StringField('Test1')
- test2 = fields.StringField('Test2')
- class TestView(ModelView):
- column_list = ('test1', 'test2', 'test3', 'test4')
- column_sortable_list = ('test1', 'test2')
- form = TestForm
- def test_model():
- app, db, admin = setup()
- view = TestView(db.test, 'Test')
- admin.add_view(view)
- # Drop existing data (if any)
- db.test.remove()
- eq_(view.name, 'Test')
- eq_(view.endpoint, 'testview')
- ok_('test1' in view._sortable_columns)
- ok_('test2' 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)
- # Make some test clients
- client = app.test_client()
- rv = client.get('/admin/testview/')
- eq_(rv.status_code, 200)
- rv = client.get('/admin/testview/new/')
- eq_(rv.status_code, 200)
- rv = client.post('/admin/testview/new/',
- data=dict(test1='test1large', test2='test2'))
- eq_(rv.status_code, 302)
- model = db.test.find()[0]
- print(model)
- eq_(model['test1'], 'test1large')
- eq_(model['test2'], 'test2')
- rv = client.get('/admin/testview/')
- eq_(rv.status_code, 200)
- ok_('test1large' in rv.data.decode('utf-8'))
- url = '/admin/testview/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)
- print(db.test.find()[0])
- model = db.test.find()[0]
- eq_(model['test1'], 'test1small')
- eq_(model['test2'], 'test2large')
- url = '/admin/testview/delete/?id=%s' % model['_id']
- rv = client.post(url)
- eq_(rv.status_code, 302)
- eq_(db.test.count(), 0)
|
