Pruebas para enfocar una directiva AngularJS

¿Cómo se puede probar el enfoque en una directiva AngularJS? Espero que lo siguiente funcione:

describe('focus test', function(){
    it('should focus element', function(){
        var element = $('<input type="text" />');
        // Append to body because otherwise it can't be foccused
        element.appendTo(document.body);
        element.focus();
        expect(element.is(':focus')).toBe(true);
    });
});

Sin embargo, esto solo funciona en IE, falla en Firefox y Chrome

Actualizar: La solución de @S McCrohan funciona. Usando esto creé un 'toHaveFocus' matcher:

beforeEach(function(){
    this.addMatchers({
        toHaveFocus: function(){
            this.message = function(){
                return 'Expected \'' + angular.mock.dump(this.actual) + '\' to have focus';
            };

            return document.activeElement === this.actual[0];
        }
    });
});

Que se utiliza de la siguiente manera:

expect(myElement).toHaveFocus();

Tenga en cuenta que para las pruebas relacionadas con focus, el elemento compilado debe adjuntarse al DOM, lo que se puede hacer de la siguiente manera:

myElement.appendTo(document.body);