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