1 /* 2 * Copyright (c) 2023 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 #ifndef MOCK_ICLOUD_DB_H 16 #define MOCK_ICLOUD_DB_H 17 #include <gmock/gmock.h> 18 #include "icloud_db.h" 19 20 namespace DistributedDB { 21 22 class MockICloudDB : public ICloudDb { 23 public: 24 MOCK_METHOD3(BatchInsert, DBStatus(const std::string &, std::vector<VBucket> &&, 25 std::vector<VBucket> &)); 26 MOCK_METHOD3(BatchUpdate, DBStatus(const std::string &, std::vector<VBucket> &&, 27 std::vector<VBucket> &)); 28 MOCK_METHOD2(BatchDelete, DBStatus(const std::string &, std::vector<VBucket> &)); 29 MOCK_METHOD3(Query, DBStatus(const std::string &, VBucket &, std::vector<VBucket> &)); 30 MOCK_METHOD1(GetEmptyCursor, std::pair<DBStatus, std::string>(const std::string &)); 31 MOCK_METHOD0(Lock, std::pair<DBStatus, uint32_t>(void)); 32 MOCK_METHOD0(UnLock, DBStatus(void)); 33 MOCK_METHOD0(HeartBeat, DBStatus(void)); 34 MOCK_METHOD0(Close, DBStatus(void)); 35 }; 36 } 37 #endif // #define MOCK_ICLOUD_DB_H