5 files changed, 50 insertions, 30 deletions
diff --git a/test/ng/directive/ngIfSpec.js b/test/ng/directive/ngIfSpec.js
index 5726e092..0cca57d5 100755
--- a/test/ng/directive/ngIfSpec.js
+++ b/test/ng/directive/ngIfSpec.js
@@ -77,18 +77,22 @@ describe('ngIf', function () {
 
 describe('ngIf ngAnimate', function () {
   var vendorPrefix, window;
-  var body, element;
+  var body, element, $rootElement;
 
   function html(html) {
-    body.html(html);
-    element = body.children().eq(0);
+    $rootElement.html(html);
+    element = $rootElement.children().eq(0);
     return element;
   }
 
-  beforeEach(function() {
+  beforeEach(module(function() {
     // we need to run animation on attached elements;
-    body = jqLite(document.body);
-  });
+    return function(_$rootElement_) {
+      $rootElement = _$rootElement_;
+      body = jqLite(document.body);
+      body.append($rootElement);
+    };
+  }));
 
   afterEach(function(){
     dealoc(body);
diff --git a/test/ng/directive/ngIncludeSpec.js b/test/ng/directive/ngIncludeSpec.js
index 2e8d7023..93709431 100644
--- a/test/ng/directive/ngIncludeSpec.js
+++ b/test/ng/directive/ngIncludeSpec.js
@@ -299,11 +299,11 @@ describe('ngInclude', function() {
 
 describe('ngInclude ngAnimate', function() {
   var vendorPrefix, window;
-  var body, element;
+  var body, element, $rootElement;
 
   function html(html) {
-    body.html(html);
-    element = body.children().eq(0);
+    $rootElement.html(html);
+    element = $rootElement.children().eq(0);
     return element;
   }
 
@@ -312,10 +312,14 @@ describe('ngInclude ngAnimate', function() {
     element.css(vendorPrefix + cssProp, cssValue);
   }
 
-  beforeEach(function() {
+  beforeEach(module(function() {
     // we need to run animation on attached elements;
-    body = jqLite(document.body);
-  });
+    return function(_$rootElement_) {
+      $rootElement = _$rootElement_;
+      body = jqLite(document.body);
+      body.append($rootElement);
+    };
+  }));
 
   afterEach(function(){
     dealoc(body);
diff --git a/test/ng/directive/ngRepeatSpec.js b/test/ng/directive/ngRepeatSpec.js
index 492f851d..21ea21e9 100644
--- a/test/ng/directive/ngRepeatSpec.js
+++ b/test/ng/directive/ngRepeatSpec.js
@@ -708,11 +708,11 @@ describe('ngRepeat', function() {
 
 describe('ngRepeat ngAnimate', function() {
   var vendorPrefix, window;
-  var body, element;
+  var body, element, $rootElement;
 
   function html(html) {
-    body.html(html);
-    element = body.children().eq(0);
+    $rootElement.html(html);
+    element = $rootElement.children().eq(0);
     return element;
   }
 
@@ -721,10 +721,14 @@ describe('ngRepeat ngAnimate', function() {
     element.css(vendorPrefix + cssProp, cssValue);
   }
 
-  beforeEach(function() {
+  beforeEach(module(function() {
     // we need to run animation on attached elements;
-    body = jqLite(document.body);
-  });
+    return function(_$rootElement_) {
+      $rootElement = _$rootElement_;
+      body = jqLite(document.body);
+      body.append($rootElement);
+    };
+  }));
 
   afterEach(function(){
     dealoc(body);
diff --git a/test/ng/directive/ngSwitchSpec.js b/test/ng/directive/ngSwitchSpec.js
index 9ccb3b12..ab231ec2 100644
--- a/test/ng/directive/ngSwitchSpec.js
+++ b/test/ng/directive/ngSwitchSpec.js
@@ -216,18 +216,22 @@ describe('ngSwitch', function() {
 
 describe('ngSwitch ngAnimate', function() {
   var vendorPrefix, window;
-  var body, element;
+  var body, element, $rootElement;
 
   function html(html) {
-    body.html(html);
-    element = body.children().eq(0);
+    $rootElement.html(html);
+    element = $rootElement.children().eq(0);
     return element;
   }
 
-  beforeEach(function() {
+  beforeEach(module(function() {
     // we need to run animation on attached elements;
-    body = jqLite(document.body);
-  });
+    return function(_$rootElement_) {
+      $rootElement = _$rootElement_;
+      body = jqLite(document.body);
+      body.append($rootElement);
+    };
+  }));
 
   afterEach(function(){
     dealoc(body);
diff --git a/test/ngRoute/directive/ngViewSpec.js b/test/ngRoute/directive/ngViewSpec.js
index 6beb3d27..50531c18 100644
--- a/test/ngRoute/directive/ngViewSpec.js
+++ b/test/ngRoute/directive/ngViewSpec.js
@@ -511,11 +511,12 @@ describe('ngView', function() {
 
   describe('ngAnimate ', function() {
     var window, vendorPrefix;
-    var body, element;
+    var body, element, $rootElement;
 
     function html(html) {
-      body.html(html);
-      element = body.children().eq(0);
+      $rootElement.html(html);
+      body.append($rootElement);
+      element = $rootElement.children().eq(0);
       return element;
     }
 
@@ -524,10 +525,13 @@ describe('ngView', function() {
       element.css(vendorPrefix + cssProp, cssValue);
     }
 
-    beforeEach(function() {
+    beforeEach(module(function() {
       // we need to run animation on attached elements;
-      body = jqLite(document.body);
-    });
+      return function(_$rootElement_) {
+        $rootElement = _$rootElement_;
+        body = jqLite(document.body);
+      };
+    }));
 
     afterEach(function(){
       dealoc(body);