Rehydrate based on resolution keys

ericclemmons · Apr 22, 2015 · ae54d7c · ae54d7c
1 parent 5f0efec
commit ae54d7c
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 7 deletions.
diff --git a/src/Resolver.js b/src/Resolver.js
@@ -47,11 +47,7 @@ export default class Resolver {
       throw new ReferenceError(`${container.constructor.displayName} should have an ID`);
     }
 
-    const state = this.states.hasOwnProperty(id) ? this.states[id] : {
-      fulfilled: false,
-      rejected: false,
-      values: {},
-    };
+    const state = this.states.hasOwnProperty(id) ? this.states[id] : this.rehydrate(id);
 
     if (!this.states.hasOwnProperty(id)) {
       this.states[id] = state;
@@ -72,6 +68,17 @@ export default class Resolver {
     throw new Error(`${this.constructor.displayName} was rejected: ${error}`);
   }
 
+  rehydrate(id) {
+    if (typeof __resolver__ === 'undefined') {
+      return {
+        fulfilled: false,
+        rejected: false,
+        values: {},
+      };
+    }
+    return __resolver__[id];
+  }
+
   resolve(container, callback) {
     const asyncProps = container.props.resolve || {};
     const state = this.getContainerState(container);

diff --git a/test/Container.element.test.js b/test/Container.element.test.js
@@ -100,7 +100,15 @@ describe("<Container />", function() {
 
       context("when keys are rehydrating", function() {
         before(function() {
-          global.__resolver__ = { user: "Exists" };
+          global.__resolver__ = {
+            '.0': {
+              values: {
+                fulfilled: false,
+                rejected: false,
+                user: "Exists"
+              }
+            }
+          };
         });
 
         after(function() {
@@ -130,7 +138,7 @@ describe("<Container />", function() {
             />
           );
 
-          assert.equal(actual, `<code>${JSON.stringify(global.__resolver__)}</code>`);
+          assert.equal(actual, `<code>${JSON.stringify(global.__resolver__['.0'].values)}</code>`);
         });
       });
     });