[EnhancedButton] Fix not setting focus when keyboardFocused prop set

mui · May 10, 2016 · 88ffe5d · 88ffe5d
1 parent f35eed0
commit 88ffe5d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/internal/EnhancedButton.js b/src/internal/EnhancedButton.js
@@ -94,6 +94,10 @@ class EnhancedButton extends Component {
   componentDidMount() {
     injectStyle();
     listenForTabPresses();
+    if (this.state.isKeyboardFocused) {
+      this.refs.enhancedButton.focus();
+      this.props.onKeyboardFocus(null, true);
+    }
   }
 
   componentWillReceiveProps(nextProps) {
@@ -304,6 +308,7 @@ class EnhancedButton extends Component {
     const buttonProps = {
       ...other,
       style: prepareStyles(mergedStyles),
+      ref: 'enhancedButton',
       disabled: disabled,
       onBlur: this.handleBlur,
       onClick: this.handleClick,