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 LNN_SERVICE_MOCK_H
17 #define LNN_SERVICE_MOCK_H
18 
19 #include <gmock/gmock.h>
20 #include <mutex>
21 
22 #include "bus_center_event.h"
23 #include "bus_center_manager.h"
24 #include "disc_interface.h"
25 #include "lnn_async_callback_utils.h"
26 #include "lnn_connection_addr_utils.h"
27 #include "lnn_deviceinfo_to_profile.h"
28 #include "lnn_feature_capability.h"
29 #include "lnn_heartbeat_strategy.h"
30 #include "lnn_ohos_account_adapter.h"
31 #include "lnn_settingdata_event_monitor.h"
32 #include "message_handler.h"
33 #include "softbus_adapter_crypto.h"
34 #include "softbus_bus_center.h"
35 #include "softbus_common.h"
36 #include "softbus_feature_config.h"
37 #include "softbus_wifi_api_adapter.h"
38 #include "lnn_event_monitor_impl.h"
39 
40 namespace OHOS {
41 class LnnServiceInterface {
42 public:
LnnServiceInterface()43     LnnServiceInterface() {};
~LnnServiceInterface()44     virtual ~LnnServiceInterface() {};
45     virtual int32_t LnnInitBusCenterEvent(void) = 0;
46     virtual void LnnDeinitBusCenterEvent(void) = 0;
47     virtual int32_t LnnRegisterEventHandler(LnnEventType event, LnnEventHandler handler) = 0;
48     virtual void LnnNotifyJoinResult(ConnectionAddr *addr,
49         const char *networkId, int32_t retCode) = 0;
50     virtual void LnnNotifyLeaveResult(const char *networkId, int32_t retCode) = 0;
51     virtual void LnnNotifyOnlineState(bool isOnline, NodeBasicInfo *info) = 0;
52     virtual void LnnNotifyBasicInfoChanged(NodeBasicInfo *info, NodeBasicInfoType type) = 0;
53     virtual void LnnNotifyWlanStateChangeEvent(void *state) = 0;
54     virtual void LnnNotifyBtStateChangeEvent(void *state) = 0;
55     virtual void LnnNotifyLnnRelationChanged(const char *udid, ConnectionAddrType type,
56         uint8_t relation, bool isJoin) = 0;
57     virtual void LnnNotifyMasterNodeChanged(bool isMaster, const char* masterNodeUdid,
58         int32_t weight) = 0;
59     virtual int32_t LnnInitGetDeviceName(LnnDeviceNameHandler handler) = 0;
60     virtual void RegisterNameMonitor(void) = 0;
61     virtual void LnnUnregisterEventHandler(LnnEventType event, LnnEventHandler handler) = 0;
62     virtual int32_t LnnOfflineTimingByHeartbeat(const char *networkId, ConnectionAddrType addrType) = 0;
63     virtual int32_t LnnGetSettingDeviceName(char *deviceName, uint32_t len) = 0;
64     virtual uint32_t AuthGenRequestId(void) = 0;
65     virtual void AuthHandleLeaveLNN(AuthHandle authHandle) = 0;
66     virtual int32_t AuthGetDeviceUuid(int64_t authId, char *uuid, uint16_t size) = 0;
67     virtual int32_t SoftBusGetWifiDeviceConfig(SoftBusWifiDevConf *configList, uint32_t *num) = 0;
68     virtual int32_t SoftBusConnectToDevice(const SoftBusWifiDevConf *wifiConfig) = 0;
69     virtual int32_t SoftBusDisconnectDevice(void) = 0;
70     virtual ConnectionAddrType LnnDiscTypeToConnAddrType(DiscoveryType type) = 0;
71     virtual void UpdateProfile(const NodeInfo *info) = 0;
72     virtual bool IsFeatureSupport(uint64_t feature, FeatureCapability capaBit) = 0;
73     virtual int32_t LnnStartHbByTypeAndStrategy(
74         LnnHeartbeatType hbType, LnnHeartbeatStrategyType strategy, bool isRelay) = 0;
75     virtual bool SoftBusIsWifiTripleMode(void) = 0;
76     virtual SoftBusBand SoftBusGetLinkBand(void) = 0;
77     virtual SoftBusWifiDetailState SoftBusGetWifiState(void) = 0;
78     virtual bool SoftBusHasWifiDirectCapability(void) = 0;
79     virtual char* SoftBusGetWifiInterfaceCoexistCap(void) = 0;
80     virtual int32_t LnnAsyncCallbackHelper(SoftBusLooper *looper,
81         LnnAsyncCallbackFunc callback, void *para);
82     virtual int32_t LnnAsyncCallbackDelayHelper(SoftBusLooper *looper, LnnAsyncCallbackFunc callback,
83         void *para, uint64_t delayMillis);
84     virtual int32_t SoftBusGenerateRandomArray(unsigned char *randStr, uint32_t len) = 0;
85     virtual int32_t LnnGetDeviceDisplayName(const char *nickName,
86         const char *defaultName, char *deviceName, uint32_t len);
87     virtual int32_t LnnGetUnifiedDeviceName(char *unifiedName, uint32_t len);
88     virtual int32_t LnnGetUnifiedDefaultDeviceName(char *unifiedDefaultName, uint32_t len);
89     virtual int32_t GetCurrentAccount(int64_t *account);
90     virtual int32_t LnnSetLocalUnifiedName(const char *unifiedName);
91     virtual void LnnNotifyLocalNetworkIdChanged(void);
92     virtual int32_t LnnGetSettingNickName(const char *defaultName,
93         const char *unifiedName, char *nickName, uint32_t len);
94     virtual int SoftbusGetConfig(ConfigType type, unsigned char *val, uint32_t len) = 0;
95     virtual int32_t LnnSubscribeAccountBootEvent(AccountEventHandle handle) = 0;
96 };
97 
98 class LnnServicetInterfaceMock : public LnnServiceInterface {
99 public:
100     LnnServicetInterfaceMock();
101     ~LnnServicetInterfaceMock() override;
102     MOCK_METHOD0(LnnInitBusCenterEvent, int32_t (void));
103     MOCK_METHOD0(LnnDeinitBusCenterEvent, void (void));
104     MOCK_METHOD2(LnnRegisterEventHandler, int32_t (LnnEventType, LnnEventHandler));
105     MOCK_METHOD3(LnnNotifyJoinResult, void (ConnectionAddr *, const char *, int32_t));
106     MOCK_METHOD2(LnnNotifyLeaveResult, void (const char *, int32_t));
107     MOCK_METHOD2(LnnNotifyOnlineState, void (bool, NodeBasicInfo *));
108     MOCK_METHOD2(LnnNotifyBasicInfoChanged, void (NodeBasicInfo *, NodeBasicInfoType));
109     MOCK_METHOD1(LnnNotifyWlanStateChangeEvent, void (void *));
110     MOCK_METHOD1(LnnNotifyBtStateChangeEvent, void (void *));
111     MOCK_METHOD4(LnnNotifyLnnRelationChanged, void (const char *, ConnectionAddrType, uint8_t, bool));
112     MOCK_METHOD3(LnnNotifyMasterNodeChanged, void (bool, const char*, int32_t));
113     MOCK_METHOD1(LnnInitGetDeviceName, int32_t (LnnDeviceNameHandler));
114     MOCK_METHOD0(RegisterNameMonitor, void (void));
115     MOCK_METHOD2(LnnUnregisterEventHandler, void (LnnEventType, LnnEventHandler));
116     MOCK_METHOD2(LnnOfflineTimingByHeartbeat, int32_t (const char *, ConnectionAddrType));
117     MOCK_METHOD2(LnnGetSettingDeviceName, int32_t (char *, uint32_t));
118     MOCK_METHOD0(AuthGenRequestId, uint32_t ());
119     MOCK_METHOD1(AuthHandleLeaveLNN, void (AuthHandle));
120     MOCK_METHOD3(AuthGetDeviceUuid, int32_t (int64_t, char*, uint16_t));
121     MOCK_METHOD2(SoftBusGetWifiDeviceConfig, int32_t (SoftBusWifiDevConf *, uint32_t *));
122     MOCK_METHOD1(SoftBusConnectToDevice, int32_t (const SoftBusWifiDevConf *));
123     MOCK_METHOD0(SoftBusDisconnectDevice, int32_t ());
124     MOCK_METHOD1(LnnDiscTypeToConnAddrType, ConnectionAddrType (DiscoveryType));
125     MOCK_METHOD1(UpdateProfile, void (const NodeInfo *));
126     MOCK_METHOD2(IsFeatureSupport, bool (uint64_t, FeatureCapability));
127     MOCK_METHOD3(LnnStartHbByTypeAndStrategy, int32_t (LnnHeartbeatType, LnnHeartbeatStrategyType, bool));
128     MOCK_METHOD0(SoftBusIsWifiTripleMode, bool ());
129     MOCK_METHOD0(SoftBusGetLinkBand, SoftBusBand ());
130     MOCK_METHOD0(SoftBusGetWifiState, SoftBusWifiDetailState ());
131     MOCK_METHOD0(SoftBusHasWifiDirectCapability, bool ());
132     MOCK_METHOD0(SoftBusGetWifiInterfaceCoexistCap, char* ());
133     MOCK_METHOD3(LnnAsyncCallbackHelper, int32_t (SoftBusLooper *, LnnAsyncCallbackFunc, void *));
134     MOCK_METHOD4(LnnAsyncCallbackDelayHelper, int32_t (SoftBusLooper *, LnnAsyncCallbackFunc, void *, uint64_t));
135     MOCK_METHOD2(SoftBusGenerateRandomArray, int32_t (unsigned char *, uint32_t));
136     MOCK_METHOD4(LnnGetDeviceDisplayName, int32_t (const char *, const char *, char *, uint32_t));
137     MOCK_METHOD2(LnnGetUnifiedDeviceName, int32_t (char *, uint32_t));
138     MOCK_METHOD2(LnnGetUnifiedDefaultDeviceName, int32_t (char *, uint32_t));
139     MOCK_METHOD1(GetCurrentAccount, int32_t (int64_t *));
140     MOCK_METHOD1(LnnSetLocalUnifiedName, int32_t (const char *));
141     MOCK_METHOD0(LnnNotifyLocalNetworkIdChanged, void ());
142     MOCK_METHOD4(LnnGetSettingNickName, int32_t (const char *, const char *, char *, uint32_t));
143     MOCK_METHOD3(SoftbusGetConfig, int (ConfigType, unsigned char *, uint32_t));
144     MOCK_METHOD1(LnnSubscribeAccountBootEvent, int32_t (AccountEventHandle handle));
145     static int32_t ActionOfLnnRegisterEventHandler(LnnEventType event, LnnEventHandler handler);
146     static int32_t ActionOfLnnInitGetDeviceName(LnnDeviceNameHandler handler);
147     static int32_t ActionOfLnnGetSettingDeviceName(char *deviceName, uint32_t len);
148     static inline std::map<LnnEventType, LnnEventHandler> g_lnnEventHandlers;
149     static inline LnnDeviceNameHandler g_deviceNameHandler;
150 };
151 } // namespace OHOS
152 #endif // LNN_SERVICE_MOCK_H