1 /* 2 * Copyright (c) 2025 Huawei Device Co., Ltd. 3 * Licensed under the Apache License, Version 2.0 (the "License"); 4 * you may not use this file except in compliance with the License. 5 * You may obtain a copy of the License at 6 * 7 * http://www.apache.org/licenses/LICENSE-2.0 8 * 9 * Unless required by applicable law or agreed to in writing, software 10 * distributed under the License is distributed on an "AS IS" BASIS, 11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 * See the License for the specific language governing permissions and 13 * limitations under the License. 14 */ 15 16 #ifndef BASE_COMMON_EVENT_SERVICE_TEST_UNITEST_MOCK_IABILITYMANAGER_H 17 #define BASE_COMMON_EVENT_SERVICE_TEST_UNITEST_MOCK_IABILITYMANAGER_H 18 19 #include <gmock/gmock.h> 20 21 #include "ability_manager_interface.h" 22 23 namespace OHOS { 24 namespace AAFwk { 25 class MockIAbilityManager : public IAbilityManager { 26 public: 27 MOCK_METHOD(sptr<IRemoteObject>, AsObject, (), (override)); 28 MOCK_METHOD(int, StartAbility, (const Want &want, int32_t userId, int requestCode), (override)); 29 MOCK_METHOD(int, StartAbility, 30 (const Want &want, const sptr<IRemoteObject> &callerToken, int32_t userId, int requestCode), (override)); 31 MOCK_METHOD(int, StartAbilityWithSpecifyTokenId, 32 (const Want &want, const sptr<IRemoteObject> &callerToken, uint32_t specifyTokenId, int32_t userId, 33 int requestCode), 34 (override)); 35 MOCK_METHOD(int32_t, StartAbilityByInsightIntent, 36 (const Want &want, const sptr<IRemoteObject> &callerToken, uint64_t intentId, int32_t userId), (override)); 37 MOCK_METHOD(int, StartAbility, 38 (const Want &want, const AbilityStartSetting &abilityStartSetting, const sptr<IRemoteObject> &callerToken, 39 int32_t userId, int requestCode), 40 (override)); 41 MOCK_METHOD(int, StartAbility, 42 (const Want &want, const StartOptions &startOptions, const sptr<IRemoteObject> &callerToken, int32_t userId, 43 int requestCode), 44 (override)); 45 MOCK_METHOD( 46 int, TerminateAbility, (const sptr<IRemoteObject> &token, int resultCode, const Want *resultWant), (override)); 47 MOCK_METHOD( 48 int, CloseAbility, (const sptr<IRemoteObject> &token, int resultCode, const Want *resultWant), (override)); 49 MOCK_METHOD(int, MinimizeAbility, (const sptr<IRemoteObject> &token, bool fromUser), (override)); 50 MOCK_METHOD(int32_t, ConnectAbility, 51 (const Want &want, const sptr<IAbilityConnection> &connect, const sptr<IRemoteObject> &callerToken, 52 int32_t userId), 53 (override)); 54 MOCK_METHOD(int, DisconnectAbility, (sptr<IAbilityConnection> connect), (override)); 55 MOCK_METHOD(sptr<IAbilityScheduler>, AcquireDataAbility, 56 (const Uri &uri, bool tryBind, const sptr<IRemoteObject> &callerToken), (override)); 57 MOCK_METHOD(int, ReleaseDataAbility, 58 (sptr<IAbilityScheduler> dataAbilityScheduler, const sptr<IRemoteObject> &callerToken), (override)); 59 MOCK_METHOD(int, AttachAbilityThread, (const sptr<IAbilityScheduler> &scheduler, const sptr<IRemoteObject> &token), 60 (override)); 61 MOCK_METHOD( 62 int, AbilityTransitionDone, (const sptr<IRemoteObject> &token, int state, const PacMap &saveData), (override)); 63 MOCK_METHOD(int, ScheduleConnectAbilityDone, 64 (const sptr<IRemoteObject> &token, const sptr<IRemoteObject> &remoteObject), (override)); 65 MOCK_METHOD(int, ScheduleDisconnectAbilityDone, (const sptr<IRemoteObject> &token), (override)); 66 MOCK_METHOD(int, ScheduleCommandAbilityDone, (const sptr<IRemoteObject> &token), (override)); 67 MOCK_METHOD(int, ScheduleCommandAbilityWindowDone, 68 (const sptr<IRemoteObject> &token, const sptr<AAFwk::SessionInfo> &sessionInfo, AAFwk::WindowCommand winCmd, 69 AAFwk::AbilityCommand abilityCmd), 70 (override)); 71 MOCK_METHOD(void, DumpState, (const std::string &args, std::vector<std::string> &state), (override)); 72 MOCK_METHOD(void, DumpSysState, 73 (const std::string &args, std::vector<std::string> &state, bool isClient, bool isUserID, int userID), 74 (override)); 75 MOCK_METHOD( 76 int, StopServiceAbility, (const Want &want, int32_t userId, const sptr<IRemoteObject> &token), (override)); 77 MOCK_METHOD(int, KillProcess, (const std::string &bundleName, bool clearPageStack, int32_t appIndex), (override)); 78 #ifdef ABILITY_COMMAND_FOR_TEST 79 MOCK_METHOD(int, ForceTimeoutForTest, (const std::string &abilityName, const std::string &state), (override)); 80 #endif 81 MOCK_METHOD(sptr<IWantSender>, GetWantSender, 82 (const WantSenderInfo &wantSenderInfo, const sptr<IRemoteObject> &callerToken, int32_t uid), (override)); 83 MOCK_METHOD(int, SendWantSender, (sptr<IWantSender> target, SenderInfo &senderInfo), (override)); 84 MOCK_METHOD(void, CancelWantSender, (const sptr<IWantSender> &sender), (override)); 85 MOCK_METHOD(int, GetPendingWantUid, (const sptr<IWantSender> &target), (override)); 86 MOCK_METHOD(int, GetPendingWantUserId, (const sptr<IWantSender> &target), (override)); 87 MOCK_METHOD(std::string, GetPendingWantBundleName, (const sptr<IWantSender> &target), (override)); 88 MOCK_METHOD(int, GetPendingWantCode, (const sptr<IWantSender> &target), (override)); 89 MOCK_METHOD(int, GetPendingWantType, (const sptr<IWantSender> &target), (override)); 90 MOCK_METHOD(void, RegisterCancelListener, (const sptr<IWantSender> &sender, const sptr<IWantReceiver> &receiver), 91 (override)); 92 MOCK_METHOD(void, UnregisterCancelListener, (const sptr<IWantSender> &sender, const sptr<IWantReceiver> &receiver), 93 (override)); 94 MOCK_METHOD(int, GetPendingRequestWant, (const sptr<IWantSender> &target, std::shared_ptr<Want> &want), (override)); 95 MOCK_METHOD( 96 int, GetWantSenderInfo, (const sptr<IWantSender> &target, std::shared_ptr<WantSenderInfo> &info), (override)); 97 MOCK_METHOD(int, ContinueMission, 98 (const std::string &srcDeviceId, const std::string &dstDeviceId, int32_t missionId, 99 const sptr<IRemoteObject> &callBack, AAFwk::WantParams &wantParams), 100 (override)); 101 MOCK_METHOD( 102 int, ContinueAbility, (const std::string &deviceId, int32_t missionId, uint32_t versionCode), (override)); 103 MOCK_METHOD(int, StartContinuation, (const Want &want, const sptr<IRemoteObject> &abilityToken, int32_t status), 104 (override)); 105 MOCK_METHOD( 106 void, NotifyCompleteContinuation, (const std::string &deviceId, int32_t sessionId, bool isSuccess), (override)); 107 MOCK_METHOD(int, NotifyContinuationResult, (int32_t missionId, int32_t result), (override)); 108 MOCK_METHOD(int, LockMissionForCleanup, (int32_t missionId), (override)); 109 MOCK_METHOD(int, UnlockMissionForCleanup, (int32_t missionId), (override)); 110 MOCK_METHOD(int, RegisterMissionListener, (const sptr<IMissionListener> &listener), (override)); 111 MOCK_METHOD(int, UnRegisterMissionListener, (const sptr<IMissionListener> &listener), (override)); 112 MOCK_METHOD(int, GetMissionInfos, 113 (const std::string &deviceId, int32_t numMax, std::vector<MissionInfo> &missionInfos), (override)); 114 MOCK_METHOD( 115 int, GetMissionInfo, (const std::string &deviceId, int32_t missionId, MissionInfo &missionInfo), (override)); 116 MOCK_METHOD(int, GetMissionSnapshot, 117 (const std::string &deviceId, int32_t missionId, MissionSnapshot &snapshot, bool isLowResolution), (override)); 118 MOCK_METHOD(int, CleanMission, (int32_t missionId), (override)); 119 MOCK_METHOD(int, CleanAllMissions, (), (override)); 120 MOCK_METHOD(int, MoveMissionToFront, (int32_t missionId), (override)); 121 MOCK_METHOD(int, MoveMissionToFront, (int32_t missionId, const StartOptions &startOptions), (override)); 122 MOCK_METHOD(int, StartAbilityByCall, 123 (const Want &want, const sptr<IAbilityConnection> &connect, const sptr<IRemoteObject> &callerToken, 124 int32_t accountId, bool isSilent), 125 (override)); 126 MOCK_METHOD(int, ReleaseCall, (const sptr<IAbilityConnection> &connect, const AppExecFwk::ElementName &element), 127 (override)); 128 MOCK_METHOD(int, StartUser, (int userId, sptr<IUserCallback> callback, bool isAppRecovery), (override)); 129 MOCK_METHOD(int, StopUser, (int userId, const sptr<IUserCallback> &callback), (override)); 130 #ifdef SUPPORT_SCREEN 131 MOCK_METHOD(int, SetMissionLabel, (const sptr<IRemoteObject> &abilityToken, const std::string &label), (override)); 132 MOCK_METHOD(int, SetMissionIcon, 133 (const sptr<IRemoteObject> &token, const std::shared_ptr<OHOS::Media::PixelMap> &icon), (override)); 134 MOCK_METHOD(void, CompleteFirstFrameDrawing, (const sptr<IRemoteObject> &abilityToken), (override)); 135 #endif 136 MOCK_METHOD(int, GetAbilityRunningInfos, (std::vector<AbilityRunningInfo> & info), (override)); 137 MOCK_METHOD(int, GetExtensionRunningInfos, (int upperLimit, std::vector<ExtensionRunningInfo> &info), (override)); 138 MOCK_METHOD(int, GetProcessRunningInfos, (std::vector<AppExecFwk::RunningProcessInfo> & info), (override)); 139 MOCK_METHOD(int, StartSyncRemoteMissions, (const std::string &devId, bool fixConflict, int64_t tag), (override)); 140 MOCK_METHOD(int, StopSyncRemoteMissions, (const std::string &devId), (override)); 141 MOCK_METHOD(int, RegisterMissionListener, 142 (const std::string &deviceId, const sptr<IRemoteMissionListener> &listener), (override)); 143 MOCK_METHOD(int, UnRegisterMissionListener, 144 (const std::string &deviceId, const sptr<IRemoteMissionListener> &listener), (override)); 145 MOCK_METHOD(int, SetAbilityController, 146 (const sptr<AppExecFwk::IAbilityController> &abilityController, bool imAStabilityTest), (override)); 147 MOCK_METHOD(bool, IsRunningInStabilityTest, (), (override)); 148 MOCK_METHOD(int, RegisterSnapshotHandler, (const sptr<ISnapshotHandler> &handler), (override)); 149 MOCK_METHOD(int, StartUserTest, (const Want &want, const sptr<IRemoteObject> &observer), (override)); 150 MOCK_METHOD(int, FinishUserTest, (const std::string &msg, const int64_t &resultCode, const std::string &bundleName), 151 (override)); 152 MOCK_METHOD(int, GetTopAbility, (sptr<IRemoteObject> & token), (override)); 153 MOCK_METHOD(int, DelegatorDoAbilityForeground, (const sptr<IRemoteObject> &token), (override)); 154 MOCK_METHOD(int, DelegatorDoAbilityBackground, (const sptr<IRemoteObject> &token), (override)); 155 MOCK_METHOD(int, DoAbilityForeground, (const sptr<IRemoteObject> &token, uint32_t flag), (override)); 156 MOCK_METHOD(int, DoAbilityBackground, (const sptr<IRemoteObject> &token, uint32_t flag), (override)); 157 MOCK_METHOD(int32_t, GetMissionIdByToken, (const sptr<IRemoteObject> &token), (override)); 158 MOCK_METHOD(void, GetAbilityTokenByCalleeObj, (const sptr<IRemoteObject> &callStub, sptr<IRemoteObject> &token), 159 (override)); 160 MOCK_METHOD(int32_t, IsValidMissionIds, 161 (const std::vector<int32_t> &missionIds, std::vector<MissionValidResult> &results), (override)); 162 MOCK_METHOD(int32_t, ReportDrawnCompleted, (const sptr<IRemoteObject> &callerToken), (override)); 163 MOCK_METHOD(int32_t, RegisterAppDebugListener, (sptr<AppExecFwk::IAppDebugListener> listener), (override)); 164 MOCK_METHOD(int32_t, UnregisterAppDebugListener, (sptr<AppExecFwk::IAppDebugListener> listener), (override)); 165 MOCK_METHOD(int32_t, AttachAppDebug, (const std::string &bundleName, bool isDebugFromLocal), (override)); 166 MOCK_METHOD(int32_t, DetachAppDebug, (const std::string &bundleName, bool isDebugFromLocal), (override)); 167 MOCK_METHOD(int32_t, ExecuteIntent, 168 (uint64_t key, const sptr<IRemoteObject> &callerToken, const InsightIntentExecuteParam ¶m), (override)); 169 MOCK_METHOD(int32_t, ExecuteInsightIntentDone, 170 (const sptr<IRemoteObject> &token, uint64_t intentId, const InsightIntentExecuteResult &result), (override)); 171 MOCK_METHOD(int32_t, SetApplicationAutoStartupByEDM, (const AutoStartupInfo &info, bool flag), (override)); 172 MOCK_METHOD(int32_t, CancelApplicationAutoStartupByEDM, (const AutoStartupInfo &info, bool flag), (override)); 173 MOCK_METHOD(int32_t, GetForegroundUIAbilities, (std::vector<AppExecFwk::AbilityStateData> & list), (override)); 174 MOCK_METHOD(void, ScheduleClearRecoveryPageStack, (), (override)); 175 MOCK_METHOD(void, NotifyFrozenProcessByRSS, (const std::vector<int32_t> &pidList, int32_t uid), (override)); 176 MOCK_METHOD(void, KillProcessWithPrepareTerminateDone, 177 (const std::string &moduleName, int32_t prepareTermination, bool isExist), (override)); 178 bool mockFunctionExcuted = false; 179 }; 180 } // namespace AAFwk 181 } // namespace OHOS 182 #endif // BASE_COMMON_EVENT_SERVICE_TEST_UNITEST_MOCK_IABILITYMANAGER_H