1 files changed, 49 insertions, 0 deletions
diff --git a/test/ng/directive/ngCloakSpec.js b/test/ng/directive/ngCloakSpec.js
new file mode 100644
index 00000000..f3c28b60
--- /dev/null
+++ b/test/ng/directive/ngCloakSpec.js
@@ -0,0 +1,49 @@
+'use strict';
+
+describe('ng-cloak', function() {
+  var element;
+
+
+  afterEach(function() {
+    dealoc(element);
+  });
+
+
+  it('should get removed when an element is compiled', inject(function($rootScope, $compile) {
+    element = jqLite('<div ng-cloak></div>');
+    expect(element.attr('ng-cloak')).toBe('');
+    $compile(element);
+    expect(element.attr('ng-cloak')).toBeUndefined();
+  }));
+
+
+  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);
+    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);
+  }));
+
+
+  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);
+  }));
+});