adapt mongoengine config change

.env.test.sample

@@ -5,7 +5,7 @@ SITE_NAME=萌翻TEST
 SECRET_KEY=SECRET
 [email protected]
 
-MONGODB_URI="mongomock://moeflow:[email protected]:27017/moeflow_test?authSource=admin"
+MONGODB_URI="mongodb://moeflow:[email protected]:27017/moeflow_test?authSource=admin"
 # MONGODB_DB_NAME=moeflow
 # MONGODB_USER=moeflow
 # MONGODB_PASS=CHANGE_ME

app/models/__init__.py

@@ -13,6 +13,11 @@ def connect_db(config):
     logger.info("Connect mongodb")
     uri = config["DB_URI"]
     logger.debug(" - $DB_URI: {}".format(uri))
+    if config.get("TESTING"):
+        import mongomock
+
+        return connect(host=uri, mongo_client_class=mongomock.MongoClient)
+
     return connect(host=uri)