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