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 "disc_interface.h" 23 #include "softbus_conn_interface.h" 24 25 namespace OHOS { 26 class LnnConnectInterface { 27 public: LnnConnectInterface()28 LnnConnectInterface() {}; ~LnnConnectInterface()29 virtual ~LnnConnectInterface() {}; 30 31 virtual int32_t ConnGetConnectionInfo(uint32_t connectionId, ConnectionInfo *info) = 0; 32 virtual int32_t ConnSetConnectCallback(ConnModule moduleId, const ConnectCallback *callback) = 0; 33 virtual void ConnUnSetConnectCallback(ConnModule moduleId) = 0; 34 virtual int32_t ConnConnectDevice(const ConnectOption *option, uint32_t requestId, const ConnectResult *result) = 0; 35 virtual int32_t ConnDisconnectDevice(uint32_t connectionId) = 0; 36 virtual uint32_t ConnGetHeadSize(void) = 0; 37 virtual int32_t ConnPostBytes(uint32_t connectionId, ConnPostData *data) = 0; 38 virtual bool CheckActiveConnection(const ConnectOption *option) = 0; 39 virtual int32_t ConnStartLocalListening(const LocalListenerInfo *info) = 0; 40 virtual int32_t ConnStopLocalListening(const LocalListenerInfo *info) = 0; 41 virtual uint32_t ConnGetNewRequestId(ConnModule moduleId) = 0; 42 virtual void DiscDeviceInfoChanged(InfoTypeChanged type) = 0; 43 virtual int32_t ConnUpdateConnection(uint32_t connectionId, UpdateOption *option) = 0; 44 }; 45 class LnnConnectInterfaceMock : public LnnConnectInterface { 46 public: 47 LnnConnectInterfaceMock(); 48 ~LnnConnectInterfaceMock() override; 49 MOCK_METHOD2(ConnGetConnectionInfo, int32_t (uint32_t, ConnectionInfo *)); 50 MOCK_METHOD2(ConnSetConnectCallback, int32_t (ConnModule, const ConnectCallback *)); 51 MOCK_METHOD1(ConnUnSetConnectCallback, void (ConnModule)); 52 MOCK_METHOD3(ConnConnectDevice, int32_t (const ConnectOption *, uint32_t, const ConnectResult *)); 53 MOCK_METHOD1(ConnDisconnectDevice, int32_t (uint32_t)); 54 MOCK_METHOD0(ConnGetHeadSize, uint32_t (void)); 55 MOCK_METHOD2(ConnPostBytes, int32_t (uint32_t, ConnPostData *)); 56 MOCK_METHOD1(CheckActiveConnection, bool (const ConnectOption *)); 57 MOCK_METHOD1(ConnStartLocalListening, int32_t (const LocalListenerInfo *)); 58 MOCK_METHOD1(ConnStopLocalListening, int32_t (const LocalListenerInfo *)); 59 MOCK_METHOD1(ConnGetNewRequestId, uint32_t (ConnModule)); 60 MOCK_METHOD1(DiscDeviceInfoChanged, void (InfoTypeChanged)); 61 MOCK_METHOD2(ConnUpdateConnection, int32_t(uint32_t, UpdateOption *)); 62 }; 63 } // namespace OHOS 64 #endif // AUTH_CONNECTION_MOCK_H