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_NET_LEDGER_MOCK_H 17 #define LNN_NET_LEDGER_MOCK_H 18 19 #include <gmock/gmock.h> 20 #include <mutex> 21 22 #include "bus_center_manager.h" 23 #include "lnn_node_info.h" 24 #include "lnn_distributed_net_ledger.h" 25 #include "bus_center_info_key.h" 26 #include "lnn_device_info.h" 27 #include "softbus_bus_center.h" 28 #include "bus_center_event.h" 29 30 namespace OHOS { 31 class LnnNetLedgerInterface { 32 public: LnnNetLedgerInterface()33 LnnNetLedgerInterface() {}; ~LnnNetLedgerInterface()34 virtual ~LnnNetLedgerInterface() {}; 35 virtual int32_t LnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len) = 0; 36 virtual const NodeInfo *LnnGetLocalNodeInfo(void) = 0; 37 virtual int32_t LnnGetAuthPort(const NodeInfo *info) = 0; 38 virtual int32_t LnnGetSessionPort(const NodeInfo *info) = 0; 39 virtual int32_t LnnGetProxyPort(const NodeInfo *info) = 0; 40 virtual const char *LnnGetBtMac(const NodeInfo *info) = 0; 41 virtual const char *LnnGetDeviceName(const DeviceBasicInfo *info) = 0; 42 virtual char *LnnConvertIdToDeviceType(uint16_t typeId) = 0; 43 virtual int32_t LnnGetP2pRole(const NodeInfo *info) = 0; 44 virtual const char *LnnGetP2pMac(const NodeInfo *info) = 0; 45 virtual uint64_t LnnGetSupportedProtocols(const NodeInfo *info) = 0; 46 virtual int32_t LnnConvertDeviceTypeToId(const char *deviceType, uint16_t *typeId) = 0; 47 virtual int32_t LnnGetLocalNumInfo(InfoKey key, int32_t *info) = 0; 48 virtual int32_t LnnSetNetCapability(uint32_t *capability, NetCapability type) = 0; 49 virtual int32_t LnnClearNetCapability(uint32_t *capability, NetCapability type) = 0; 50 virtual int32_t LnnSetLocalNumInfo(InfoKey key, int32_t info) = 0; 51 virtual int32_t LnnSetLocalStrInfo(InfoKey key, const char *info) = 0; 52 virtual int32_t LnnConvertDlId(const char *srcId, IdCategory srcIdType, IdCategory dstIdType, 53 char *dstIdBuf, uint32_t dstIdBufLen) = 0; 54 virtual bool LnnSetDLDeviceInfoName(const char *udid, const char *name) = 0; 55 virtual bool LnnSetDLP2pInfo(const char *networkId, const P2pInfo *info) = 0; 56 virtual int32_t LnnSetP2pRole(NodeInfo *info, int32_t role) = 0; 57 virtual int32_t LnnSetP2pMac(NodeInfo *info, const char *p2pMac) = 0; 58 virtual int32_t LnnSetP2pGoMac(NodeInfo *info, const char *goMac) = 0; 59 virtual int32_t LnnGetAllOnlineAndMetaNodeInfo(NodeBasicInfo **info, int32_t *infoNum) = 0; 60 virtual int32_t LnnGetAllOnlineNodeInfo(NodeBasicInfo **info, int32_t *infoNum) = 0; 61 virtual NodeInfo *LnnGetNodeInfoById(const char *id, IdCategory type) = 0; 62 virtual int32_t LnnGetLnnRelation(const char *id, IdCategory type, uint8_t *relation, uint32_t len) = 0; 63 virtual int32_t LnnSetDLConnCapability(const char *networkId, uint64_t connCapability) = 0; 64 virtual bool LnnHasDiscoveryType(const NodeInfo *info, DiscoveryType type) = 0; 65 virtual bool LnnIsNodeOnline(const NodeInfo *info) = 0; 66 virtual short LnnGetCnnCode(const char *uuid, DiscoveryType type) = 0; 67 virtual ReportCategory LnnAddOnlineNode(NodeInfo *info) = 0; 68 virtual int32_t LnnGetBasicInfoByUdid(const char *udid, NodeBasicInfo *basicInfo) = 0; 69 virtual int32_t LnnInsertSpecificTrustedDevInfo(const char *udid) = 0; 70 virtual const char *LnnGetDeviceUdid(const NodeInfo *info) = 0; 71 virtual ReportCategory LnnSetNodeOffline(const char *udid, ConnectionAddrType type, int32_t authId) = 0; 72 virtual void LnnRemoveNode(const char *udid) = 0; 73 }; 74 class LnnNetLedgertInterfaceMock : public LnnNetLedgerInterface { 75 public: 76 LnnNetLedgertInterfaceMock(); 77 ~LnnNetLedgertInterfaceMock() override; 78 MOCK_METHOD3(LnnGetLocalStrInfo, int32_t (InfoKey, char *, uint32_t)); 79 MOCK_METHOD0(LnnGetLocalNodeInfo, const NodeInfo *()); 80 MOCK_METHOD1(LnnGetAuthPort, int32_t (const NodeInfo *)); 81 MOCK_METHOD1(LnnGetSessionPort, int32_t (const NodeInfo *)); 82 MOCK_METHOD1(LnnGetProxyPort, int32_t (const NodeInfo *)); 83 MOCK_METHOD1(LnnGetBtMac, const char *(const NodeInfo *)); 84 MOCK_METHOD1(LnnGetDeviceName, const char *(const DeviceBasicInfo *)); 85 MOCK_METHOD1(LnnConvertIdToDeviceType, char *(uint16_t)); 86 MOCK_METHOD1(LnnGetP2pRole, int32_t (const NodeInfo *)); 87 MOCK_METHOD1(LnnGetP2pMac, const char *(const NodeInfo *)); 88 MOCK_METHOD1(LnnGetSupportedProtocols, uint64_t (const NodeInfo *)); 89 MOCK_METHOD2(LnnConvertDeviceTypeToId, int32_t (const char *, uint16_t *)); 90 MOCK_METHOD2(LnnGetLocalNumInfo, int32_t (InfoKey, int32_t *)); 91 MOCK_METHOD2(LnnSetNetCapability, int32_t (uint32_t *, NetCapability)); 92 MOCK_METHOD2(LnnClearNetCapability, int32_t (uint32_t *, NetCapability)); 93 MOCK_METHOD2(LnnSetLocalNumInfo, int32_t (InfoKey, int32_t)); 94 MOCK_METHOD2(LnnSetLocalStrInfo, int32_t (InfoKey, const char *)); 95 MOCK_METHOD5(LnnConvertDlId, int32_t (const char *, IdCategory, IdCategory, char *, uint32_t)); 96 MOCK_METHOD2(LnnSetDLDeviceInfoName, bool (const char *, const char *)); 97 MOCK_METHOD2(LnnSetDLP2pInfo, bool (const char *, const P2pInfo *)); 98 MOCK_METHOD2(LnnSetP2pRole, int32_t (NodeInfo *, int32_t)); 99 MOCK_METHOD2(LnnSetP2pMac, int32_t (NodeInfo *, const char *)); 100 MOCK_METHOD2(LnnSetP2pGoMac, int32_t (NodeInfo *, const char *)); 101 MOCK_METHOD2(LnnGetAllOnlineAndMetaNodeInfo, int32_t (NodeBasicInfo **, int32_t *)); 102 MOCK_METHOD2(LnnGetAllOnlineNodeInfo, int32_t (NodeBasicInfo **, int32_t *)); 103 MOCK_METHOD2(LnnGetNodeInfoById, NodeInfo *(const char *, IdCategory)); 104 MOCK_METHOD4(LnnGetLnnRelation, int32_t(const char *, IdCategory, uint8_t *, uint32_t)); 105 MOCK_METHOD2(LnnSetDLConnCapability, int32_t (const char *, uint64_t)); 106 MOCK_METHOD2(LnnHasDiscoveryType, bool (const NodeInfo *, DiscoveryType)); 107 MOCK_METHOD1(LnnIsNodeOnline, bool (const NodeInfo *)); 108 MOCK_METHOD2(LnnGetCnnCode, short (const char *, DiscoveryType)); 109 MOCK_METHOD1(LnnAddOnlineNode, ReportCategory (NodeInfo *)); 110 MOCK_METHOD2(LnnGetBasicInfoByUdid, int32_t (const char *, NodeBasicInfo *)); 111 MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, int32_t (const char *)); 112 MOCK_METHOD1(LnnInsertSpecificTrustedDevInfo, char * (const NodeInfo *)); 113 MOCK_METHOD3(LnnSetNodeOffline, ReportCategory (const char *, ConnectionAddrType, int32_t)); 114 MOCK_METHOD1(LnnRemoveNode, void (const char *)); 115 MOCK_METHOD1(LnnGetDeviceUdid, const char *(const NodeInfo *)); 116 static int32_t ActionOfLnnGetAllOnline(NodeBasicInfo **info, int32_t *infoNum); 117 static int32_t ActionOfLnnConvertDlId(const char *srcId, IdCategory srcIdType, IdCategory dstIdType, 118 char *dstIdBuf, uint32_t dstIdBufLen); 119 static int32_t ActionOfLnnConvertDlId1(const char *srcId, IdCategory srcIdType, IdCategory dstIdType, 120 char *dstIdBuf, uint32_t dstIdBufLen); 121 static inline std::map<LnnEventType, LnnEventHandler> g_lnnevent_handlers; 122 static int32_t ActionOfLnnGetAllOnlineNodeInfo(NodeBasicInfo **info, int32_t *infoNum); 123 static int32_t ActionOfLnnGetLnnRelation(const char *id, IdCategory type, 124 uint8_t *relation, uint32_t len); 125 static int32_t ActionOfLnnGetLocalStrInfo(InfoKey key, char *info, uint32_t len); 126 }; 127 } // namespace OHOS 128 #endif // LNN_NET_LEDGER_MOCK_H