Expose some internals for easier testing

react-dnd · May 18, 2015 · 19f6740 · 19f6740
1 parent c8ad695
commit 19f6740
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 6 deletions.
diff --git a/src/DragDropContext.js b/src/DragDropContext.js
@@ -26,18 +26,29 @@ export default function DragDropContext(backend) {
       'Component';
 
     return class DragDropContextContainer extends Component {
+      static DecoratedComponent = DecoratedComponent;
+
       static displayName = `DragDropContext(${displayName})`;
 
       static childContextTypes = {
         dragDropManager: PropTypes.object.isRequired
       };
 
+      getDecoratedComponentInstance() {
+        return this.refs.child;
+      }
+
+      getManager() {
+        return childContext.dragDropManager;
+      }
+
       getChildContext() {
         return childContext;
       }
 
       render() {
-        return <DecoratedComponent {...this.props} />;
+        return <DecoratedComponent {...this.props}
+                                   ref='child' />;
       }
     };
   };

diff --git a/src/DragLayer.js b/src/DragLayer.js
@@ -31,12 +31,18 @@ export default function DragLayer(collect, options = {}) {
       'Component';
 
     return class DragLayerContainer extends Component {
+      static DecoratedComponent = DecoratedComponent;
+
       static displayName = `DragLayer(${displayName})`;
 
       static contextTypes = {
         dragDropManager: PropTypes.object.isRequired
       }
 
+      getDecoratedComponentInstance() {
+        return this.refs.child;
+      }
+
       shouldComponentUpdate(nextProps, nextState) {
         return !arePropsEqual(nextProps, this.props) ||
                !shallowEqual(nextState, this.state);
@@ -83,7 +89,8 @@ export default function DragLayer(collect, options = {}) {
       render() {
         return (
           <DecoratedComponent {...this.props}
-                              {...this.state} />
+                              {...this.state}
+                              ref='child' />
         );
       }
     };

diff --git a/src/decorateHandler.js b/src/decorateHandler.js
@@ -24,21 +24,27 @@ export default function decorateHandler({
     'Component';
 
   return class DragDropContainer extends Component {
+    static DecoratedComponent = DecoratedComponent;
+
     static displayName = `${containerDisplayName}(${displayName})`;
 
     static contextTypes = {
       dragDropManager: PropTypes.object.isRequired
     }
 
+    getHandlerId() {
+      return this.handlerId;
+    }
+
+    getDecoratedComponentInstance() {
+      return this.decoratedComponentInstance;
+    }
+
     shouldComponentUpdate(nextProps, nextState) {
       return !arePropsEqual(nextProps, this.props) ||
              !shallowEqual(nextState, this.state);
     }
 
-    getInstance() {
-      return this.refs.child;
-    }
-
     constructor(props, context) {
       super(props, context);
       this.handleChange = this.handleChange.bind(this);
@@ -116,6 +122,7 @@ export default function decorateHandler({
 
       this.handlerMonitor.receiveHandlerId(handlerId);
       this.handlerConnector = handlerConnector;
+      this.handlerId = handlerId;
     }
 
     handleChange() {
@@ -126,6 +133,7 @@ export default function decorateHandler({
     }
 
     handleChildRef(component) {
+      this.decoratedComponentInstance = component;
       this.handler.receiveComponent(component);
     }