Update root container attributes on client transition with reuseDom

Fixes #73.
redfin · May 25, 2016 · 9ea54a8 · 9ea54a8
1 parent 2ad4497
commit 9ea54a8
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/react-server/core/ClientController.js b/packages/react-server/core/ClientController.js
@@ -577,6 +577,10 @@ class ClientController extends EventEmitter {
 				//
 				if (this._reuseDom && element.containerOpen && oldRootContainer) {
 					mountNode = oldRootContainer;
+					this._updateContainerNodeAttributes(
+						mountNode,
+						element.containerOpen
+					);
 				} else if (this._reuseDom && element.containerClose && !oldRootContainer && !oldRootElement) {
 					mountNode = mountNode.parentNode;
 				} else if (this._reuseDom && oldRootElement) {
@@ -737,6 +741,10 @@ class ClientController extends EventEmitter {
 		return node;
 	}
 
+	_updateContainerNodeAttributes(node, attrs) {
+		_.forEach(attrs, (v, k) => node.setAttribute(k, v));
+	}
+
 	init () {
 
 		var unloadHandler = () => {this.terminate(); };