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