[EnhancedButton] Fix not setting focus when keyboardFocused prop set

mui · May 7, 2016 · 5294c26 · 5294c26
1 parent f35eed0
commit 5294c26
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/internal/EnhancedButton.js b/src/internal/EnhancedButton.js
@@ -1,4 +1,5 @@
 import React, {Component, PropTypes} from 'react';
+import ReactDOM from 'react-dom';
 import {createChildFragment} from '../utils/childUtils';
 import Events from '../utils/events';
 import keycode from 'keycode';
@@ -94,6 +95,10 @@ class EnhancedButton extends Component {
   componentDidMount() {
     injectStyle();
     listenForTabPresses();
+    if (this.state.isKeyboardFocused) {
+      ReactDOM.findDOMNode(this).focus();
+      this.props.onKeyboardFocus(null, true);
+    }
   }
 
   componentWillReceiveProps(nextProps) {