diff --git a/packages/midway-mock/bootstrap.js b/packages/midway-mock/bootstrap.js
index ef85e56fc579..59bc65e0d2c7 100644
--- a/packages/midway-mock/bootstrap.js
+++ b/packages/midway-mock/bootstrap.js
@@ -1,12 +1,24 @@
 'use strict';
 
-const {
+const assert = require('power-assert');
+const mock = require('./dist').default;
+
+const options = {
+  framework: 'midway'
+};
+
+if (process.env.EGG_BASE_DIR) options.baseDir = process.env.EGG_BASE_DIR;
+const app = mock.app(options);
+
+before(() => app.ready());
+afterEach(mock.restore);
+
+module.exports = {
   assert,
   app,
   mock,
-  mm
-} = require('egg-mock/bootstrap');
-module.exports = require('egg-mock/bootstrap');
+  mm: mock,
+};
 
 exports.assert = assert;
 exports.app = app;
diff --git a/packages/midway-mock/package.json b/packages/midway-mock/package.json
index 2d6faeaa11f0..cac842208652 100644
--- a/packages/midway-mock/package.json
+++ b/packages/midway-mock/package.json
@@ -40,6 +40,7 @@
     "injection": "^0.2.4",
     "is-type-of": "^1.2.0",
     "midway-core": "^0.2.4",
+    "power-assert": "^1.6.0",
     "ready-callback": "^2.0.1"
   }
 }
diff --git a/packages/midway-mock/src/index.ts b/packages/midway-mock/src/index.ts
index d32e9e96b170..060261bdab74 100644
--- a/packages/midway-mock/src/index.ts
+++ b/packages/midway-mock/src/index.ts
@@ -11,6 +11,11 @@ const mm2: MidwayMock = {
   mockContainer
 };
 
+mm2.app = (options) => {
+  return mm2.app(Object.assign({
+    framework: 'midway'
+  }, options));
+};
 mm2.container = mockContainer;
 module.exports = mm2;
 exports.mm = mm2;