From 334e8d004cb8acf5385b2f5887e65f2333b1cbe5 Mon Sep 17 00:00:00 2001
From: memories <2716550306@qq.com>
Date: Tue, 11 Oct 2022 23:36:36 +0800
Subject: [PATCH] =?UTF-8?q?=E8=BE=93=E5=85=A5=E6=B3=95=E6=A1=86=E6=9E=B6?=
 =?UTF-8?q?=E6=94=AF=E6=8C=81=E5=BC=82=E5=B8=B8=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: buzhifeng <buzhifeng2@huawei.com>
---
 mock-generate/src/generate/generateCommonFunction.ts | 2 ++
 mock-generate/src/generate/generateMockJsFile.ts     | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/mock-generate/src/generate/generateCommonFunction.ts b/mock-generate/src/generate/generateCommonFunction.ts
index 9185c17f..8c337a18 100644
--- a/mock-generate/src/generate/generateCommonFunction.ts
+++ b/mock-generate/src/generate/generateCommonFunction.ts
@@ -39,6 +39,8 @@ export function generateCommonFunction(rootName: string, functionArray: Array<Fu
     if (functionEntity.returnType.returnKind !== SyntaxKind.VoidKeyword) {
       if (rootName === 'featureAbility' && functionEntity.returnType.returnKindName === 'Context') {
         functionBody += 'return _Context;';
+      } else if (rootName === 'inputMethod' && functionEntity.returnType.returnKindName === 'InputMethodSubtype') {
+        functionBody += 'return mockInputMethodSubtype().InputMethodSubtype;'
       } else {
         functionBody += getReturnStatement(functionEntity.returnType, sourceFile);
       }
diff --git a/mock-generate/src/generate/generateMockJsFile.ts b/mock-generate/src/generate/generateMockJsFile.ts
index f3775f59..660f0877 100644
--- a/mock-generate/src/generate/generateMockJsFile.ts
+++ b/mock-generate/src/generate/generateMockJsFile.ts
@@ -211,7 +211,7 @@ export function generateImportDeclaration(importEntity: ImportElementEntity, sou
  * @returns
  */
  function checIsDefaultExportClass(importName: string): boolean {
-  const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext'];
+  const defaultExportClass = ['Context', 'BaseContext', 'ExtensionContext', 'ApplicationContext', 'ExtensionAbility'];
   return defaultExportClass.includes(importName);
 }