1 /*
2  * Copyright (c) 2021 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 MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_ABILITY_SCHEDULER_STUB_H
17 #define MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_ABILITY_SCHEDULER_STUB_H
18 
19 #include "ability_scheduler_stub.h"
20 
21 namespace OHOS {
22 namespace AAFwk {
23 class MockAbilitySchedulerStub : public AbilitySchedulerStub {
24 public:
25     MOCK_METHOD3(ScheduleAbilityTransaction, bool(const Want&, const LifeCycleStateInfo&, sptr<SessionInfo>));
26     MOCK_METHOD1(ScheduleShareData, void(const int32_t &uniqueId));
27     MOCK_METHOD3(SendResult, void(int, int, const Want&));
28     MOCK_METHOD1(ScheduleConnectAbility, void(const Want&));
29     MOCK_METHOD1(ScheduleDisconnectAbility, void(const Want&));
30     MOCK_METHOD3(ScheduleCommandAbility, void(const Want&, bool, int));
31     MOCK_METHOD3(ScheduleCommandAbilityWindow, void(const Want &, const sptr<SessionInfo> &, WindowCommand));
32     MOCK_METHOD0(SchedulePrepareTerminateAbility, bool());
33     MOCK_METHOD0(ScheduleSaveAbilityState, void());
34     MOCK_METHOD1(ScheduleRestoreAbilityState, void(const PacMap&));
35     MOCK_METHOD2(GetFileTypes, std::vector<std::string>(const Uri&, const std::string&));
36     MOCK_METHOD2(OpenFile, int(const Uri&, const std::string&));
37     MOCK_METHOD2(Insert, int(const Uri&, const NativeRdb::ValuesBucket&));
38     MOCK_METHOD3(Update, int(const Uri&, const NativeRdb::ValuesBucket&, const NativeRdb::DataAbilityPredicates&));
39     MOCK_METHOD2(Delete, int(const Uri&, const NativeRdb::DataAbilityPredicates&));
40     MOCK_METHOD3(Query, std::shared_ptr<NativeRdb::AbsSharedResultSet>(
41         const Uri&, std::vector<std::string>&, const NativeRdb::DataAbilityPredicates&));
42     MOCK_METHOD4(Call, std::shared_ptr<PacMap>(const Uri&, const std::string&, const std::string&, const PacMap&));
43     MOCK_METHOD1(GetType, std::string(const Uri& uri));
44     MOCK_METHOD2(OpenRawFile, int(const Uri& uri, const std::string& mode));
45     MOCK_METHOD2(Reload, bool(const Uri& uri, const PacMap& extras));
46     MOCK_METHOD2(BatchInsert, int(const Uri& uri, const std::vector<NativeRdb::ValuesBucket>& values));
47     MOCK_METHOD1(NormalizeUri, Uri(const Uri&));
48     MOCK_METHOD1(DenormalizeUri, Uri(const Uri&));
49     MOCK_METHOD2(ScheduleRegisterObserver, bool(const Uri& uri, const sptr<AAFwk::IDataAbilityObserver>& dataObserver));
50     MOCK_METHOD2(
51         ScheduleUnregisterObserver, bool(const Uri& uri, const sptr<AAFwk::IDataAbilityObserver>& dataObserver));
52     MOCK_METHOD1(ScheduleNotifyChange, bool(const Uri& uri));
53     MOCK_METHOD1(ExecuteBatch, std::vector<std::shared_ptr<AppExecFwk::DataAbilityResult>>(
54         const std::vector<std::shared_ptr<AppExecFwk::DataAbilityOperation>>& operations));
55     MOCK_METHOD1(NotifyContinuationResult, void(int32_t result));
56     MOCK_METHOD2(ContinueAbility, void(const std::string& deviceId, uint32_t versionCode));
57     MOCK_METHOD2(DumpAbilityInfo, void(const std::vector<std::string>& params, std::vector<std::string>& info));
58 
CallRequest()59     virtual void CallRequest()
60     {
61         return;
62     }
OnExecuteIntent(const Want & want)63     void OnExecuteIntent(const Want &want) override
64     {};
65 
CreateModalUIExtension(const Want & want)66     virtual int CreateModalUIExtension(const Want &want) override
67     {
68         return 0;
69     }
70 
UpdateSessionToken(sptr<IRemoteObject> sessionToken)71     virtual void UpdateSessionToken(sptr<IRemoteObject> sessionToken) override {}
72 };
73 }  // namespace AAFwk
74 }  // namespace OHOS
75 #endif  // MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_ABILITY_SCHEDULER_STUB_H
76