device/usb/mock_usb_service.h

   1 // Copyright 2015 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef DEVICE_USB_MOCK_USB_SERVICE_H_
   6 #define DEVICE_USB_MOCK_USB_SERVICE_H_
   7
   8 #include "device/usb/usb_service.h"
   9
  10 #include "testing/gmock/include/gmock/gmock.h"
  11
  12 namespace device {
  13
  14 class MockUsbService : public UsbService {
  15  public:
  16   MockUsbService();
  17   ~MockUsbService() override;
  18
  19   void AddDevice(scoped_refptr<UsbDevice> device);
  20   void RemoveDevice(scoped_refptr<UsbDevice> device);
  21
  22  protected:
  23   // UsbService implementation:
  24   scoped_refptr<UsbDevice> GetDeviceById(uint32 unique_id) override;
  25   void GetDevices(const GetDevicesCallback& callback) override;
  26
  27   std::map<uint32, scoped_refptr<UsbDevice>> devices_;
  28 };
  29
  30 }  // namespace device
  31
  32 #endif  // DEVICE_USB_MOCK_USB_SERVICE_H_