Exemplo n.º 1
0
    def test_connection_default(self):
        self.app.config['MONGODB_SETTINGS'] = {}
        self.app.config['TESTING'] = True

        db = MongoEngine()
        # Disconnect to drop connection from setup.
        db.disconnect()
        db.init_app(self.app)
def db_connect(app):

    mongo = MongoEngine()

    if not app.config["TESTING"]:
        # allow fork() calls by pymongo while the
        # wsgi process executes this flask app
        if "UWSGI_LOADED" in environ:
            from uwsgidecorators import postfork

            @postfork
            def setup_db():
                mongo.init_app(app)

            return

    else:
        logging.info(app.config["MONGODB_DB"], app.config["MONGODB_HOST"])
        mongo.disconnect()

    mongo.init_app(app)