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_AUTO_LAUNCH_H 17 #define MOCK_AUTO_LAUNCH_H 18 19 #include <gmock/gmock.h> 20 #include "auto_launch.h" 21 22 namespace DistributedDB { 23 class MockAutoLaunch : public AutoLaunch { 24 public: SetAutoLaunchItem(const std::string & identify,const std::string & userId,const AutoLaunchItem & item)25 void SetAutoLaunchItem(const std::string &identify, const std::string &userId, const AutoLaunchItem &item) 26 { 27 std::lock_guard<std::mutex> autoLock(extLock_); 28 extItemMap_[identify][userId] = item; 29 } 30 CallExtConnectionLifeCycleCallbackTask(const std::string & identifier,const std::string & userId)31 void CallExtConnectionLifeCycleCallbackTask(const std::string &identifier, const std::string &userId) 32 { 33 AutoLaunch::ExtConnectionLifeCycleCallbackTask(identifier, userId); 34 } 35 36 MOCK_METHOD1(TryCloseConnection, void(AutoLaunchItem &)); 37 }; 38 } // namespace DistributedDB 39 #endif // #define MOCK_AUTO_LAUNCH_H