import cProfile import flask from flaskext.mongoobject import MongoObject db = MongoObject() app = flask.Flask(__name__) TESTING = True class SomeModel(db.Model): __collection__ = "tests" app.config['MONGODB_HOST'] = "mongodb://localhost:27017" app.config['MONGODB_DATABASE'] = "testdb" app.config['MONGODB_AUTOREF'] = False app.config['AUTOINCREMENT'] = False app.config['TESTING'] = True db.init_app(app) def create_model(interval): for i in interval: model = SomeModel({"test": {"name": "testing_{}".format(i)}}) model.save() def find_model(interval): for i in interval: SomeModel.query.find({"test.name": "testing_{}".format(i)})
from attest import Tests, assert_hook import flask from flaskext.attest import request_context from flaskext.mongoobject import AttrDict, MongoObject db = MongoObject() app = flask.Flask(__name__) TESTING = True class TestModel(db.Model): __collection__ = "tests" db.set_mapper(TestModel) @request_context def setup_app(): app.config['MONGODB_HOST'] = "mongodb://localhost:27017" app.config['MONGODB_DATABASE'] = "testdb" app.config['MONGODB_AUTOREF'] = True app.config['TESTING'] = True yield app mongounit = Tests() mongointegration = Tests(contexts=[setup_app]) flask_mongoobject = Tests([mongounit, mongointegration])