fix: runtime mock support initHandler (#749)

midwayjs · Dec 11, 2020 · 0b993be · 0b993be
1 parent 18c32b9
commit 0b993be
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/packages-serverless/runtime-mock/src/runtime.ts b/packages-serverless/runtime-mock/src/runtime.ts
@@ -18,6 +18,8 @@ export interface MockRuntimeOptions {
   bootstrap?;
   events?: FunctionEvent[];
   handler?: any;
+  initHandler?: any;
+  initContext?: any;
   runtime?: Runtime;
 }
 
@@ -60,8 +62,11 @@ export class MockRuntime {
       await this.bootstrap.start();
       await sleep(500);
       this.runtime = this.bootstrap.getRuntime();
-      return this.runtime;
     }
+    if (this.options.initHandler) {
+      await this.options.initHandler(this.options.initContext);
+    }
+    return this.runtime;
   }
 
   async invoke(...args) {

diff --git a/packages-serverless/runtime-mock/test/index.test.ts b/packages-serverless/runtime-mock/test/index.test.ts
@@ -5,9 +5,14 @@ const assert = require('assert');
 
 describe('/test/index.test.ts', () => {
   it('should create runtime', async () => {
+    let initHanlderResult;
     const runtime = createRuntime({
       functionDir: join(__dirname, './code'),
       events: [new HTTPEvent()],
+      initHandler: arg => {
+        initHanlderResult = arg;
+      },
+      initContext: 'test'
     });
     await runtime.start();
     const result = await runtime.invoke({
@@ -18,5 +23,6 @@ describe('/test/index.test.ts', () => {
     });
     assert.equal(result, 'hello Alan');
     await runtime.close();
+    assert.equal(initHanlderResult, 'test');
   });
 });