1 /* 2 * Copyright (c) 2023 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 USB_IMPL_MOCK_H 17 #define USB_IMPL_MOCK_H 18 19 #include <gmock/gmock.h> 20 #include <gtest/gtest.h> 21 22 #include "v1_0/iusb_interface.h" 23 #include "v1_0/usb_types.h" 24 25 using OHOS::HDI::Usb::V1_0::IUsbdBulkCallback; 26 using OHOS::HDI::Usb::V1_0::IUsbdSubscriber; 27 using OHOS::HDI::Usb::V1_0::IUsbInterface; 28 using OHOS::HDI::Usb::V1_0::PortInfo; 29 using OHOS::HDI::Usb::V1_0::UsbCtrlTransfer; 30 using OHOS::HDI::Usb::V1_0::UsbDev; 31 using OHOS::HDI::Usb::V1_0::USBDeviceInfo; 32 using OHOS::HDI::Usb::V1_0::UsbPipe; 33 34 namespace OHOS { 35 namespace USB { 36 constexpr int32_t DEV_ADDR_OK = 2; 37 constexpr int32_t DEV_ADDR_OK_2 = 3; 38 constexpr int32_t DEV_ADDR_OK_ERR_DESC = 4; 39 constexpr int32_t DEV_ADDR_OK_NULL_DESC = 5; 40 41 constexpr int32_t BUS_NUM_OK = 6; 42 constexpr int32_t BUS_NUM_OK_2 = 7; 43 constexpr int32_t DEV_ADDR_INVALID = 255; 44 constexpr int32_t BUS_NUM_INVALID = 255; 45 constexpr int32_t DEFAULT_PORT_ID = 1; 46 constexpr int32_t RETVAL_INVALID = -1; 47 constexpr int32_t BUS_NUM_ERR = 8; 48 constexpr int32_t DEV_ADDR_ERR = 8; 49 constexpr int32_t DEV_ADDR_INTERFACE_ERR = 9; 50 51 class UsbImplMock : public IUsbInterface { 52 public: 53 UsbImplMock(); 54 ~UsbImplMock(); 55 int32_t SubscriberDeviceEvent(const USBDeviceInfo &info); 56 int32_t BindUsbdSubscriber(const sptr<IUsbdSubscriber> &subscriber) override; 57 int32_t UnbindUsbdSubscriber(const sptr<IUsbdSubscriber> &subscriber) override; 58 int32_t GetDeviceDescriptor(const UsbDev& dev, std::vector<uint8_t>& descriptor) override; 59 int32_t GetRawDescriptor(const UsbDev &dev, std::vector<uint8_t> &decriptor) override; 60 int32_t GetStringDescriptor(const UsbDev &dev, uint8_t descId, std::vector<uint8_t> &decriptor) override; 61 int32_t QueryPort(int32_t &portId, int32_t &powerRole, int32_t &dataRole, int32_t &mode) override; 62 int32_t SetPortRole(int32_t portId, int32_t powerRole, int32_t dataRole) override; 63 int32_t GetConfig(const UsbDev &dev, uint8_t &configIndex) override; 64 int32_t OpenDevice(const UsbDev &dev) override; 65 MOCK_METHOD1(CloseDevice, int32_t(const UsbDev &dev)); 66 MOCK_METHOD3(GetConfigDescriptor, int32_t(const UsbDev &dev, uint8_t descId, std::vector<uint8_t> &decriptor)); 67 MOCK_METHOD2(GetFileDescriptor, int32_t(const UsbDev &dev, int32_t &fd)); 68 MOCK_METHOD2(SetConfig, int32_t(const UsbDev &dev, uint8_t configIndex)); 69 MOCK_METHOD3(ClaimInterface, int32_t(const UsbDev &dev, uint8_t interfaceid, uint8_t force)); 70 MOCK_METHOD2(ReleaseInterface, int32_t(const UsbDev &dev, uint8_t interfaceid)); 71 MOCK_METHOD3(SetInterface, int32_t(const UsbDev &dev, uint8_t interfaceid, uint8_t altIndex)); 72 MOCK_METHOD4( 73 BulkTransferRead, int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, std::vector<uint8_t> &data)); 74 MOCK_METHOD4(BulkTransferWrite, 75 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, const std::vector<uint8_t> &data)); 76 MOCK_METHOD3( 77 ControlTransferRead, int32_t(const UsbDev &dev, const UsbCtrlTransfer &ctrl, std::vector<uint8_t> &data)); 78 MOCK_METHOD3(ControlTransferWrite, 79 int32_t(const UsbDev &dev, const UsbCtrlTransfer &ctrl, const std::vector<uint8_t> &data)); 80 MOCK_METHOD4(InterruptTransferRead, 81 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, std::vector<uint8_t> &data)); 82 MOCK_METHOD4(InterruptTransferWrite, 83 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, const std::vector<uint8_t> &data)); 84 MOCK_METHOD4( 85 IsoTransferRead, int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, std::vector<uint8_t> &data)); 86 MOCK_METHOD4(IsoTransferWrite, 87 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, const std::vector<uint8_t> &data)); 88 MOCK_METHOD4(RequestQueue, 89 int32_t(const UsbDev &dev, const UsbPipe &pipe, const std::vector<uint8_t> &clientData, 90 const std::vector<uint8_t> &buffer)); 91 MOCK_METHOD4(RequestWait, 92 int32_t(const UsbDev &dev, std::vector<uint8_t> &clientData, std::vector<uint8_t> &buffer, int32_t timeout)); 93 MOCK_METHOD2(RequestCancel, int32_t(const UsbDev &dev, const UsbPipe &pipe)); 94 MOCK_METHOD1(GetCurrentFunctions, int32_t(int32_t &funcs)); 95 MOCK_METHOD1(SetCurrentFunctions, int32_t(int32_t funcs)); 96 MOCK_METHOD3(RegBulkCallback, int32_t(const UsbDev &dev, const UsbPipe &pipe, const sptr<IUsbdBulkCallback> &cb)); 97 MOCK_METHOD2(UnRegBulkCallback, int32_t(const UsbDev &dev, const UsbPipe &pipe)); 98 MOCK_METHOD3(BulkRead, int32_t(const UsbDev &dev, const UsbPipe &pipe, const sptr<Ashmem> &ashmem)); 99 MOCK_METHOD3(BulkWrite, int32_t(const UsbDev &dev, const UsbPipe &pipe, const sptr<Ashmem> &ashmem)); 100 MOCK_METHOD2(BulkCancel, int32_t(const UsbDev &dev, const UsbPipe &pipe)); 101 MOCK_METHOD3(ManageInterface, int32_t(const UsbDev &dev, uint8_t interfaceid, bool disable)); 102 MOCK_METHOD3(GetInterfaceActiveStatus, int32_t(const UsbDev &dev, uint8_t interfaceid, bool &unactivated)); 103 MOCK_METHOD2(GetDeviceSpeed, int32_t(const UsbDev &dev, uint8_t &speed)); 104 private: 105 sptr<IUsbdSubscriber> subscriber_ = nullptr; 106 }; 107 } // namespace USB 108 } // namespace OHOS 109 #endif 110