[added]: pass overlay and content element references to onAfterOpen fn (

reactjs#741)
jaksenko · Oct 8, 2019 · 4ac3ff4 · 4ac3ff4
1 parent 9be00a9
commit 4ac3ff4
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/specs/Modal.events.spec.js b/specs/Modal.events.spec.js
@@ -25,6 +25,16 @@ export default () => {
     afterOpenCallback.called.should.be.ok();
   });
 
+  it("should call onAfterOpen with overlay and content references", () => {
+    const afterOpenCallback = sinon.spy();
+    const modal = renderModal({ isOpen: true, onAfterOpen: afterOpenCallback });
+
+    sinon.assert.calledWith(afterOpenCallback, {
+      overlayEl: modal.portal.overlay,
+      contentEl: modal.portal.content
+    });
+  });
+
   it("should trigger the onAfterClose callback", () => {
     const onAfterCloseCallback = sinon.spy();
     const modal = renderModal({

diff --git a/src/components/ModalPortal.js b/src/components/ModalPortal.js
@@ -208,7 +208,10 @@ export default class ModalPortal extends Component {
         this.setState({ afterOpen: true });
 
         if (this.props.isOpen && this.props.onAfterOpen) {
-          this.props.onAfterOpen();
+          this.props.onAfterOpen({
+            overlayEl: this.overlay,
+            contentEl: this.content
+          });
         }
       });
     }