1 /* 2 * Copyright (c) 2021-2024 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 MOCK_DISTRIBUTED_SCHED_H 17 #define MOCK_DISTRIBUTED_SCHED_H 18 19 #include "distributed_sched_stub.h" 20 21 namespace OHOS { 22 namespace DistributedSchedule { 23 class MockDistributedSched : public DistributedSchedStub { 24 public: 25 explicit MockDistributedSched(bool expectedTrue = false) 26 { 27 expectedTrue_ = expectedTrue; 28 } 29 ~MockDistributedSched() = default; 30 31 int32_t StartRemoteAbility(const OHOS::AAFwk::Want& want, int32_t callerUid, int32_t requestCode, 32 uint32_t accessToken) override; 33 int32_t StartAbilityFromRemote(const OHOS::AAFwk::Want& want, 34 const OHOS::AppExecFwk::AbilityInfo& abilityInfo, int32_t requestCode, const CallerInfo& callerInfo, 35 const AccountInfo& accountInfo) override; 36 int32_t SendResultFromRemote(OHOS::AAFwk::Want& want, int32_t requestCode, const CallerInfo& callerInfo, 37 const AccountInfo& accountInfo, int32_t resultCode) override; 38 int32_t ContinueMission(const std::string& srcDeviceId, const std::string& dstDeviceId, 39 int32_t missionId, const sptr<IRemoteObject>& callback, const OHOS::AAFwk::WantParams& wantParams) override; 40 int32_t StartContinuation(const OHOS::AAFwk::Want& want, int32_t missionId, int32_t callerUid, 41 int32_t status, uint32_t accessToken) override; 42 void NotifyCompleteContinuation(const std::u16string& devId, int32_t sessionId, bool isSuccess) override; 43 int32_t NotifyContinuationResultFromRemote(int32_t sessionId, bool isSuccess, const std::string dstInfo) override; 44 int32_t ConnectRemoteAbility(const OHOS::AAFwk::Want& want, const sptr<IRemoteObject>& connect, 45 int32_t callerUid, int32_t callerPid, uint32_t accessToken) override; 46 int32_t DisconnectRemoteAbility(const sptr<IRemoteObject>& connect, int32_t callerUid, 47 uint32_t accessToken) override; 48 int32_t ConnectAbilityFromRemote(const OHOS::AAFwk::Want& want, const AppExecFwk::AbilityInfo& abilityInfo, 49 const sptr<IRemoteObject>& connect, const CallerInfo& callerInfo, const AccountInfo& accountInfo) override; 50 int32_t DisconnectAbilityFromRemote(const sptr<IRemoteObject>& connect, 51 int32_t uid, const std::string& sourceDeviceId) override; 52 int32_t NotifyProcessDiedFromRemote(const CallerInfo& callerInfo) override; 53 #ifdef SUPPORT_DISTRIBUTED_MISSION_MANAGER 54 int32_t StartSyncRemoteMissions(const std::string& devId, bool fixConflict, int64_t tag) override; 55 int32_t RegisterMissionListener(const std::u16string& devId, const sptr<IRemoteObject>& obj) override; 56 int32_t UnRegisterMissionListener(const std::u16string& devId, const sptr<IRemoteObject>& obj) override; 57 int32_t GetMissionInfos(const std::string& deviceId, int32_t numMissions, 58 std::vector<AAFwk::MissionInfo>& missionInfos) override; 59 int32_t GetRemoteMissionSnapshotInfo(const std::string& networkId, int32_t missionId, 60 std::unique_ptr<AAFwk::MissionSnapshot>& missionSnapshot) override; 61 int32_t NotifyMissionsChangedFromRemote(const std::vector<DstbMissionInfo>& missionInfos, 62 const CallerInfo& callerInfo) override; 63 int32_t StartSyncMissionsFromRemote(const CallerInfo& callerInfo, 64 std::vector<DstbMissionInfo>& missionInfos) override; 65 int32_t StopSyncMissionsFromRemote(const CallerInfo& callerInfo) override; 66 int32_t StopSyncRemoteMissions(const std::string& devId) override; 67 int32_t SetMissionContinueState(const int32_t missionId, const OHOS::AAFwk::ContinueState &state) override; 68 #endif 69 int32_t StartRemoteAbilityByCall(const OHOS::AAFwk::Want& want, const sptr<IRemoteObject>& connect, 70 int32_t callerUid, int32_t callerPid, uint32_t accessToken) override; 71 int32_t ReleaseRemoteAbility(const sptr<IRemoteObject>& connect, 72 const AppExecFwk::ElementName &element) override; 73 int32_t StartAbilityByCallFromRemote(const OHOS::AAFwk::Want& want, const sptr<IRemoteObject>& connect, 74 const CallerInfo& callerInfo, const AccountInfo& accountInfo) override; 75 int32_t ReleaseAbilityFromRemote(const sptr<IRemoteObject>& connect, const AppExecFwk::ElementName &element, 76 const CallerInfo& callerInfo) override; 77 int32_t NotifyStateChangedFromRemote(int32_t abilityState, int32_t missionId, 78 const AppExecFwk::ElementName& element) override; 79 int32_t GetDistributedComponentList(std::vector<std::string>& distributedComponents) override; 80 #ifdef SUPPORT_DISTRIBUTED_FORM_SHARE 81 int32_t StartRemoteShareForm(const std::string& remoteDeviceId, 82 const OHOS::AppExecFwk::FormShareInfo& formShareInfo) override; 83 int32_t StartShareFormFromRemote( 84 const std::string& remoteDeviceId, const OHOS::AppExecFwk::FormShareInfo& formShareInfo) override; 85 #endif 86 int32_t StartRemoteFreeInstall(const OHOS::AAFwk::Want& want, int32_t callerUid, int32_t requestCode, 87 uint32_t accessToken, const sptr<IRemoteObject>& callback) override; 88 int32_t StartFreeInstallFromRemote(const FreeInstallInfo& info, int64_t taskId) override; 89 int32_t NotifyCompleteFreeInstallFromRemote(int64_t taskId, int32_t resultCode) override; 90 int32_t NotifyCompleteFreeInstall(const FreeInstallInfo& info, int64_t taskId, int32_t resultCode); 91 int32_t StopRemoteExtensionAbility( 92 const OHOS::AAFwk::Want& want, int32_t callerUid, uint32_t accessToken, int32_t extensionType) override; 93 int32_t StopExtensionAbilityFromRemote(const OHOS::AAFwk::Want& want, 94 const CallerInfo& callerInfo, const AccountInfo& accountInfo, int32_t extensionType) override; 95 private: 96 bool expectedTrue_ = false; 97 }; 98 } // namespace DistributedSchedule 99 } // namespace OHOS 100 101 #endif // MOCK_DISTRIBUTED_SCHED_H 102