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_BT_NETWORK_IMPL_MOCK_H 17 #define LNN_BT_NETWORK_IMPL_MOCK_H 18 19 #include <gmock/gmock.h> 20 #include <mutex> 21 22 #include "auth_hichain_adapter.h" 23 #include "bus_center_event.h" 24 #include "bus_center_info_key.h" 25 #include "bus_center_manager.h" 26 #include "lnn_device_info.h" 27 #include "lnn_distributed_net_ledger.h" 28 #include "lnn_network_manager.h" 29 #include "lnn_node_info.h" 30 #include "lnn_physical_subnet_manager.h" 31 #include "softbus_adapter_bt_common.h" 32 #include "softbus_adapter_crypto.h" 33 #include "softbus_bus_center.h" 34 #include "softbus_common.h" 35 #include "softbus_protocol_def.h" 36 37 namespace OHOS { 38 class LnnBtNetworkImplInterface { 39 public: LnnBtNetworkImplInterface()40 LnnBtNetworkImplInterface() {}; ~LnnBtNetworkImplInterface()41 virtual ~LnnBtNetworkImplInterface() {}; 42 virtual int32_t LnnRequestLeaveSpecific(const char *networkId, ConnectionAddrType addrType) = 0; 43 virtual int32_t LnnRequestLeaveByAddrType(const bool *type, uint32_t typeLen) = 0; 44 virtual int32_t SoftBusGetBtState(void) = 0; 45 virtual int32_t SoftBusGetBtMacAddr(SoftBusBtAddr *mac) = 0; 46 virtual int32_t ConvertBtMacToStr(char *strMac, uint32_t strMacLen, const uint8_t *binMac, uint32_t binMacLen) = 0; 47 virtual int32_t LnnRegisterEventHandler(LnnEventType event, LnnEventHandler handler) = 0; 48 virtual int32_t LnnGetNetIfTypeByName(const char *ifName, LnnNetIfType *type) = 0; 49 virtual bool LnnVisitNetif(VisitNetifCallback callback, void *data) = 0; 50 virtual int32_t LnnRegistPhysicalSubnet(LnnPhysicalSubnet *manager) = 0; 51 virtual void LnnNotifyPhysicalSubnetStatusChanged(const char *ifName, ProtocolType protocolType, void *status) = 0; 52 virtual int32_t LnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len) = 0; 53 virtual const NodeInfo *LnnGetLocalNodeInfo(void) = 0; 54 virtual int32_t LnnGetAuthPort(const NodeInfo *info) = 0; 55 virtual int32_t LnnGetSessionPort(const NodeInfo *info) = 0; 56 virtual int32_t LnnGetProxyPort(const NodeInfo *info) = 0; 57 virtual const char *LnnGetBtMac(const NodeInfo *info) = 0; 58 virtual const char *LnnGetDeviceName(const DeviceBasicInfo *info) = 0; 59 virtual char *LnnConvertIdToDeviceType(uint16_t typeId) = 0; 60 virtual int32_t LnnGetP2pRole(const NodeInfo *info) = 0; 61 virtual const char *LnnGetP2pMac(const NodeInfo *info) = 0; 62 virtual uint64_t LnnGetSupportedProtocols(const NodeInfo *info) = 0; 63 virtual int32_t LnnConvertDeviceTypeToId(const char *deviceType, uint16_t *typeId) = 0; 64 virtual int32_t LnnGetLocalNumInfo(InfoKey key, int32_t *info) = 0; 65 virtual int32_t LnnGetLocalNumU32Info(InfoKey key, uint32_t *info) = 0; 66 virtual bool LnnHasCapability(uint32_t capability, NetCapability type) = 0; 67 virtual int32_t LnnSetNetCapability(uint32_t *capability, NetCapability type) = 0; 68 virtual int32_t LnnClearNetCapability(uint32_t *capability, NetCapability type) = 0; 69 virtual int32_t LnnSetLocalNumInfo(InfoKey key, int32_t info) = 0; 70 virtual int32_t LnnSetLocalStrInfo(InfoKey key, const char *info) = 0; 71 virtual int32_t LnnConvertDlId( 72 const char *srcId, IdCategory srcIdType, IdCategory dstIdType, char *dstIdBuf, uint32_t dstIdBufLen) = 0; 73 virtual bool LnnSetDLDeviceInfoName(const char *udid, const char *name) = 0; 74 virtual bool LnnSetDLP2pInfo(const char *networkId, const P2pInfo *info) = 0; 75 virtual int32_t LnnSetP2pRole(NodeInfo *info, int32_t role) = 0; 76 virtual int32_t LnnSetP2pMac(NodeInfo *info, const char *p2pMac) = 0; 77 virtual int32_t LnnSetP2pGoMac(NodeInfo *info, const char *goMac) = 0; 78 virtual int32_t LnnGetAllOnlineAndMetaNodeInfo(NodeBasicInfo **info, int32_t *infoNum) = 0; 79 virtual int32_t LnnGetAllOnlineNodeInfo(NodeBasicInfo **info, int32_t *infoNum) = 0; 80 virtual bool LnnIsLSANode(const NodeBasicInfo *info) = 0; 81 virtual NodeInfo *LnnGetNodeInfoById(const char *id, IdCategory type) = 0; 82 virtual int32_t LnnGetLnnRelation(const char *id, IdCategory type, uint8_t *relation, uint32_t len) = 0; 83 virtual int32_t LnnSetDLConnCapability(const char *networkId, uint64_t connCapability) = 0; 84 virtual bool LnnHasDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0; 85 virtual bool LnnIsNodeOnline(const NodeInfo *info) = 0; 86 virtual short LnnGetCnnCode(const char *uuid, DiscoveryType type) = 0; 87 virtual ReportCategory LnnAddOnlineNode(NodeInfo *info) = 0; 88 virtual int32_t LnnGetBasicInfoByUdid(const char *udid, NodeBasicInfo *basicInfo) = 0; 89 virtual int32_t LnnInsertSpecificTrustedDevInfo(const char *udid) = 0; 90 virtual int32_t LnnSetSupportDiscoveryType(char *info, const char *type) = 0; 91 virtual bool LnnHasSupportDiscoveryType(const char *destType, const char *type) = 0; 92 virtual bool LnnPeerHasExchangeDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0; 93 virtual const char *LnnGetDeviceUdid(const NodeInfo *info) = 0; 94 virtual ReportCategory LnnSetNodeOffline(const char *udid, ConnectionAddrType type, int32_t authId) = 0; 95 virtual void LnnRemoveNode(const char *udid) = 0; 96 virtual int32_t LnnGetNetworkIdByBtMac(const char *btMac, char *buf, uint32_t len) = 0; 97 virtual int32_t LnnSetLocalNum64Info(InfoKey key, int64_t info) = 0; 98 virtual int32_t LnnGetNodeKeyInfo(const char *networkId, int32_t key, uint8_t *info, uint32_t infoLen) = 0; 99 virtual int32_t LnnGetRemoteNumInfo(const char *netWorkId, InfoKey key, int32_t *info) = 0; 100 virtual int32_t LnnGetLocalDeviceInfo(NodeBasicInfo *info) = 0; 101 virtual int32_t LnnGetLocalByteInfo(InfoKey key, uint8_t *info, uint32_t len) = 0; 102 virtual bool LnnIsDefaultOhosAccount(void) = 0; 103 virtual bool IsPotentialTrustedDevice( 104 TrustedRelationIdType idType, const char *deviceId, bool isPrecise, bool isPointToPoint) = 0; 105 virtual bool LnnIsPotentialHomeGroup(const char *udid) = 0; 106 virtual int32_t LnnGetRemoteNodeInfoById(const char *id, IdCategory type, NodeInfo *info) = 0; 107 virtual int32_t LnnRegisterBleLpDeviceMediumMgr(void) = 0; 108 virtual int32_t LnnGetLocalNumU64Info(InfoKey key, uint64_t *info) = 0; 109 virtual int32_t LnnSetWifiDirectAddr(NodeInfo *info, const char *wifiDirectAddr); 110 virtual const char *LnnGetWifiDirectAddr(const NodeInfo *info) = 0; 111 virtual bool LnnSetDLWifiDirectAddr(const char *networkId, const char *addr) = 0; 112 virtual bool IsActiveOsAccountUnlocked(void) = 0; 113 virtual int32_t GetActiveOsAccountIds(void) = 0; 114 virtual int32_t AuthDeviceGetLatestIdByUuid(const char *uuid, AuthLinkType type, AuthHandle *authHandle) = 0; 115 virtual int32_t AuthGetLatestAuthSeqListByType( 116 const char *udid, int64_t *seqList, uint64_t *authVerifyTime, DiscoveryType type) = 0; 117 virtual int32_t LnnSetDLUnifiedDeviceName(const char *udid, const char *name) = 0; 118 virtual int32_t LnnSetDLUnifiedDefaultDeviceName(const char *udid, const char *name) = 0; 119 virtual int32_t LnnSetDLDeviceNickNameByUdid(const char *udid, const char *name) = 0; 120 virtual int32_t LnnSetDLDeviceStateVersion(const char *udid, int32_t stateVersion) = 0; 121 virtual int32_t LnnUpdateDistributedNodeInfo(NodeInfo *newInfo, const char *udid) = 0; 122 virtual int32_t LnnSetDLDeviceBroadcastCipherKey(const char *udid, const void *cipherKey) = 0; 123 virtual int32_t LnnSetDLDeviceBroadcastCipherIv(const char *udid, const void *cipherIv) = 0; 124 virtual int32_t LnnSetDLBssTransInfo(const char *networkId, const BssTransInfo *info) = 0; 125 virtual int32_t LnnSetDLBatteryInfo(const char *networkId, const BatteryInfo *info) = 0; 126 virtual int32_t LnnGetOsTypeByNetworkId(const char *networkId, int32_t *osType) = 0; 127 virtual bool LnnSetDLDeviceNickName(const char *networkId, const char *name); 128 virtual int32_t LnnUpdateLocalScreenStatus(bool isScreenOn) = 0; 129 }; 130 131 class LnnBtNetworkImplInterfaceMock : public LnnBtNetworkImplInterface { 132 public: 133 LnnBtNetworkImplInterfaceMock(); 134 ~LnnBtNetworkImplInterfaceMock() override; 135 MOCK_METHOD2(LnnRequestLeaveSpecific, int32_t(const char *, ConnectionAddrType)); 136 MOCK_METHOD2(LnnRequestLeaveByAddrType, int32_t(const bool *, uint32_t)); 137 MOCK_METHOD0(SoftBusGetBtState, int32_t(void)); 138 MOCK_METHOD1(SoftBusGetBtMacAddr, int32_t(SoftBusBtAddr *)); 139 MOCK_METHOD4(ConvertBtMacToStr, int32_t(char *, uint32_t, const uint8_t *, uint32_t)); 140 MOCK_METHOD2(LnnRegisterEventHandler, int32_t(LnnEventType, LnnEventHandler)); 141 MOCK_METHOD2(LnnGetNetIfTypeByName, int32_t(const char *, LnnNetIfType *)); 142 MOCK_METHOD2(LnnVisitNetif, bool(VisitNetifCallback, void *)); 143 MOCK_METHOD1(LnnRegistPhysicalSubnet, int32_t(LnnPhysicalSubnet *)); 144 MOCK_METHOD3(LnnNotifyPhysicalSubnetStatusChanged, void(const char *, ProtocolType, void *)); 145 static int32_t ActionOfLnnGetNetIfTypeByNameBr(const char *ifName, LnnNetIfType *type); 146 static int32_t ActionOfLnnGetNetIfTypeByNameBle(const char *ifName, LnnNetIfType *type); 147 MOCK_METHOD3(LnnGetLocalStrInfo, int32_t(InfoKey, char *, uint32_t)); 148 MOCK_METHOD0(LnnGetLocalNodeInfo, const NodeInfo *()); 149 MOCK_METHOD1(LnnGetAuthPort, int32_t(const NodeInfo *)); 150 MOCK_METHOD1(LnnGetSessionPort, int32_t(const NodeInfo *)); 151 MOCK_METHOD1(LnnGetProxyPort, int32_t(const NodeInfo *)); 152 MOCK_METHOD1(LnnGetBtMac, const char *(const NodeInfo *)); 153 MOCK_METHOD1(LnnGetDeviceName, const char *(const DeviceBasicInfo *)); 154 MOCK_METHOD1(LnnConvertIdToDeviceType, char *(uint16_t)); 155 MOCK_METHOD1(LnnGetP2pRole, int32_t(const NodeInfo *)); 156 MOCK_METHOD1(LnnGetP2pMac, const char *(const NodeInfo *)); 157 MOCK_METHOD1(LnnGetWifiDirectAddr, const char *(const NodeInfo *)); 158 MOCK_METHOD1(LnnGetSupportedProtocols, uint64_t(const NodeInfo *)); 159 MOCK_METHOD2(LnnConvertDeviceTypeToId, int32_t(const char *, uint16_t *)); 160 MOCK_METHOD2(LnnGetLocalNumInfo, int32_t(InfoKey, int32_t *)); 161 MOCK_METHOD2(LnnGetLocalNumU32Info, int32_t(InfoKey, uint32_t *)); 162 MOCK_METHOD2(LnnHasCapability, bool(uint32_t, NetCapability)); 163 MOCK_METHOD2(LnnSetNetCapability, int32_t(uint32_t *, NetCapability)); 164 MOCK_METHOD2(LnnClearNetCapability, int32_t(uint32_t *, NetCapability)); 165 MOCK_METHOD2(LnnSetLocalNumInfo, int32_t(InfoKey, int32_t)); 166 MOCK_METHOD2(LnnSetLocalStrInfo, int32_t(InfoKey, const char *)); 167 MOCK_METHOD5(LnnConvertDlId, int32_t(const char *, IdCategory, IdCategory, char *, uint32_t)); 168 MOCK_METHOD2(LnnSetDLDeviceInfoName, bool(const char *, const char *)); 169 MOCK_METHOD2(LnnSetDLP2pInfo, bool(const char *, const P2pInfo *)); 170 MOCK_METHOD2(LnnSetDLWifiDirectAddr, bool(const char *, const char *addr)); 171 MOCK_METHOD2(LnnSetP2pRole, int32_t(NodeInfo *, int32_t)); 172 MOCK_METHOD2(LnnSetP2pMac, int32_t(NodeInfo *, const char *)); 173 MOCK_METHOD2(LnnSetP2pGoMac, int32_t(NodeInfo *, const char *)); 174 MOCK_METHOD2(LnnSetWifiDirectAddr, int32_t(NodeInfo *, const char *)); 175 MOCK_METHOD2(LnnGetAllOnlineAndMetaNodeInfo, int32_t(NodeBasicInfo **, int32_t *)); 176 MOCK_METHOD2(LnnGetAllOnlineNodeInfo, int32_t(NodeBasicInfo **, int32_t *)); 177 MOCK_METHOD1(LnnIsLSANode, bool(const NodeBasicInfo *)); 178 MOCK_METHOD2(LnnGetNodeInfoById, NodeInfo *(const char *, IdCategory)); 179 MOCK_METHOD4(LnnGetLnnRelation, int32_t(const char *, IdCategory, uint8_t *, uint32_t)); 180 MOCK_METHOD2(LnnSetDLConnCapability, int32_t(const char *, uint64_t)); 181 MOCK_METHOD2(LnnHasDiscoveryType, bool(const NodeInfo *, DiscoveryType)); 182 MOCK_METHOD1(LnnIsNodeOnline, bool(const NodeInfo *)); 183 MOCK_METHOD2(LnnGetCnnCode, short(const char *, DiscoveryType)); 184 MOCK_METHOD1(LnnAddOnlineNode, ReportCategory(NodeInfo *)); 185 MOCK_METHOD2(LnnGetBasicInfoByUdid, int32_t(const char *, NodeBasicInfo *)); 186 MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, int32_t(const char *)); 187 MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, char *(const NodeInfo *)); 188 MOCK_METHOD3(LnnSetNodeOffline, ReportCategory(const char *, ConnectionAddrType, int32_t)); 189 MOCK_METHOD1(LnnRemoveNode, void(const char *)); 190 MOCK_METHOD2(LnnSetSupportDiscoveryType, int32_t(char *, const char *)); 191 MOCK_METHOD2(LnnHasSupportDiscoveryType, bool(const char *, const char *)); 192 MOCK_METHOD2(LnnPeerHasExchangeDiscoveryType, bool(const NodeInfo *, DiscoveryType)); 193 MOCK_METHOD1(LnnGetDeviceUdid, const char *(const NodeInfo *)); 194 MOCK_METHOD3(LnnGetNetworkIdByBtMac, int32_t(const char *, char *, uint32_t)); 195 MOCK_METHOD2(LnnSetLocalNum64Info, int32_t(InfoKey, int64_t)); 196 MOCK_METHOD4(LnnGetNodeKeyInfo, int32_t(const char *, int, uint8_t *, uint32_t)); 197 MOCK_METHOD3(LnnGetRemoteNumInfo, int32_t(const char *, InfoKey, int32_t *)); 198 MOCK_METHOD1(LnnGetLocalDeviceInfo, int32_t(NodeBasicInfo *)); 199 MOCK_METHOD3(LnnGetLocalByteInfo, int32_t(InfoKey key, uint8_t *info, uint32_t len)); 200 MOCK_METHOD0(LnnIsDefaultOhosAccount, bool(void)); 201 MOCK_METHOD4(IsPotentialTrustedDevice, 202 bool(TrustedRelationIdType idType, const char *deviceId, bool isPrecise, bool isPointToPoint)); 203 MOCK_METHOD1(LnnIsPotentialHomeGroup, bool(const char *udid)); 204 MOCK_METHOD3(LnnGetRemoteNodeInfoById, int32_t(const char *id, IdCategory type, NodeInfo *info)); 205 MOCK_METHOD0(LnnRegisterBleLpDeviceMediumMgr, int32_t(void)); 206 MOCK_METHOD2(LnnGetLocalNumU64Info, int32_t(InfoKey, uint64_t *)); 207 MOCK_METHOD0(IsActiveOsAccountUnlocked, bool(void)); 208 MOCK_METHOD0(GetActiveOsAccountIds, int32_t(void)); 209 MOCK_METHOD3(AuthDeviceGetLatestIdByUuid, int32_t(const char *, AuthLinkType, AuthHandle *)); 210 MOCK_METHOD4(AuthGetLatestAuthSeqListByType, int32_t(const char *, int64_t *, uint64_t *, DiscoveryType)); 211 MOCK_METHOD2(LnnSetDLUnifiedDeviceName, int32_t(const char *, const char *)); 212 MOCK_METHOD2(LnnSetDLUnifiedDefaultDeviceName, int32_t(const char *, const char *)); 213 MOCK_METHOD2(LnnSetDLDeviceNickNameByUdid, int32_t(const char *, const char *)); 214 MOCK_METHOD2(LnnSetDLDeviceStateVersion, int32_t(const char *, int32_t)); 215 MOCK_METHOD2(LnnUpdateDistributedNodeInfo, int32_t(NodeInfo *, const char *)); 216 MOCK_METHOD2(LnnSetDLDeviceBroadcastCipherKey, int32_t(const char *, const void *)); 217 MOCK_METHOD2(LnnSetDLDeviceBroadcastCipherIv, int32_t(const char *, const void *)); 218 MOCK_METHOD2(LnnSetDLBssTransInfo, int32_t(const char *, const BssTransInfo *)); 219 MOCK_METHOD2(LnnSetDLBatteryInfo, int32_t(const char *, const BatteryInfo *)); 220 MOCK_METHOD2(LnnGetOsTypeByNetworkId, int32_t(const char *, int32_t *)); 221 MOCK_METHOD2(LnnSetDLDeviceNickName, bool(const char *, const char *)); 222 MOCK_METHOD1(LnnUpdateLocalScreenStatus, int32_t(bool)); 223 }; 224 } // namespace OHOS 225 #endif // LNN_BT_NETWORK_IMPL_MOCK_H