chore: use mocha global hook to register before/after

bootstrap.js

@@ -20,12 +20,9 @@ if (process.env.ENABLE_MOCHA_PARALLEL && process.env.AUTO_AGENT) {
   if (typeof beforeAll === 'function') {
     // jest
     beforeAll(() => app.ready());
-  } else {
-    // mocha
-    before(() => app.ready());
+    afterEach(() => app.backgroundTasksFinished());
+    afterEach(mock.restore);
   }
-  afterEach(() => app.backgroundTasksFinished());
-  afterEach(mock.restore);
 }
 
 module.exports = {

lib/agent.js

@@ -0,0 +1,18 @@
+let agent;
+
+const Agent = require('./parallel/agent');
+const { getEggOptions } = require('./utils');
+
+exports.setupAgent = async () => {
+  if (process.env.ENABLE_MOCHA_PARALLEL && process.env.AUTO_AGENT) {
+    agent = Agent(getEggOptions());
+    await agent.ready();
+  }
+  return agent;
+};
+
+exports.closeAgent = async () => {
+  if (agent) {
+    await agent.close();
+  }
+};

lib/parallel/agent_register.js

@@ -1,31 +1,3 @@
-const Agent = require('./agent');
-const { getEggOptions } = require('../utils');
-const mock = require('../../index').default;
+console.warn('[deprecated] use `egg-mock/register.js` instead of `egg-mock/lib/parallel/agent_register.js`');
 
-let agent;
-exports.mochaGlobalSetup = async () => {
-  agent = Agent(getEggOptions());
-  await agent.ready();
-};
-
-exports.mochaGlobalTeardown = async () => {
-  if (agent) {
-    await agent.close();
-  }
-};
-
-exports.mochaHooks = {
-  async beforeAll() {
-    const { app } = require('../../bootstrap');
-    if (app) {
-      await app.ready();
-    }
-  },
-  async afterEach() {
-    const { app } = require('../../bootstrap');
-    if (app) {
-      await app.backgroundTasksFinished();
-    }
-    await mock.restore();
-  },
-};
+module.exports = require('../../register');

package.json

@@ -19,8 +19,8 @@
     "lint": "eslint lib app index.js test/*.test.js",
     "tsd": "tsd",
     "test": "npm run lint && npm run tsd && npm run test-local",
-    "test-local": "egg-bin test --full-trace",
-    "cov": "egg-bin cov",
+    "test-local": "egg-bin test --full-trace -r ./register.js",
+    "cov": "egg-bin cov -r ./register.js",
     "ci": "npm run lint && npm run tsd && npm run cov",
     "contributors": "git-contributor"
   },

register.js

@@ -0,0 +1,26 @@
+const mock = require('./index').default;
+const agent = require('./lib/agent');
+
+exports.mochaGlobalSetup = async () => {
+  await agent.setupAgent();
+};
+
+exports.mochaGlobalTeardown = async () => {
+  await agent.closeAgent();
+};
+
+exports.mochaHooks = {
+  async beforeAll() {
+    const { app } = require('./bootstrap');
+    if (app) {
+      await app.ready();
+    }
+  },
+  async afterEach() {
+    const { app } = require('./bootstrap');
+    if (app) {
+      await app.backgroundTasksFinished();
+    }
+    await mock.restore();
+  },
+};