1 /* 2 * Copyright (c) 2025 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_USB_NETWORK_IMPL_MOCK_H 17 #define LNN_USB_NETWORK_IMPL_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 "lnn_distributed_net_ledger.h" 25 #include "lnn_network_manager.h" 26 #include "lnn_node_info.h" 27 #include "lnn_physical_subnet_manager.h" 28 #include "softbus_bus_center.h" 29 #include "softbus_conn_interface.h" 30 31 namespace OHOS { 32 class LnnUsbNetworkImplInterface { 33 public: LnnUsbNetworkImplInterface()34 LnnUsbNetworkImplInterface() {}; ~LnnUsbNetworkImplInterface()35 virtual ~LnnUsbNetworkImplInterface() {}; 36 virtual int32_t LnnRegisterEventHandler(LnnEventType event, LnnEventHandler handler) = 0; 37 virtual int32_t LnnRegistPhysicalSubnet(LnnPhysicalSubnet *manager) = 0; 38 virtual void DiscLinkStatusChanged(LinkStatus status, ExchangeMedium medium, int32_t ifnameIdx) = 0; 39 virtual bool LnnVisitPhysicalSubnet(LnnVisitPhysicalSubnetCallback callback, void *data) = 0; 40 virtual void LnnIpAddrChangeEventHandler(void) = 0; 41 virtual void AuthStopListening(AuthLinkType type) = 0; 42 virtual int32_t TransTdcStopSessionListener(ListenerModule module) = 0; 43 virtual int32_t LnnGetAddrTypeByIfName(const char *ifName, ConnectionAddrType *type) = 0; 44 virtual int32_t AuthStartListening(AuthLinkType type, const char *ip, int32_t port) = 0; 45 virtual int32_t TransTdcStartSessionListener(ListenerModule module, const LocalListenerInfo *info) = 0; 46 virtual void LnnNotifyPhysicalSubnetStatusChanged(const char *ifName, ProtocolType protocolType, void *status) = 0; 47 virtual bool LnnVisitNetif(VisitNetifCallback callback, void *data) = 0; 48 virtual int32_t LnnRequestLeaveByAddrType(const bool *type, uint32_t typeLen) = 0; 49 virtual int32_t GetNetworkIpv6ByIfName(const char *ifName, char *ip, uint32_t len) = 0; 50 virtual int32_t LnnRegistProtocol(LnnProtocolManager *protocolMgr) = 0; 51 virtual int32_t LnnGetLocalNumU32Info(InfoKey key, uint32_t *info) = 0; 52 virtual int32_t LnnSetNetCapability(uint32_t *capability, NetCapability type) = 0; 53 virtual int32_t LnnClearNetCapability(uint32_t *capability, NetCapability type) = 0; 54 virtual int32_t LnnSetLocalNumInfo(InfoKey key, int32_t info) = 0; 55 virtual int32_t LnnGetLocalStrInfoByIfnameIdx(InfoKey key, char *info, uint32_t len, int32_t ifIdx) = 0; 56 virtual int32_t LnnGetLocalNumInfoByIfnameIdx(InfoKey key, int32_t *info, int32_t ifIdx) = 0; 57 virtual int32_t LnnSetLocalStrInfoByIfnameIdx(InfoKey key, const char *info, int32_t ifIdx) = 0; 58 virtual int32_t LnnSetLocalNumInfoByIfnameIdx(InfoKey key, int32_t info, int32_t ifIdx) = 0; 59 }; 60 61 class LnnUsbNetworkImplInterfaceMock : public LnnUsbNetworkImplInterface { 62 public: 63 LnnUsbNetworkImplInterfaceMock(); 64 ~LnnUsbNetworkImplInterfaceMock() override; 65 MOCK_METHOD2(LnnRegisterEventHandler, int32_t(LnnEventType, LnnEventHandler)); 66 MOCK_METHOD1(LnnRegistPhysicalSubnet, int32_t(LnnPhysicalSubnet *)); 67 MOCK_METHOD3(DiscLinkStatusChanged, void(LinkStatus, ExchangeMedium, int32_t)); 68 MOCK_METHOD2(LnnVisitPhysicalSubnet, bool(LnnVisitPhysicalSubnetCallback, void *)); 69 MOCK_METHOD0(LnnIpAddrChangeEventHandler, void(void)); 70 MOCK_METHOD1(AuthStopListening, void(AuthLinkType)); 71 MOCK_METHOD1(TransTdcStopSessionListener, int32_t(ListenerModule)); 72 MOCK_METHOD2(LnnGetAddrTypeByIfName, int32_t(const char *, ConnectionAddrType *)); 73 MOCK_METHOD3(AuthStartListening, int32_t(AuthLinkType, const char *, int32_t)); 74 MOCK_METHOD2(TransTdcStartSessionListener, int32_t(ListenerModule, const LocalListenerInfo *)); 75 MOCK_METHOD3(LnnNotifyPhysicalSubnetStatusChanged, void(const char *, ProtocolType, void *)); 76 MOCK_METHOD2(LnnVisitNetif, bool(VisitNetifCallback, void *)); 77 MOCK_METHOD2(LnnRequestLeaveByAddrType, int32_t(const bool *, uint32_t)); 78 MOCK_METHOD3(GetNetworkIpv6ByIfName, int32_t(const char *, char *, uint32_t)); 79 MOCK_METHOD1(LnnRegistProtocol, int32_t(LnnProtocolManager *)); 80 MOCK_METHOD2(LnnGetLocalNumU32Info, int32_t(InfoKey, uint32_t *)); 81 MOCK_METHOD2(LnnSetNetCapability, int32_t(uint32_t *, NetCapability)); 82 MOCK_METHOD2(LnnClearNetCapability, int32_t(uint32_t *, NetCapability)); 83 MOCK_METHOD2(LnnSetLocalNumInfo, int32_t(InfoKey, int32_t)); 84 MOCK_METHOD4(LnnGetLocalStrInfoByIfnameIdx, int32_t(InfoKey, char *, uint32_t, int32_t)); 85 MOCK_METHOD3(LnnGetLocalNumInfoByIfnameIdx, int32_t(InfoKey, int32_t *, int32_t)); 86 MOCK_METHOD3(LnnSetLocalStrInfoByIfnameIdx, int32_t(InfoKey, const char *, int32_t)); 87 MOCK_METHOD3(LnnSetLocalNumInfoByIfnameIdx, int32_t(InfoKey, int32_t, int32_t)); 88 static int32_t ActionOfGetNetworkIpv6ByIfName(const char *ifName, char *ip, uint32_t len); 89 static int32_t ActionOfGetNetworkIpv6ByIfName2(const char *ifName, char *ip, uint32_t len); 90 static int32_t ActionOfLnnGetLocalStrInfoByIfnameIdx(InfoKey key, char *info, uint32_t len, int32_t ifIdx); 91 }; 92 } // namespace OHOS 93 #endif // LNN_USB_NETWORK_IMPL_MOCK_H