fix: mockClassFunction missing (#807)

midwayjs · Jan 15, 2021 · 88fa763 · 88fa763
1 parent 30cd26c
commit 88fa763
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/packages/mock/src/interface.ts b/packages/mock/src/interface.ts
@@ -95,5 +95,7 @@ declare module 'egg' {
      * mock csrf
      */
     mockCsrf(): Application;
+
+    mockClassFunction(className: string, methodName: string, fnOrData: any): Application;
   }
 }
diff --git a/packages/mock/src/utils.ts b/packages/mock/src/utils.ts
@@ -86,6 +86,10 @@ export async function create<
             enable: true,
             package: 'egg-mock',
           },
+          'midway-mock': {
+            enable: true,
+            package: '@midwayjs/mock',
+          },
           watcher: false,
           development: false,
         },