| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/extensions/api/usb/usb_api.h" | 5 #include "chrome/browser/extensions/api/usb/usb_api.h" |
| 6 #include "chrome/browser/extensions/extension_apitest.h" | 6 #include "chrome/browser/extensions/extension_apitest.h" |
| 7 #include "chrome/browser/ui/browser.h" | 7 #include "chrome/browser/ui/browser.h" |
| 8 #include "chrome/browser/usb/usb_service.h" | 8 #include "chrome/browser/usb/usb_service.h" |
| 9 #include "chrome/browser/usb/usb_service_factory.h" | 9 #include "chrome/browser/usb/usb_service_factory.h" |
| 10 #include "chrome/common/chrome_switches.h" | 10 #include "chrome/common/chrome_switches.h" |
| (...skipping 27 matching lines...) Expand all Loading... |
| 38 #if defined(OS_WIN) | 38 #if defined(OS_WIN) |
| 39 #pragma warning(push) | 39 #pragma warning(push) |
| 40 #pragma warning(disable:4373) | 40 #pragma warning(disable:4373) |
| 41 #endif | 41 #endif |
| 42 class MockUsbDevice : public UsbDevice { | 42 class MockUsbDevice : public UsbDevice { |
| 43 public: | 43 public: |
| 44 MockUsbDevice() : UsbDevice() {} | 44 MockUsbDevice() : UsbDevice() {} |
| 45 | 45 |
| 46 MOCK_METHOD1(Close, void(const base::Callback<void()>& callback)); | 46 MOCK_METHOD1(Close, void(const base::Callback<void()>& callback)); |
| 47 | 47 |
| 48 MOCK_METHOD10(ControlTransfer, void(const TransferDirection direction, | 48 MOCK_METHOD10(ControlTransfer, void(const UsbEndpointDirection direction, |
| 49 const TransferRequestType request_type, const TransferRecipient recipient, | 49 const TransferRequestType request_type, const TransferRecipient recipient, |
| 50 const uint8 request, const uint16 value, const uint16 index, | 50 const uint8 request, const uint16 value, const uint16 index, |
| 51 net::IOBuffer* buffer, const size_t length, const unsigned int timeout, | 51 net::IOBuffer* buffer, const size_t length, const unsigned int timeout, |
| 52 const UsbTransferCallback& callback)); | 52 const UsbTransferCallback& callback)); |
| 53 | 53 |
| 54 MOCK_METHOD6(BulkTransfer, void(const TransferDirection direction, | 54 MOCK_METHOD6(BulkTransfer, void(const UsbEndpointDirection direction, |
| 55 const uint8 endpoint, net::IOBuffer* buffer, const size_t length, | 55 const uint8 endpoint, net::IOBuffer* buffer, const size_t length, |
| 56 const unsigned int timeout, const UsbTransferCallback& callback)); | 56 const unsigned int timeout, const UsbTransferCallback& callback)); |
| 57 | 57 |
| 58 MOCK_METHOD6(InterruptTransfer, void(const TransferDirection direction, | 58 MOCK_METHOD6(InterruptTransfer, void(const UsbEndpointDirection direction, |
| 59 const uint8 endpoint, net::IOBuffer* buffer, const size_t length, | 59 const uint8 endpoint, net::IOBuffer* buffer, const size_t length, |
| 60 const unsigned int timeout, const UsbTransferCallback& callback)); | 60 const unsigned int timeout, const UsbTransferCallback& callback)); |
| 61 | 61 |
| 62 MOCK_METHOD8(IsochronousTransfer, void(const TransferDirection direction, | 62 MOCK_METHOD8(IsochronousTransfer, void(const UsbEndpointDirection direction, |
| 63 const uint8 endpoint, net::IOBuffer* buffer, const size_t length, | 63 const uint8 endpoint, net::IOBuffer* buffer, const size_t length, |
| 64 const unsigned int packets, const unsigned int packet_length, | 64 const unsigned int packets, const unsigned int packet_length, |
| 65 const unsigned int timeout, const UsbTransferCallback& callback)); | 65 const unsigned int timeout, const UsbTransferCallback& callback)); |
| 66 | 66 |
| 67 MOCK_METHOD1(ResetDevice, void(const base::Callback<void(bool)>& callback)); | 67 MOCK_METHOD1(ResetDevice, void(const base::Callback<void(bool)>& callback)); |
| 68 | 68 |
| 69 MOCK_METHOD2(ListInterfaces, void(UsbConfigDescriptor* config, |
| 70 const UsbInterfaceCallback& callback)); |
| 71 |
| 69 protected: | 72 protected: |
| 70 virtual ~MockUsbDevice() {} | 73 virtual ~MockUsbDevice() {} |
| 71 }; | 74 }; |
| 72 #if defined(OS_WIN) | 75 #if defined(OS_WIN) |
| 73 #pragma warning(pop) | 76 #pragma warning(pop) |
| 74 #endif | 77 #endif |
| 75 | 78 |
| 76 class UsbApiTest : public ExtensionApiTest { | 79 class UsbApiTest : public ExtensionApiTest { |
| 77 public: | 80 public: |
| 78 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { | 81 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
| (...skipping 12 matching lines...) Expand all Loading... |
| 91 | 94 |
| 92 } // namespace | 95 } // namespace |
| 93 | 96 |
| 94 IN_PROC_BROWSER_TEST_F(UsbApiTest, DeviceHandling) { | 97 IN_PROC_BROWSER_TEST_F(UsbApiTest, DeviceHandling) { |
| 95 EXPECT_CALL(*mock_device_, Close(_)) | 98 EXPECT_CALL(*mock_device_, Close(_)) |
| 96 .WillRepeatedly(InvokeUsbCallback()); | 99 .WillRepeatedly(InvokeUsbCallback()); |
| 97 EXPECT_CALL(*mock_device_, ResetDevice(_)) | 100 EXPECT_CALL(*mock_device_, ResetDevice(_)) |
| 98 .WillOnce(InvokeUsbResultCallback<0>(true)) | 101 .WillOnce(InvokeUsbResultCallback<0>(true)) |
| 99 .WillOnce(InvokeUsbResultCallback<0>(false)); | 102 .WillOnce(InvokeUsbResultCallback<0>(false)); |
| 100 EXPECT_CALL(*mock_device_, | 103 EXPECT_CALL(*mock_device_, |
| 101 InterruptTransfer(UsbDevice::OUTBOUND, 2, _, 1, _, _)) | 104 InterruptTransfer(USB_DIRECTION_OUTBOUND, 2, _, 1, _, _)) |
| 102 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); | 105 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); |
| 103 ASSERT_TRUE(RunExtensionTest("usb/device_handling")); | 106 ASSERT_TRUE(RunExtensionTest("usb/device_handling")); |
| 104 } | 107 } |
| 105 | 108 |
| 109 IN_PROC_BROWSER_TEST_F(UsbApiTest, ListInterfaces) { |
| 110 EXPECT_CALL(*mock_device_, ListInterfaces(_, _)) |
| 111 .WillOnce(InvokeUsbResultCallback<1>(false)); |
| 112 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); |
| 113 ASSERT_TRUE(RunExtensionTest("usb/list_interfaces")); |
| 114 } |
| 115 |
| 106 IN_PROC_BROWSER_TEST_F(UsbApiTest, TransferEvent) { | 116 IN_PROC_BROWSER_TEST_F(UsbApiTest, TransferEvent) { |
| 107 EXPECT_CALL(*mock_device_, | 117 EXPECT_CALL(*mock_device_, |
| 108 ControlTransfer(UsbDevice::OUTBOUND, UsbDevice::STANDARD, | 118 ControlTransfer(USB_DIRECTION_OUTBOUND, UsbDevice::STANDARD, |
| 109 UsbDevice::DEVICE, 1, 2, 3, _, 1, _, _)) | 119 UsbDevice::DEVICE, 1, 2, 3, _, 1, _, _)) |
| 110 .WillOnce(InvokeUsbTransferCallback<9>(USB_TRANSFER_COMPLETED)); | 120 .WillOnce(InvokeUsbTransferCallback<9>(USB_TRANSFER_COMPLETED)); |
| 111 EXPECT_CALL(*mock_device_, | 121 EXPECT_CALL(*mock_device_, |
| 112 BulkTransfer(UsbDevice::OUTBOUND, 1, _, 1, _, _)) | 122 BulkTransfer(USB_DIRECTION_OUTBOUND, 1, _, 1, _, _)) |
| 113 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); | 123 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); |
| 114 EXPECT_CALL(*mock_device_, | 124 EXPECT_CALL(*mock_device_, |
| 115 InterruptTransfer(UsbDevice::OUTBOUND, 2, _, 1, _, _)) | 125 InterruptTransfer(USB_DIRECTION_OUTBOUND, 2, _, 1, _, _)) |
| 116 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); | 126 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); |
| 117 EXPECT_CALL(*mock_device_, | 127 EXPECT_CALL(*mock_device_, |
| 118 IsochronousTransfer(UsbDevice::OUTBOUND, 3, _, 1, 1, 1, _, _)) | 128 IsochronousTransfer(USB_DIRECTION_OUTBOUND, 3, _, 1, 1, 1, _, _)) |
| 119 .WillOnce(InvokeUsbTransferCallback<7>(USB_TRANSFER_COMPLETED)); | 129 .WillOnce(InvokeUsbTransferCallback<7>(USB_TRANSFER_COMPLETED)); |
| 120 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); | 130 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); |
| 121 ASSERT_TRUE(RunExtensionTest("usb/transfer_event")); | 131 ASSERT_TRUE(RunExtensionTest("usb/transfer_event")); |
| 122 } | 132 } |
| 123 | 133 |
| 124 IN_PROC_BROWSER_TEST_F(UsbApiTest, ZeroLengthTransfer) { | 134 IN_PROC_BROWSER_TEST_F(UsbApiTest, ZeroLengthTransfer) { |
| 125 EXPECT_CALL(*mock_device_, BulkTransfer(_, _, _, 0, _, _)) | 135 EXPECT_CALL(*mock_device_, BulkTransfer(_, _, _, 0, _, _)) |
| 126 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); | 136 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)); |
| 127 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); | 137 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); |
| 128 ASSERT_TRUE(RunExtensionTest("usb/zero_length_transfer")); | 138 ASSERT_TRUE(RunExtensionTest("usb/zero_length_transfer")); |
| 129 } | 139 } |
| 130 | 140 |
| 131 IN_PROC_BROWSER_TEST_F(UsbApiTest, TransferFailure) { | 141 IN_PROC_BROWSER_TEST_F(UsbApiTest, TransferFailure) { |
| 132 EXPECT_CALL(*mock_device_, BulkTransfer(_, _, _, _, _, _)) | 142 EXPECT_CALL(*mock_device_, BulkTransfer(_, _, _, _, _, _)) |
| 133 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)) | 143 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_COMPLETED)) |
| 134 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_ERROR)) | 144 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_ERROR)) |
| 135 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_TIMEOUT)); | 145 .WillOnce(InvokeUsbTransferCallback<5>(USB_TRANSFER_TIMEOUT)); |
| 136 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); | 146 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); |
| 137 ASSERT_TRUE(RunExtensionTest("usb/transfer_failure")); | 147 ASSERT_TRUE(RunExtensionTest("usb/transfer_failure")); |
| 138 } | 148 } |
| 139 | 149 |
| 140 IN_PROC_BROWSER_TEST_F(UsbApiTest, InvalidLengthTransfer) { | 150 IN_PROC_BROWSER_TEST_F(UsbApiTest, InvalidLengthTransfer) { |
| 141 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); | 151 EXPECT_CALL(*mock_device_, Close(_)).Times(AnyNumber()); |
| 142 ASSERT_TRUE(RunExtensionTest("usb/invalid_length_transfer")); | 152 ASSERT_TRUE(RunExtensionTest("usb/invalid_length_transfer")); |
| 143 } | 153 } |
| 144 | 154 |
| OLD | NEW |