fix(ng-cloak): work with class

Author: mhevery

src/directives.js

@@ -937,7 +937,7 @@ var ngStyleDirective = ngDirective(function(scope, element, attr) {
  */
 var ngCloakDirective = ngDirective({
   compile: function(element, attr) {
-    attr.$set(attr.$attr.ngCloak, undefined);
+    attr.$set('ngCloak', undefined);
     element.removeClass('ng-cloak');
   }
 });

test/directivesSpec.js

@@ -543,7 +543,7 @@ describe("directive", function() {
     }));
 
 
-    it('should remove ng-cloak class from a compiled element', inject(function($rootScope, $compile) {
+    it('should remove ng-cloak class from a compiled element with attribute', inject(function($rootScope, $compile) {
       element = jqLite('<div ng:cloak class="foo ng-cloak bar"></div>');
 
       expect(element.hasClass('foo')).toBe(true);
@@ -556,6 +556,21 @@ describe("directive", function() {
       expect(element.hasClass('ng-cloak')).toBe(false);
       expect(element.hasClass('bar')).toBe(true);
     }));
+
+
+    it('should remove ng-cloak class from a compiled element', inject(function($rootScope, $compile) {
+      element = jqLite('<div class="foo ng-cloak bar"></div>');
+
+      expect(element.hasClass('foo')).toBe(true);
+      expect(element.hasClass('ng-cloak')).toBe(true);
+      expect(element.hasClass('bar')).toBe(true);
+
+      $compile(element);
+
+      expect(element.hasClass('foo')).toBe(true);
+      expect(element.hasClass('ng-cloak')).toBe(false);
+      expect(element.hasClass('bar')).toBe(true);
+    }));
   });