test: more test fixes re: #6754

Automattic · Jul 29, 2018 · d5b43da · d5b43da
1 parent e91d404
commit d5b43da
Showing 1 changed file with 13 additions and 15 deletions.
diff --git a/test/docs/transactions.test.js b/test/docs/transactions.test.js
@@ -160,26 +160,24 @@ describe('transactions', function() {
 
     beforeEach(function() {
       if (_skipped) {
-        return this.skip(); // https://github.com/mochajs/mocha/issues/2546
-      }
-      return Author.deleteMany({}).then(() => Article.deleteMany({}));
-    });
-
-    beforeEach(function() {
-      if (_skipped) {
-        return this.skip(); // https://github.com/mochajs/mocha/issues/2546
+        this.skip();
+        return Promise.resolve(); // https://github.com/mochajs/mocha/issues/2546
       }
-      return db.startSession().then(_session => {
-        session = _session;
-        session.startTransaction();
-      });
+      return Author.deleteMany({}).
+        then(() => Article.deleteMany({})).
+        then(() => db.startSession()).
+        then(_session => {
+          session = _session;
+          session.startTransaction();
+        });
     });
 
-    afterEach(function() {
+    afterEach(function(done) {
       if (_skipped) {
-        return; // https://github.com/mochajs/mocha/issues/2546
+        return done(); // https://github.com/mochajs/mocha/issues/2546
       }
-      return session.commitTransaction();
+      session.commitTransaction();
+      done();
     });
 
     it('`populate()` uses the querys session', function() {