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 #include "concurrent_adapter.h" 22 #include "res_finalizer.h" 23 24 namespace DistributedDB { 25 class MockAutoLaunch : public AutoLaunch { 26 public: SetAutoLaunchItem(const std::string & identify,const std::string & userId,const AutoLaunchItem & item)27 void SetAutoLaunchItem(const std::string &identify, const std::string &userId, const AutoLaunchItem &item) 28 { 29 ConcurrentAdapter::AdapterAutoLock(extLock_); 30 ResFinalizer finalizer([this]() { ConcurrentAdapter::AdapterAutoUnLock(extLock_); }); 31 extItemMap_[identify][userId] = item; 32 } 33 CallExtConnectionLifeCycleCallbackTask(const std::string & identifier,const std::string & userId)34 void CallExtConnectionLifeCycleCallbackTask(const std::string &identifier, const std::string &userId) 35 { 36 AutoLaunch::ExtConnectionLifeCycleCallbackTask(identifier, userId); 37 } 38 SetWhiteListItem(const std::string & id,const std::string & userId,const AutoLaunchItem & item)39 void SetWhiteListItem(const std::string &id, const std::string &userId, const AutoLaunchItem &item) 40 { 41 autoLaunchItemMap_[id][userId] = item; 42 } 43 ClearWhiteList()44 void ClearWhiteList() 45 { 46 autoLaunchItemMap_.clear(); 47 } 48 CallGetAutoLaunchItemUid(const std::string & identifier,const std::string & originalUserId,bool & ext)49 std::string CallGetAutoLaunchItemUid(const std::string &identifier, const std::string &originalUserId, bool &ext) 50 { 51 return AutoLaunch::GetAutoLaunchItemUid(identifier, originalUserId, ext); 52 } 53 54 MOCK_METHOD1(TryCloseConnection, void(AutoLaunchItem &)); 55 }; 56 } // namespace DistributedDB 57 #endif // #define MOCK_AUTO_LAUNCH_H