Update the current user on disk when saved or fetched

Author: andrewimm

src/ParseUser.js

@@ -401,6 +401,32 @@ export default class ParseUser extends ParseObject {
     return controller.logIn(this, loginOptions)._thenRunCallbacks(options, this);
   }
 
+  /**
+   * Wrap the default save behavior with functionality to save to local
+   * storage if this is current user.
+   */
+  save(...args) {
+    return super.save.apply(this, args).then(() => {
+      if (this.isCurrent()) {
+        return CoreManager.getUserController().updateUserOnDisk(this);
+      }
+      return this;
+    });
+  }
+
+  /**
+   * Wrap the default fetch behavior with functionality to save to local
+   * storage if this is current user.
+   */
+  fetch(...args) {
+    return super.fetch.apply(this, args).then(() => {
+      if (this.isCurrent()) {
+        return CoreManager.getUserController().updateUserOnDisk(this);
+      }
+      return this;
+    });
+  }
+
   static readOnlyAttributes() {
     return ['sessionToken'];
   }
@@ -692,10 +718,7 @@ export default class ParseUser extends ParseObject {
 ParseObject.registerSubclass('_User', ParseUser);
 
 var DefaultController = {
-  setCurrentUser(user) {
-    currentUserCache = user;
-    user._cleanupAuthData();
-    user._synchronizeAllAuthData();
+  updateUserOnDisk(user) {
     var path = Storage.generatePath(CURRENT_USER_KEY);
     var json = user.toJSON();
     json.className = '_User';
@@ -706,6 +729,13 @@ var DefaultController = {
     });
   },
 
+  setCurrentUser(user) {
+    currentUserCache = user;
+    user._cleanupAuthData();
+    user._synchronizeAllAuthData();
+    return DefaultController.updateUserOnDisk(user);
+  },
+
   currentUser(): ?ParseUser {
     if (currentUserCache) {
       return currentUserCache;

src/__tests__/ParseUser-test.js

@@ -359,6 +359,80 @@ describe('ParseUser', () => {
     });
   }));
 
+  it('updates the current user on disk when saved', asyncHelper((done) => {
+    ParseUser.enableUnsafeCurrentUser();
+    ParseUser._clearCache();
+    Storage._clear();
+    CoreManager.setRESTController({
+      request() {
+        return ParsePromise.as({
+          objectId: 'uid5',
+        }, 201);
+      },
+      ajax() {}
+    });
+
+    ParseUser.signUp('updater', 'password').then((u) => {
+      expect(u.isCurrent()).toBe(true);
+      ParseUser._clearCache();
+      CoreManager.setRESTController({
+        request() {
+          return ParsePromise.as({}, 200);
+        },
+        ajax() {}
+      });
+      return u.save({ count: 12 });
+    }).then((u) => {
+      ParseUser._clearCache();
+      ParseObject._clearAllState();
+      expect(u.attributes).toEqual({});
+      expect(u.get('count')).toBe(undefined);
+      return ParseUser.currentAsync();
+    }).then((current) => {
+      expect(current.id).toBe('uid5');
+      expect(current.get('count')).toBe(12);
+      done();
+    });
+  }));
+
+  it('updates the current user on disk when fetched', asyncHelper((done) => {
+    ParseUser.enableUnsafeCurrentUser();
+    ParseUser._clearCache();
+    Storage._clear();
+    CoreManager.setRESTController({
+      request() {
+        return ParsePromise.as({
+          objectId: 'uid6'
+        }, 200);
+      },
+      ajax() {}
+    });
+
+    ParseUser.signUp('spot', 'fetch').then((u) => {
+      expect(u.isCurrent()).toBe(true);
+      ParseUser._clearCache();
+      CoreManager.setRESTController({
+        request() {
+          return ParsePromise.as({
+            count: 15
+          }, 200);
+        },
+        ajax() {}
+      });
+      return u.fetch();
+    }).then((u) => {
+      ParseUser._clearCache();
+      ParseObject._clearAllState();
+      expect(u.attributes).toEqual({});
+      expect(u.get('count')).toBe(undefined);
+      return ParseUser.currentAsync();
+    }).then((current) => {
+      expect(current.id).toBe('uid6');
+      expect(current.get('count')).toBe(15);
+      done();
+    });
+  }));
+
   it('can get error when recursive _linkWith call fails', asyncHelper((done) => {
     CoreManager.setRESTController({
       request(method, path, body, options) {