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_CONNECTION_MOCK_H 17 #define LNN_CONNECTION_MOCK_H 18 19 #include <gmock/gmock.h> 20 #include <mutex> 21 22 #include "auth_common.h" 23 #include "disc_interface.h" 24 #include "lnn_node_info.h" 25 #include "map" 26 #include "securec.h" 27 #include "softbus_adapter_mem.h" 28 #include "softbus_conn_interface.h" 29 30 namespace OHOS { 31 class LnnConnectInterface { 32 public: LnnConnectInterface()33 LnnConnectInterface() {}; ~LnnConnectInterface()34 virtual ~LnnConnectInterface() {}; 35 36 virtual int32_t ConnGetConnectionInfo(uint32_t connectionId, ConnectionInfo *info) = 0; 37 virtual int32_t ConnSetConnectCallback(ConnModule moduleId, const ConnectCallback *callback) = 0; 38 virtual void ConnUnSetConnectCallback(ConnModule moduleId) = 0; 39 virtual int32_t ConnConnectDevice( 40 const ConnectOption *option, uint32_t requestId, const ConnectResult *result) = 0; 41 virtual int32_t ConnDisconnectDevice(uint32_t connectionId) = 0; 42 virtual uint32_t ConnGetHeadSize(void) = 0; 43 virtual int32_t ConnPostBytes(uint32_t connectionId, ConnPostData *data) = 0; 44 virtual bool CheckActiveConnection(const ConnectOption *option, bool needOccupy) = 0; 45 virtual int32_t ConnStartLocalListening(const LocalListenerInfo *info) = 0; 46 virtual int32_t ConnStopLocalListening(const LocalListenerInfo *info) = 0; 47 virtual uint32_t ConnGetNewRequestId(ConnModule moduleId) = 0; 48 virtual void DiscDeviceInfoChanged(InfoTypeChanged type) = 0; 49 virtual int32_t ConnUpdateConnection(uint32_t connectionId, UpdateOption *option) = 0; 50 }; 51 class LnnConnectInterfaceMock : public LnnConnectInterface { 52 public: 53 LnnConnectInterfaceMock(); 54 ~LnnConnectInterfaceMock() override; 55 MOCK_METHOD2(ConnGetConnectionInfo, int32_t(uint32_t, ConnectionInfo *)); 56 MOCK_METHOD2(ConnSetConnectCallback, int32_t(ConnModule, const ConnectCallback *)); 57 MOCK_METHOD1(ConnUnSetConnectCallback, void(ConnModule)); 58 MOCK_METHOD3(ConnConnectDevice, int32_t(const ConnectOption *, uint32_t, const ConnectResult *)); 59 MOCK_METHOD1(ConnDisconnectDevice, int32_t(uint32_t)); 60 MOCK_METHOD0(ConnGetHeadSize, uint32_t(void)); 61 MOCK_METHOD2(ConnPostBytes, int32_t(uint32_t, ConnPostData *)); 62 MOCK_METHOD2(CheckActiveConnection, bool(const ConnectOption *, bool)); 63 MOCK_METHOD1(ConnStartLocalListening, int32_t(const LocalListenerInfo *)); 64 MOCK_METHOD1(ConnStopLocalListening, int32_t(const LocalListenerInfo *)); 65 MOCK_METHOD1(ConnGetNewRequestId, uint32_t(ConnModule)); 66 MOCK_METHOD1(DiscDeviceInfoChanged, void(InfoTypeChanged)); 67 MOCK_METHOD2(ConnUpdateConnection, int32_t(uint32_t, UpdateOption *)); 68 static inline char *g_encryptData; 69 static inline ConnectCallback g_conncallback; 70 static inline ConnectResult g_connresultcb; 71 static int32_t ActionOfConnPostBytes(uint32_t connectionId, ConnPostData *data); 72 static int32_t ActionofConnSetConnectCallback(ConnModule moduleId, const ConnectCallback *callback); 73 static int32_t ActionofOnConnectSuccessed( 74 const ConnectOption *option, uint32_t requestId, const ConnectResult *result); 75 static int32_t ActionofOnConnectFailed( 76 const ConnectOption *option, uint32_t requestId, const ConnectResult *result); 77 static int32_t ActionofConnGetConnectionInfo(uint32_t connectionId, ConnectionInfo *info); 78 static void ActionofConnUnSetConnectCallback(ConnModule moduleId); OnVerifyPassed(uint32_t requestId,AuthHandle authHandle,const NodeInfo * info)79 static void OnVerifyPassed(uint32_t requestId, AuthHandle authHandle, const NodeInfo *info) 80 { 81 (void)requestId; 82 (void)authHandle; 83 (void)info; 84 return; 85 } onVerifyFailed(uint32_t requestId,int32_t reason)86 static void onVerifyFailed(uint32_t requestId, int32_t reason) 87 { 88 (void)requestId; 89 (void)reason; 90 return; 91 } onConnOpened(uint32_t requestId,AuthHandle authHandle)92 static void onConnOpened(uint32_t requestId, AuthHandle authHandle) 93 { 94 (void)requestId; 95 (void)authHandle; 96 return; 97 } onConnOpenFailed(uint32_t requestId,int32_t reason)98 static void onConnOpenFailed(uint32_t requestId, int32_t reason) 99 { 100 (void)requestId; 101 (void)reason; 102 return; 103 } 104 }; 105 } // namespace OHOS 106 #endif // AUTH_CONNECTION_MOCK_H