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 
SetWhiteListItem(const std::string & id,const std::string & userId,const AutoLaunchItem & item)36     void SetWhiteListItem(const std::string &id, const std::string &userId, const AutoLaunchItem &item)
37     {
38         autoLaunchItemMap_[id][userId] = item;
39     }
40 
ClearWhiteList()41     void ClearWhiteList()
42     {
43         autoLaunchItemMap_.clear();
44     }
45 
CallGetAutoLaunchItemUid(const std::string & identifier,const std::string & originalUserId,bool & ext)46     std::string CallGetAutoLaunchItemUid(const std::string &identifier, const std::string &originalUserId, bool &ext)
47     {
48         return AutoLaunch::GetAutoLaunchItemUid(identifier, originalUserId, ext);
49     }
50 
51     MOCK_METHOD1(TryCloseConnection, void(AutoLaunchItem &));
52 };
53 } // namespace DistributedDB
54 #endif  // #define MOCK_AUTO_LAUNCH_H