1 /* 2 * Copyright (c) 2025 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_DM_MOCK_SINGLE_KV_STORE_H 17 #define OHOS_DM_MOCK_SINGLE_KV_STORE_H 18 19 #include <gmock/gmock.h> 20 21 #include "single_kvstore.h" 22 23 namespace OHOS { 24 namespace DistributedKv { 25 class MockSingleKvStore : public SingleKvStore { 26 public: 27 MOCK_METHOD(Status, GetEntries, (const Key &, std::vector<Entry> &), (const)); 28 MOCK_METHOD(Status, GetEntries, (const DataQuery &, std::vector<Entry> &), (const)); 29 MOCK_METHOD(Status, GetResultSet, (const Key &, std::shared_ptr<KvStoreResultSet> &), (const)); 30 MOCK_METHOD(Status, GetResultSet, (const DataQuery &, std::shared_ptr<KvStoreResultSet> &), (const)); 31 MOCK_METHOD(Status, CloseResultSet, (std::shared_ptr<KvStoreResultSet> &)); 32 MOCK_METHOD(Status, GetCount, (const DataQuery &, int &), (const)); 33 MOCK_METHOD(Status, RemoveDeviceData, (const std::string &)); 34 MOCK_METHOD(Status, GetSecurityLevel, (SecurityLevel &), (const)); 35 MOCK_METHOD(Status, Sync, (const std::vector<std::string> &, SyncMode, uint32_t)); 36 MOCK_METHOD(Status, Delete, (const Key &)); 37 MOCK_METHOD(Status, Put, (const Key &, const Value &)); 38 MOCK_METHOD(Status, Get, (const Key &, Value &)); 39 MOCK_METHOD(Status, SubscribeKvStore, (SubscribeType, std::shared_ptr<KvStoreObserver>)); 40 MOCK_METHOD(Status, UnSubscribeKvStore, (SubscribeType, std::shared_ptr<KvStoreObserver>)); 41 MOCK_METHOD(Status, RegisterSyncCallback, (std::shared_ptr<KvStoreSyncCallback>)); 42 MOCK_METHOD(Status, UnRegisterSyncCallback, ()); 43 MOCK_METHOD(Status, PutBatch, (const std::vector<Entry> &)); 44 MOCK_METHOD(Status, DeleteBatch, (const std::vector<Key> &)); 45 MOCK_METHOD(Status, StartTransaction, ()); 46 MOCK_METHOD(Status, Commit, ()); 47 MOCK_METHOD(Status, Rollback, ()); 48 MOCK_METHOD(Status, SetSyncParam, (const KvSyncParam &)); 49 MOCK_METHOD(Status, GetSyncParam, (KvSyncParam &)); 50 MOCK_METHOD(Status, SetCapabilityEnabled, (bool), (const)); 51 MOCK_METHOD(Status, SetCapabilityRange, (const std::vector<std::string> &, 52 const std::vector<std::string> &), (const)); 53 MOCK_METHOD(StoreId, GetStoreId, (), (const)); 54 MOCK_METHOD(Status, Sync, (const std::vector<std::string> &, SyncMode, const DataQuery &, 55 std::shared_ptr<KvStoreSyncCallback>)); 56 MOCK_METHOD(Status, SubscribeWithQuery, (const std::vector<std::string> &, const DataQuery &)); 57 MOCK_METHOD(Status, UnsubscribeWithQuery, (const std::vector<std::string> &, const DataQuery &)); 58 MOCK_METHOD(Status, Backup, (const std::string &, const std::string &)); 59 MOCK_METHOD(Status, Restore, (const std::string &, const std::string &)); 60 MOCK_METHOD(Status, DeleteBackup, (const std::vector<std::string> &, const std::string &, 61 (std::map<std::string, DistributedKv::Status> &))); 62 }; 63 } // namespace DistributedKv 64 } // namespace OHOS 65 #endif // OHOS_DM_MOCK_SINGLE_KV_STORE_H 66