Small fix on test.

appirio-tech · Nov 7, 2014 · cbdd355 · cbdd355
1 parent 7838f14
commit cbdd355
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/test/models/file.js b/test/models/file.js
@@ -193,19 +193,21 @@ describe('<Unit Test>', function() {
     });
 
     afterEach(function(done) {
-      var entities = [];
       if (entity) {
-        entities.push(entity);
+        entity.destroy().success(done).error(done);
         entity = undefined;
+      } else {
+        done();
       }
+    });
+
+    after(function(done) {
       if (challenge) {
-        entities.push(challenge);
+        challenge.destroy().success(done).error(done);
         challenge = undefined;
+      } else {
+        done();
       }
-      async.forEach(entities, function (entity, callback) {
-        entity.destroy().success(callback)
-        .error(callback);
-      }, done);
     });
   });
 });