From 3139e851539ac69b45cd734be3bae34b49c90c04 Mon Sep 17 00:00:00 2001
From: Lucas Simeon <lsimeon@wanadev.fr>
Date: Fri, 3 Mar 2017 17:11:03 +0100
Subject: [PATCH] [added] refresh portalClassName on componentWillUpdate

---
 lib/components/Modal.js |  6 ++++++
 specs/Modal.spec.js     | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 44 insertions(+)

diff --git a/lib/components/Modal.js b/lib/components/Modal.js
index e41c8b4f..dd56b6fb 100644
--- a/lib/components/Modal.js
+++ b/lib/components/Modal.js
@@ -73,6 +73,12 @@ var Modal = createReactClass({
     this.renderPortal(this.props);
   },
 
+  componentWillUpdate: function(newProps) {
+    if(newProps.portalClassName !== this.props.portalClassName) {
+      this.node.className = newProps.portalClassName;
+    }
+  },
+
   componentWillReceiveProps: function(newProps) {
     if (newProps.isOpen) refCount.add(this);
     if (!newProps.isOpen) refCount.remove(this);
diff --git a/specs/Modal.spec.js b/specs/Modal.spec.js
index 176a2711..620df920 100644
--- a/specs/Modal.spec.js
+++ b/specs/Modal.spec.js
@@ -341,4 +341,42 @@ describe('State', () => {
 
     document.body.removeChild(currentDiv);
   });
+
+  it('verify that portalClassName is refreshed on component update', () => {
+    var node = document.createElement('div');
+    var modal = null;
+
+    var App = React.createClass({
+      getInitialState: function () {
+        return { testHasChanged: false };
+      },
+
+      componentDidMount: function() {
+        expect(modal.node.className).toEqual('myPortalClass');
+
+        this.setState({
+          testHasChanged: true
+        });
+      },
+
+      componentDidUpdate: function() {
+        expect(modal.node.className).toEqual('myPortalClass-modifier');
+      },
+
+      render: function() {
+        return (
+          <div>
+            <Modal
+              ref={modalComponent => { modal = modalComponent; }}
+              isOpen={true}
+              portalClassName={this.state.testHasChanged === true ? 'myPortalClass-modifier' : 'myPortalClass'}>
+            </Modal>
+          </div>
+        );
+      }
+    });
+
+    Modal.setAppElement(node);
+    ReactDOM.render(<App />, node);
+  });
 });