!743 修改自动生成mock代码,添加class特殊处理列表项

Merge pull request !743 from fupengfei/master
This commit is contained in:
openharmony_ci
2022-10-29 14:23:32 +00:00
committed by Gitee
@@ -211,7 +211,7 @@ export function generateImportDeclaration(importEntity: ImportElementEntity, sou
* @returns
*/
function checIsDefaultExportClass(importName: string): boolean {
const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext', 'ExtensionAbility'];
const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext', 'ExtensionAbility', 'Ability'];
return defaultExportClass.includes(importName);
}