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_SYNC_ENGINE_H 17 #define MOCK_SYNC_ENGINE_H 18 19 #include <gmock/gmock.h> 20 #include "sync_engine.h" 21 22 namespace DistributedDB { 23 class MockSyncEngine : public SyncEngine { 24 public: 25 MOCK_METHOD1(CreateSyncTaskContext, ISyncTaskContext *(const ISyncInterface &syncInterface)); 26 MOCK_METHOD1(GetTargetUserId, std::string(const std::string &dev)); 27 InitSubscribeManager()28 void InitSubscribeManager() 29 { 30 subManager_ = std::make_shared<SubscribeManager>(); 31 } 32 CallGetSyncTaskContext(const DeviceSyncTarget & target,int & errCode)33 ISyncTaskContext *CallGetSyncTaskContext(const DeviceSyncTarget &target, int &errCode) 34 { 35 return SyncEngine::GetSyncTaskContext(target, errCode); 36 } 37 CallFindSyncTaskContext(const DeviceSyncTarget & target,bool isNeedCorrectUserId)38 ISyncTaskContext *CallFindSyncTaskContext(const DeviceSyncTarget &target, bool isNeedCorrectUserId) 39 { 40 return SyncEngine::FindSyncTaskContext(target, isNeedCorrectUserId); 41 } 42 }; 43 } // namespace DistributedDB 44 #endif // #define MOCK_SYNC_ENGINE_H 45