Autofocus corrects the second time that the modal is open

misc/test-lib/helpers.js

@@ -13,6 +13,13 @@ beforeEach(function() {
       var element = angular.element(this.actual);
       return element.hasClass('ng-hide') ||
         element.css('display') == 'none';
+    },
+    toHaveFocus: function () {
+      this.message = function () {
+        return 'Expected \'' + angular.mock.dump(this.actual) + '\' to have focus';
+      };
+
+      return document.activeElement === this.actual[0];
     }
   });
 });

src/modal/modal.js

@@ -92,6 +92,8 @@ angular.module('ui.bootstrap.modal', ['ui.bootstrap.transition'])
         scope.size = attrs.size;
 
         $timeout(function () {
+          var inputsWithAutofocus = element[0].querySelectorAll('[autofocus]');
+
           // trigger CSS transitions
           scope.animate = true;
 
@@ -103,7 +105,9 @@ angular.module('ui.bootstrap.modal', ['ui.bootstrap.transition'])
            * the onscreen keyboard. Fixed by updated the focusing logic to only autofocus
            * the modal element if the modal does not contain an autofocus element.
            */
-          if (!element[0].querySelectorAll('[autofocus]').length) {
+          if (inputsWithAutofocus.length) {
+            inputsWithAutofocus[0].focus();
+          } else {
             element[0].focus();
           }
         });

src/modal/test/modal.spec.js

@@ -238,6 +238,22 @@ describe('$modal', function () {
       expect(modal.opened).toBeRejectedWith(false);
     });
 
+    it('should focus on the element that has autofocus attribute when the modal is open/reopen', function () {
+      function openAndCloseModalWithAutofocusElement() {
+        var modal = open({template: '<div><input type="text" id="auto-focus-element" autofocus></div>'});
+
+        waitForBackdropAnimation();
+
+        expect(angular.element('#auto-focus-element')).toHaveFocus();
+
+        close(modal, 'closed ok');
+
+        expect(modal.result).toBeResolvedWith('closed ok');
+      }
+
+      openAndCloseModalWithAutofocusElement();
+      openAndCloseModalWithAutofocusElement();
+    });
   });
 
   describe('default options can be changed in a provider', function () {