1 /* 2 * Copyright (c) 2021-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 MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_ABILITY_SCHEDULER_H 17 #define MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_ABILITY_SCHEDULER_H 18 19 #include "want.h" 20 #include "lifecycle_state_info.h" 21 #include "gmock/gmock.h" 22 #include "ability_scheduler_interface.h" 23 24 namespace OHOS { 25 namespace AAFwk { 26 class MockAbilityScheduler : public IAbilityScheduler { 27 public: 28 MOCK_METHOD2(ScheduleAbilityTransaction, void(const Want &want, const LifeCycleStateInfo &targetState)); 29 MOCK_METHOD3(SendResult, void(int requestCode, int resultCode, const Want &resultWant)); 30 MOCK_METHOD1(ScheduleConnectAbility, void(const Want &want)); 31 MOCK_METHOD1(ScheduleDisconnectAbility, void(const Want &want)); 32 MOCK_METHOD3(ScheduleCommandAbility, void(const Want &want, bool restart, int startid)); 33 MOCK_METHOD0(AsObject, sptr<IRemoteObject>()); 34 MOCK_METHOD0(ScheduleSaveAbilityState, void()); 35 MOCK_METHOD1(ScheduleRestoreAbilityState, void(const PacMap &inState)); 36 MOCK_METHOD1(ScheduleNewWant, void(const Want &want)); 37 MOCK_METHOD2(ScheduleRegisterObserver, bool(const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &dataObserver)); 38 MOCK_METHOD2( 39 ScheduleUnregisterObserver, bool(const Uri &uri, const sptr<AAFwk::IDataAbilityObserver> &dataObserver)); 40 MOCK_METHOD1(ScheduleNotifyChange, bool(const Uri &uri)); 41 MOCK_METHOD1(ExecuteBatch, std::vector<std::shared_ptr<AppExecFwk::DataAbilityResult>>( 42 const std::vector<std::shared_ptr<AppExecFwk::DataAbilityOperation>> &operations)); 43 MOCK_METHOD1(NotifyContinuationResult, void(int32_t result)); 44 MOCK_METHOD2(ContinueAbility, void(const std::string& deviceId, uint32_t versionCode)); 45 MOCK_METHOD2(DumpAbilityInfo, void(const std::vector<std::string> ¶ms, std::vector<std::string> &info)); GetFileTypes(const Uri & uri,const std::string & mimeTypeFilter)46 std::vector<std::string> GetFileTypes(const Uri &uri, const std::string &mimeTypeFilter) 47 { 48 std::vector<std::string> types; 49 return types; 50 } 51 OpenFile(const Uri & uri,const std::string & mode)52 int OpenFile(const Uri &uri, const std::string &mode) 53 { 54 return -1; 55 } 56 Insert(const Uri & uri,const NativeRdb::ValuesBucket & value)57 int Insert(const Uri &uri, const NativeRdb::ValuesBucket &value) 58 { 59 return -1; 60 } 61 Update(const Uri & uri,const NativeRdb::ValuesBucket & value,const NativeRdb::DataAbilityPredicates & predicates)62 int Update(const Uri &uri, const NativeRdb::ValuesBucket &value, const NativeRdb::DataAbilityPredicates &predicates) 63 { 64 return -1; 65 } 66 Delete(const Uri & uri,const NativeRdb::DataAbilityPredicates & predicates)67 int Delete(const Uri &uri, const NativeRdb::DataAbilityPredicates &predicates) 68 { 69 return -1; 70 } 71 Query(const Uri & uri,std::vector<std::string> & columns,const NativeRdb::DataAbilityPredicates & predicates)72 std::shared_ptr<NativeRdb::AbsSharedResultSet> Query( 73 const Uri &uri, std::vector<std::string> &columns, const NativeRdb::DataAbilityPredicates &predicates) 74 { 75 return nullptr; 76 } 77 Call(const Uri & uri,const std::string & method,const std::string & arg,const AppExecFwk::PacMap & pacMap)78 std::shared_ptr<AppExecFwk::PacMap> Call( 79 const Uri &uri, const std::string &method, const std::string &arg, const AppExecFwk::PacMap &pacMap) 80 { 81 return nullptr; 82 } 83 GetType(const Uri & uri)84 virtual std::string GetType(const Uri &uri) override 85 { 86 return " "; 87 } 88 OpenRawFile(const Uri & uri,const std::string & mode)89 virtual int OpenRawFile(const Uri &uri, const std::string &mode) override 90 { 91 return -1; 92 } 93 Reload(const Uri & uri,const PacMap & extras)94 virtual bool Reload(const Uri &uri, const PacMap &extras) override 95 { 96 return false; 97 } 98 BatchInsert(const Uri & uri,const std::vector<NativeRdb::ValuesBucket> & values)99 virtual int BatchInsert(const Uri &uri, const std::vector<NativeRdb::ValuesBucket> &values) override 100 { 101 return -1; 102 } 103 NormalizeUri(const Uri & uri)104 virtual Uri NormalizeUri(const Uri &uri) override 105 { 106 Uri urivalue(""); 107 return urivalue; 108 } 109 DenormalizeUri(const Uri & uri)110 virtual Uri DenormalizeUri(const Uri &uri) override 111 { 112 Uri urivalue(""); 113 return urivalue; 114 } CallRequest()115 virtual void CallRequest() override 116 { 117 return; 118 } 119 #ifdef ABILITY_COMMAND_FOR_TEST BlockAbility()120 virtual int BlockAbility() override 121 { 122 return 0; 123 } 124 #endif 125 }; 126 } // namespace AAFwk 127 } // namespace OHOS 128 129 #endif // MODULETEST_OHOS_ABILITY_RUNTIME_MOCK_ABILITY_SCHEDULER_H 130