fix(ng-cloak): work with class

1 files changed, 16 insertions, 1 deletions

diff --git a/test/directivesSpec.js b/test/directivesSpec.js
index c0533745..1ce6090b 100644
--- a/test/directivesSpec.js
+++ b/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);
+    }));
   });