1 /* 2 * Copyright (c) 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 MOCK_SINGLE_VER_DATA_SYNC_H 17 #define MOCK_SINGLE_VER_DATA_SYNC_H 18 19 #include <gmock/gmock.h> 20 #include "single_ver_data_sync.h" 21 22 namespace DistributedDB { 23 class MockSingleVerDataSync : public SingleVerDataSync { 24 public: CallRequestStart(SingleVerSyncTaskContext * context,int mode)25 int CallRequestStart(SingleVerSyncTaskContext *context, int mode) 26 { 27 return SingleVerDataSync::RequestStart(context, mode); 28 } 29 CallPullRequestStart(SingleVerSyncTaskContext * context)30 int CallPullRequestStart(SingleVerSyncTaskContext *context) 31 { 32 return SingleVerDataSync::PullRequestStart(context); 33 } 34 CallUpdateSendInfo(SyncTimeRange dataTimeRange,SingleVerSyncTaskContext * context)35 void CallUpdateSendInfo(SyncTimeRange dataTimeRange, SingleVerSyncTaskContext *context) 36 { 37 SingleVerDataSync::UpdateSendInfo(dataTimeRange, context); 38 } 39 CallRemoveDeviceDataIfNeed(SingleVerSyncTaskContext * context)40 int CallRemoveDeviceDataIfNeed(SingleVerSyncTaskContext *context) 41 { 42 return SingleVerDataSync::RemoveDeviceDataIfNeed(context); 43 } 44 CallDoAbilitySyncIfNeed(SingleVerSyncTaskContext * context,const Message * message,bool isControlMsg)45 int CallDoAbilitySyncIfNeed(SingleVerSyncTaskContext *context, const Message *message, bool isControlMsg) 46 { 47 return SingleVerDataSync::DoAbilitySyncIfNeed(context, message, isControlMsg); 48 } 49 SetCommunicatorHandle(ICommunicator * communicator)50 void SetCommunicatorHandle(ICommunicator *communicator) 51 { 52 communicateHandle_ = communicator; 53 } 54 55 MOCK_METHOD1(RemoveDeviceDataIfNeed, int(SingleVerSyncTaskContext *)); 56 }; 57 } // namespace DistributedDB 58 #endif // #define MOCK_SINGLE_VER_DATA_SYNC_H