1 /* 2 * Copyright (c) 2022 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 OHOS_ABILITY_RUNTIME_MOCK_ABILITY_MANAGER_STUB_H 17 #define OHOS_ABILITY_RUNTIME_MOCK_ABILITY_MANAGER_STUB_H 18 19 #include "gmock/gmock.h" 20 21 #include "string_ex.h" 22 #include "ability_manager_errors.h" 23 #include "ability_manager_stub.h" 24 #include "hilog_wrapper.h" 25 26 namespace OHOS { 27 namespace AAFwk { 28 class MockAbilityManagerStub : public AbilityManagerStub { 29 public: 30 int StartAbility(const Want &want, int32_t userId = DEFAULT_INVAL_VALUE, int requestCode = -1); 31 32 MOCK_METHOD4(StartAbility, int(const Want &want, const sptr<IRemoteObject> &callerToken, 33 int32_t userId, int requestCode)); 34 MOCK_METHOD3(TerminateAbility, int(const sptr<IRemoteObject> &token, int resultCode, const Want *resultWant)); 35 int CloseAbility(const sptr<IRemoteObject> &token, int resultCode = DEFAULT_INVAL_VALUE, 36 const Want *resultWant = nullptr) override 37 { 38 return 0; 39 } 40 int MinimizeAbility(const sptr<IRemoteObject> &token, bool fromUser = false) override 41 { 42 return 0; 43 } 44 MOCK_METHOD4(ConnectAbility, int(const Want &want, const sptr<IAbilityConnection> &connect, 45 const sptr<IRemoteObject> &callerToken, int32_t userId)); 46 MOCK_METHOD1(DisconnectAbility, int(const sptr<IAbilityConnection> &connect)); 47 MOCK_METHOD3(AcquireDataAbility, 48 sptr<IAbilityScheduler>(const Uri &uri, bool tryBind, const sptr<IRemoteObject> &callerToken)); 49 MOCK_METHOD2( 50 ReleaseDataAbility, int(sptr<IAbilityScheduler> dataAbilityScheduler, const sptr<IRemoteObject> &callerToken)); 51 MOCK_METHOD2(AttachAbilityThread, int(const sptr<IAbilityScheduler> &scheduler, const sptr<IRemoteObject> &token)); 52 MOCK_METHOD3(AbilityTransitionDone, int(const sptr<IRemoteObject> &token, int state, const PacMap &)); 53 MOCK_METHOD2( 54 ScheduleConnectAbilityDone, int(const sptr<IRemoteObject> &token, const sptr<IRemoteObject> &remoteObject)); 55 MOCK_METHOD1(ScheduleDisconnectAbilityDone, int(const sptr<IRemoteObject> &token)); 56 MOCK_METHOD1(ScheduleCommandAbilityDone, int(const sptr<IRemoteObject> &token)); 57 58 void DumpState(const std::string &args, std::vector<std::string> &state); 59 60 MOCK_METHOD2(TerminateAbilityResult, int(const sptr<IRemoteObject> &token, int startId)); 61 62 int StopServiceAbility(const Want &want, int32_t userId = DEFAULT_INVAL_VALUE); 63 64 MOCK_METHOD2(TerminateAbilityByCaller, int(const sptr<IRemoteObject> &callerToken, int requestCode)); 65 MOCK_METHOD1(KillProcess, int(const std::string &bundleName)); 66 MOCK_METHOD2(UninstallApp, int(const std::string &bundleName, int32_t uid)); 67 68 MOCK_METHOD2( 69 GetWantSender, sptr<IWantSender>(const WantSenderInfo &wantSenderInfo, const sptr<IRemoteObject> &callerToken)); 70 MOCK_METHOD2(SendWantSender, int(const sptr<IWantSender> &target, const SenderInfo &senderInfo)); 71 MOCK_METHOD1(CancelWantSender, void(const sptr<IWantSender> &sender)); 72 MOCK_METHOD1(GetPendingWantUid, int(const sptr<IWantSender> &target)); 73 MOCK_METHOD1(GetPendingWantUserId, int(const sptr<IWantSender> &target)); 74 MOCK_METHOD1(GetPendingWantBundleName, std::string(const sptr<IWantSender> &target)); 75 MOCK_METHOD1(GetPendingWantCode, int(const sptr<IWantSender> &target)); 76 MOCK_METHOD1(GetPendingWantType, int(const sptr<IWantSender> &target)); 77 MOCK_METHOD2(RegisterCancelListener, void(const sptr<IWantSender> &sender, const sptr<IWantReceiver> &receiver)); 78 MOCK_METHOD2(UnregisterCancelListener, void(const sptr<IWantSender> &sender, const sptr<IWantReceiver> &receiver)); 79 MOCK_METHOD2(GetPendingRequestWant, int(const sptr<IWantSender> &target, std::shared_ptr<Want> &want)); 80 MOCK_METHOD5(StartAbility, int(const Want &want, const AbilityStartSetting &abilityStartSetting, 81 const sptr<IRemoteObject> &callerToken, int32_t userId, int requestCode)); 82 MOCK_METHOD1(GetMissionIdByToken, int32_t(const sptr<IRemoteObject> &token)); 83 MOCK_METHOD1(GetPendinTerminateAbilityTestgRequestWant, void(int id)); 84 MOCK_METHOD3(StartContinuation, int(const Want &want, const sptr<IRemoteObject> &abilityToken, int32_t status)); 85 MOCK_METHOD2(NotifyContinuationResult, int(int32_t missionId, int32_t result)); 86 MOCK_METHOD5(ContinueMission, int(const std::string &srcDeviceId, const std::string &dstDeviceId, 87 int32_t missionId, const sptr<IRemoteObject> &callBack, AAFwk::WantParams &wantParams)); 88 MOCK_METHOD3(ContinueAbility, int(const std::string &deviceId, int32_t missionId, uint32_t versionCode)); 89 MOCK_METHOD3(NotifyCompleteContinuation, void(const std::string &deviceId, int32_t sessionId, bool isSuccess)); 90 91 MOCK_METHOD1(LockMissionForCleanup, int(int32_t missionId)); 92 MOCK_METHOD1(UnlockMissionForCleanup, int(int32_t missionId)); 93 MOCK_METHOD1(RegisterMissionListener, int(const sptr<IMissionListener> &listener)); 94 MOCK_METHOD1(UnRegisterMissionListener, int(const sptr<IMissionListener> &listener)); 95 MOCK_METHOD3( 96 GetMissionInfos, int(const std::string& deviceId, int32_t numMax, std::vector<MissionInfo> &missionInfos)); 97 MOCK_METHOD3(GetMissionInfo, int(const std::string& deviceId, int32_t missionId, MissionInfo &missionInfo)); 98 MOCK_METHOD1(CleanMission, int(int32_t missionId)); 99 MOCK_METHOD0(CleanAllMissions, int()); 100 MOCK_METHOD1(MoveMissionToFront, int(int32_t missionId)); 101 MOCK_METHOD2(MoveMissionToFront, int(int32_t missionId, const StartOptions &startOptions)); 102 MOCK_METHOD2(SetMissionLabel, int(const sptr<IRemoteObject> &token, const std::string &label)); 103 MOCK_METHOD2(SetMissionIcon, int(const sptr<IRemoteObject> &token, 104 const std::shared_ptr<OHOS::Media::PixelMap> &icon)); 105 MOCK_METHOD1(ClearUpApplicationData, int(const std::string &)); 106 MOCK_METHOD1(GetAbilityRunningInfos, int(std::vector<AbilityRunningInfo> &info)); 107 MOCK_METHOD2(GetExtensionRunningInfos, int(int upperLimit, std::vector<ExtensionRunningInfo> &info)); 108 MOCK_METHOD1(GetProcessRunningInfos, int(std::vector<AppExecFwk::RunningProcessInfo> &info)); 109 MOCK_METHOD2(GetWantSenderInfo, int(const sptr<IWantSender> &target, std::shared_ptr<WantSenderInfo> &info)); 110 111 virtual int StartAbility( 112 const Want &want, 113 const StartOptions &startOptions, 114 const sptr<IRemoteObject> &callerToken, 115 int32_t userId = DEFAULT_INVAL_VALUE, 116 int requestCode = DEFAULT_INVAL_VALUE) 117 { 118 return 0; 119 } 120 StartUser(int userId)121 int StartUser(int userId) override 122 { 123 return 0; 124 } 125 StopUser(int userId,const sptr<IStopUserCallback> & callback)126 int StopUser(int userId, const sptr<IStopUserCallback> &callback) override 127 { 128 return 0; 129 } StartSyncRemoteMissions(const std::string & devId,bool fixConflict,int64_t tag)130 int StartSyncRemoteMissions(const std::string& devId, bool fixConflict, int64_t tag) override 131 { 132 return 0; 133 } StopSyncRemoteMissions(const std::string & devId)134 int StopSyncRemoteMissions(const std::string& devId) override 135 { 136 return 0; 137 } RegisterMissionListener(const std::string & deviceId,const sptr<IRemoteMissionListener> & listener)138 int RegisterMissionListener(const std::string &deviceId, 139 const sptr<IRemoteMissionListener> &listener) override 140 { 141 return 0; 142 } UnRegisterMissionListener(const std::string & deviceId,const sptr<IRemoteMissionListener> & listener)143 int UnRegisterMissionListener(const std::string &deviceId, 144 const sptr<IRemoteMissionListener> &listener) override 145 { 146 return 0; 147 } 148 StartAbilityByCall(const Want & want,const sptr<IAbilityConnection> & connect,const sptr<IRemoteObject> & callerToken)149 virtual int StartAbilityByCall( 150 const Want &want, const sptr<IAbilityConnection> &connect, const sptr<IRemoteObject> &callerToken) 151 { 152 return 0; 153 } ReleaseCall(const sptr<IAbilityConnection> & connect,const AppExecFwk::ElementName & element)154 virtual int ReleaseCall(const sptr<IAbilityConnection> &connect, 155 const AppExecFwk::ElementName &element) 156 { 157 return 0; 158 } 159 GetMissionSnapshot(const std::string & deviceId,int32_t missionId,MissionSnapshot & snapshot,bool isLowResolution)160 virtual int GetMissionSnapshot(const std::string& deviceId, int32_t missionId, 161 MissionSnapshot& snapshot, bool isLowResolution) 162 { 163 return 0; 164 } 165 UpdateMissionSnapShot(const sptr<IRemoteObject> & token)166 virtual void UpdateMissionSnapShot(const sptr<IRemoteObject>& token) 167 { 168 return; 169 } 170 RegisterSnapshotHandler(const sptr<ISnapshotHandler> & handler)171 virtual int RegisterSnapshotHandler(const sptr<ISnapshotHandler>& handler) 172 { 173 return 0; 174 } 175 RegisterWindowManagerServiceHandler(const sptr<IWindowManagerServiceHandler> & handler)176 int RegisterWindowManagerServiceHandler(const sptr<IWindowManagerServiceHandler>& handler) override 177 { 178 return 0; 179 } 180 CompleteFirstFrameDrawing(const sptr<IRemoteObject> & abilityToken)181 void CompleteFirstFrameDrawing(const sptr<IRemoteObject> &abilityToken) override {} 182 SendANRProcessID(int pid)183 virtual int SendANRProcessID(int pid) 184 { 185 return 0; 186 } 187 SetAbilityController(const sptr<AppExecFwk::IAbilityController> & abilityController,bool imAStabilityTest)188 int SetAbilityController(const sptr<AppExecFwk::IAbilityController> &abilityController, 189 bool imAStabilityTest) override 190 { 191 return 0; 192 } IsRunningInStabilityTest()193 bool IsRunningInStabilityTest() override 194 { 195 return true; 196 } DumpSysState(const std::string & args,std::vector<std::string> & info,bool isClient,bool isUserID,int UserID)197 void DumpSysState( 198 const std::string& args, std::vector<std::string>& info, bool isClient, bool isUserID, int UserID) override 199 {} 200 StartUserTest(const Want & want,const sptr<IRemoteObject> & observer)201 int StartUserTest(const Want &want, const sptr<IRemoteObject> &observer) override 202 { 203 return 0; 204 } 205 FinishUserTest(const std::string & msg,const int64_t & resultCode,const std::string & bundleName)206 int FinishUserTest( 207 const std::string &msg, const int64_t &resultCode, const std::string &bundleName) override 208 { 209 return 0; 210 } 211 GetTopAbility(sptr<IRemoteObject> & token)212 int GetTopAbility(sptr<IRemoteObject> &token) override 213 { 214 return 0; 215 } 216 DelegatorDoAbilityForeground(const sptr<IRemoteObject> & token)217 int DelegatorDoAbilityForeground(const sptr<IRemoteObject> &token) override 218 { 219 return 0; 220 } 221 DelegatorDoAbilityBackground(const sptr<IRemoteObject> & token)222 int DelegatorDoAbilityBackground(const sptr<IRemoteObject> &token) override 223 { 224 return 0; 225 } 226 227 #ifdef ABILITY_COMMAND_FOR_TEST ForceTimeoutForTest(const std::string & abilityName,const std::string & state)228 int ForceTimeoutForTest(const std::string &abilityName, const std::string &state) override 229 { 230 return 0; 231 } 232 BlockAmsService()233 virtual int BlockAmsService() 234 { 235 return 0; 236 } 237 BlockAppService()238 virtual int BlockAppService() 239 { 240 return 0; 241 } 242 BlockAbility(int32_t abilityRecordId)243 virtual int BlockAbility(int32_t abilityRecordId) 244 { 245 return 0; 246 } 247 #endif 248 249 public: 250 std::string powerState_; 251 }; 252 } // namespace AAFwk 253 } // namespace OHOS 254 255 #endif // OHOS_ABILITY_RUNTIME_MOCK_ABILITY_MANAGER_STUB_H 256