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 48 class UsbImplMock : public IUsbInterface { 49 public: 50 UsbImplMock(); 51 ~UsbImplMock(); 52 int32_t SubscriberDeviceEvent(const USBDeviceInfo &info); 53 int32_t BindUsbdSubscriber(const sptr<IUsbdSubscriber> &subscriber) override; 54 int32_t UnbindUsbdSubscriber(const sptr<IUsbdSubscriber> &subscriber) override; 55 int32_t GetDeviceDescriptor(const UsbDev& dev, std::vector<uint8_t>& descriptor) override; 56 int32_t GetRawDescriptor(const UsbDev &dev, std::vector<uint8_t> &decriptor) override; 57 int32_t GetStringDescriptor(const UsbDev &dev, uint8_t descId, std::vector<uint8_t> &decriptor) override; 58 int32_t QueryPort(int32_t &portId, int32_t &powerRole, int32_t &dataRole, int32_t &mode) override; 59 int32_t SetPortRole(int32_t portId, int32_t powerRole, int32_t dataRole) override; 60 int32_t GetConfig(const UsbDev &dev, uint8_t &configIndex) override; 61 MOCK_METHOD1(OpenDevice, int32_t(const UsbDev &dev)); 62 MOCK_METHOD1(CloseDevice, int32_t(const UsbDev &dev)); 63 MOCK_METHOD3(GetConfigDescriptor, int32_t(const UsbDev &dev, uint8_t descId, std::vector<uint8_t> &decriptor)); 64 MOCK_METHOD2(GetFileDescriptor, int32_t(const UsbDev &dev, int32_t &fd)); 65 MOCK_METHOD2(SetConfig, int32_t(const UsbDev &dev, uint8_t configIndex)); 66 MOCK_METHOD3(ClaimInterface, int32_t(const UsbDev &dev, uint8_t interfaceid, uint8_t force)); 67 MOCK_METHOD2(ReleaseInterface, int32_t(const UsbDev &dev, uint8_t interfaceid)); 68 MOCK_METHOD3(SetInterface, int32_t(const UsbDev &dev, uint8_t interfaceid, uint8_t altIndex)); 69 MOCK_METHOD4( 70 BulkTransferRead, int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, std::vector<uint8_t> &data)); 71 MOCK_METHOD4(BulkTransferWrite, 72 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, const std::vector<uint8_t> &data)); 73 MOCK_METHOD3( 74 ControlTransferRead, int32_t(const UsbDev &dev, const UsbCtrlTransfer &ctrl, std::vector<uint8_t> &data)); 75 MOCK_METHOD3(ControlTransferWrite, 76 int32_t(const UsbDev &dev, const UsbCtrlTransfer &ctrl, const std::vector<uint8_t> &data)); 77 MOCK_METHOD4(InterruptTransferRead, 78 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, std::vector<uint8_t> &data)); 79 MOCK_METHOD4(InterruptTransferWrite, 80 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, const std::vector<uint8_t> &data)); 81 MOCK_METHOD4( 82 IsoTransferRead, int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, std::vector<uint8_t> &data)); 83 MOCK_METHOD4(IsoTransferWrite, 84 int32_t(const UsbDev &dev, const UsbPipe &pipe, int32_t timeout, const std::vector<uint8_t> &data)); 85 MOCK_METHOD4(RequestQueue, 86 int32_t(const UsbDev &dev, const UsbPipe &pipe, const std::vector<uint8_t> &clientData, 87 const std::vector<uint8_t> &buffer)); 88 MOCK_METHOD4(RequestWait, 89 int32_t(const UsbDev &dev, std::vector<uint8_t> &clientData, std::vector<uint8_t> &buffer, int32_t timeout)); 90 MOCK_METHOD2(RequestCancel, int32_t(const UsbDev &dev, const UsbPipe &pipe)); 91 MOCK_METHOD1(GetCurrentFunctions, int32_t(int32_t &funcs)); 92 MOCK_METHOD1(SetCurrentFunctions, int32_t(int32_t funcs)); 93 MOCK_METHOD3(RegBulkCallback, int32_t(const UsbDev &dev, const UsbPipe &pipe, const sptr<IUsbdBulkCallback> &cb)); 94 MOCK_METHOD2(UnRegBulkCallback, int32_t(const UsbDev &dev, const UsbPipe &pipe)); 95 MOCK_METHOD3(BulkRead, int32_t(const UsbDev &dev, const UsbPipe &pipe, const sptr<Ashmem> &ashmem)); 96 MOCK_METHOD3(BulkWrite, int32_t(const UsbDev &dev, const UsbPipe &pipe, const sptr<Ashmem> &ashmem)); 97 MOCK_METHOD2(BulkCancel, int32_t(const UsbDev &dev, const UsbPipe &pipe)); 98 MOCK_METHOD3(ManageInterface, int32_t(const UsbDev &dev, uint8_t interfaceid, bool disable)); 99 MOCK_METHOD3(GetInterfaceActiveStatus, int32_t(const UsbDev &dev, uint8_t interfaceid, bool &unactivated)); 100 MOCK_METHOD2(GetDeviceSpeed, int32_t(const UsbDev &dev, uint8_t &speed)); 101 private: 102 sptr<IUsbdSubscriber> subscriber_ = nullptr; 103 }; 104 } // namespace USB 105 } // namespace OHOS 106 #endif 107