feat: support init hook before mock app init (#109)

eggjs · Dec 12, 2019 · 4c31c9e · 4c31c9e
1 parent cbab52a
commit 4c31c9e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/lib/app.js b/lib/app.js
@@ -56,6 +56,10 @@ class MockApplication extends EventEmitter {
   }
 
   * [INIT]() {
+    if (this.options.beforeInit) {
+      yield this.options.beforeInit(this);
+      delete this.options.beforeInit;
+    }
     if (this.options.clean !== false) {
       const logDir = path.join(this.options.baseDir, 'logs');
       try {

diff --git a/test/app.test.js b/test/app.test.js
@@ -66,6 +66,26 @@ describe('test/app.test.js', () => {
     yield app.close();
   });
 
+  it('support options.beforeInit', function* () {
+    const baseDir = path.join(fixtures, 'app');
+    const app = mm.app({
+      baseDir,
+      customEgg: path.join(__dirname, '../node_modules/egg'),
+      cache: false,
+      beforeInit(instance) {
+        return new Promise(resolve => {
+          setTimeout(() => {
+            instance.options.test = 'abc';
+            resolve();
+          }, 100);
+        });
+      },
+    });
+    yield app.ready();
+    assert(!app.options.beforeInit);
+    assert(app.options.test === 'abc');
+  });
+
   // TODO: implement ready(err)
   it.skip('should emit error when load Application fail', done => {
     const baseDir = path.join(fixtures, 'app-fail');