fix: ensure ContextInitiator be called after ctx ready

Author: killagu

plugin/controller/test/fixtures/apps/controller-app/app/controller/MiddlewareController.ts

@@ -8,6 +8,7 @@ import {
 import AppService from '../../modules/multi-module-service/AppService';
 import { countMw } from '../middleware/count_mw';
 import { logMwFactory } from '../middleware/log_mw';
+import { callModuleCtx } from '../middleware/call_module';
 
 @HTTPController({
   path: '/middleware',
@@ -33,4 +34,13 @@ export class MiddlewareController {
   async middleware() {
     return {};
   }
+
+  @HTTPMethod({
+    method: HTTPMethodEnum.GET,
+    path: '/methodCallModule',
+  })
+  @Middleware(callModuleCtx)
+  async middlewareCallModule() {
+    return {};
+  }
 }

plugin/controller/test/fixtures/apps/controller-app/app/middleware/call_module.ts

@@ -0,0 +1,7 @@
+import { Context } from 'egg';
+import { Next } from '@eggjs/tegg';
+
+export async function callModuleCtx(ctx: Context, next: Next) {
+  await (ctx.module as any).multiModuleService.appService.findApp('foo');
+  await next();
+}

plugin/controller/test/http/middleware.test.ts

@@ -48,4 +48,11 @@ describe('test/middleware.test.ts', () => {
         assert(res.body.log === 'use middleware');
       });
   });
+
+  it('method middleware call module should work', async () => {
+    app.mockCsrf();
+    await app.httpRequest()
+      .get('/middleware/methodCallModule')
+      .expect(200);
+  });
 });

plugin/tegg/lib/EggContextCompatibleHook.ts

@@ -35,4 +35,12 @@ export class EggContextCompatibleHook implements LifecycleHook<EggContextLifecyc
       await EggContainerFactory.getOrCreateEggObject(protoObj as EggPrototype);
     }
   }
+
+  async postCreate(_, ctx: EggContext): Promise<void> {
+    const rootProto = ctx.get(ROOT_PROTO);
+    if (rootProto) {
+      // Ensure ContextInitiator be called.
+      await EggContainerFactory.getOrCreateEggObject(rootProto as EggPrototype);
+    }
+  }
 }