From b1ac2598c0601547d226e64e8beac804539f0428 Mon Sep 17 00:00:00 2001
From: Matt Wright <matt@nobien.net>
Date: Wed, 16 Oct 2013 10:55:56 -0400
Subject: [PATCH] Be sure to save user model when using `add_role_to_user`
 method for MongoEngineUserDatastore. Fixes #170

---
 flask_security/datastore.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/flask_security/datastore.py b/flask_security/datastore.py
index b454dc8..4716931 100644
--- a/flask_security/datastore.py
+++ b/flask_security/datastore.py
@@ -234,6 +234,12 @@ class MongoEngineUserDatastore(MongoEngineDatastore, UserDatastore):
     def find_role(self, role):
         return self.role_model.objects(name=role).first()
 
+    def add_role_to_user(self, user, role):
+        rv = super(MongoEngineUserDatastore, self).add_role_to_user(user, role)
+        if rv:
+            self.put(user)
+        return rv
+
 
 class PeeweeUserDatastore(PeeweeDatastore, UserDatastore):
     """A PeeweeD datastore implementation for Flask-Security that assumes