OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/extension_apitest.h" | 5 #include "chrome/browser/extensions/extension_apitest.h" |
6 #include "chrome/browser/ui/browser.h" | 6 #include "chrome/browser/ui/browser.h" |
7 #include "components/usb_service/usb_service.h" | 7 #include "components/usb_service/usb_service.h" |
8 #include "content/public/browser/browser_thread.h" | 8 #include "content/public/browser/browser_thread.h" |
9 #include "content/public/test/test_utils.h" | 9 #include "content/public/test/test_utils.h" |
10 #include "extensions/browser/api/usb/usb_api.h" | 10 #include "extensions/browser/api/usb/usb_api.h" |
11 #include "net/base/io_buffer.h" | 11 #include "net/base/io_buffer.h" |
12 #include "testing/gmock/include/gmock/gmock.h" | 12 #include "testing/gmock/include/gmock/gmock.h" |
13 | 13 |
14 using testing::AnyNumber; | 14 using testing::AnyNumber; |
15 using testing::_; | 15 using testing::_; |
16 using testing::Return; | 16 using testing::Return; |
17 using content::BrowserThread; | 17 using content::BrowserThread; |
18 using usb_service::UsbConfigDescriptor; | 18 using usb_service::UsbConfigDescriptor; |
19 using usb_service::UsbDevice; | 19 using usb_service::UsbDevice; |
20 using usb_service::UsbDeviceHandle; | 20 using usb_service::UsbDeviceHandle; |
21 using usb_service::UsbEndpointDirection; | 21 using usb_service::UsbEndpointDirection; |
| 22 using usb_service::UsbInterfaceDescriptor; |
22 using usb_service::UsbService; | 23 using usb_service::UsbService; |
23 using usb_service::UsbTransferCallback; | 24 using usb_service::UsbTransferCallback; |
24 | 25 |
25 namespace { | 26 namespace { |
26 | 27 |
27 ACTION_TEMPLATE(InvokeUsbTransferCallback, | 28 ACTION_TEMPLATE(InvokeUsbTransferCallback, |
28 HAS_1_TEMPLATE_PARAMS(int, k), | 29 HAS_1_TEMPLATE_PARAMS(int, k), |
29 AND_1_VALUE_PARAMS(p1)) { | 30 AND_1_VALUE_PARAMS(p1)) { |
30 net::IOBuffer* io_buffer = new net::IOBuffer(1); | 31 net::IOBuffer* io_buffer = new net::IOBuffer(1); |
31 memset(io_buffer->data(), 0, 1); // Avoid uninitialized reads. | 32 memset(io_buffer->data(), 0, 1); // Avoid uninitialized reads. |
(...skipping 56 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
88 | 89 |
89 void set_device(UsbDevice* device) { device_ = device; } | 90 void set_device(UsbDevice* device) { device_ = device; } |
90 | 91 |
91 protected: | 92 protected: |
92 virtual ~MockUsbDeviceHandle() {} | 93 virtual ~MockUsbDeviceHandle() {} |
93 }; | 94 }; |
94 | 95 |
95 class MockUsbConfigDescriptor : public UsbConfigDescriptor { | 96 class MockUsbConfigDescriptor : public UsbConfigDescriptor { |
96 public: | 97 public: |
97 MOCK_CONST_METHOD0(GetNumInterfaces, size_t()); | 98 MOCK_CONST_METHOD0(GetNumInterfaces, size_t()); |
| 99 MOCK_CONST_METHOD1(GetInterface, |
| 100 scoped_refptr<const UsbInterfaceDescriptor>(size_t index)); |
98 | 101 |
99 protected: | 102 protected: |
100 virtual ~MockUsbConfigDescriptor() {} | 103 virtual ~MockUsbConfigDescriptor() {} |
101 }; | 104 }; |
102 | 105 |
103 class MockUsbDevice : public UsbDevice { | 106 class MockUsbDevice : public UsbDevice { |
104 public: | 107 public: |
105 explicit MockUsbDevice(MockUsbDeviceHandle* mock_handle) | 108 explicit MockUsbDevice(MockUsbDeviceHandle* mock_handle) |
106 : UsbDevice(0, 0, 0), mock_handle_(mock_handle) { | 109 : UsbDevice(0, 0, 0), mock_handle_(mock_handle) { |
107 mock_handle->set_device(this); | 110 mock_handle->set_device(this); |
(...skipping 160 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
268 .WillOnce( | 271 .WillOnce( |
269 InvokeUsbTransferCallback<5>(usb_service::USB_TRANSFER_TIMEOUT)); | 272 InvokeUsbTransferCallback<5>(usb_service::USB_TRANSFER_TIMEOUT)); |
270 EXPECT_CALL(*mock_device_handle_.get(), Close()).Times(AnyNumber()); | 273 EXPECT_CALL(*mock_device_handle_.get(), Close()).Times(AnyNumber()); |
271 ASSERT_TRUE(RunExtensionTest("usb/transfer_failure")); | 274 ASSERT_TRUE(RunExtensionTest("usb/transfer_failure")); |
272 } | 275 } |
273 | 276 |
274 IN_PROC_BROWSER_TEST_F(UsbApiTest, InvalidLengthTransfer) { | 277 IN_PROC_BROWSER_TEST_F(UsbApiTest, InvalidLengthTransfer) { |
275 EXPECT_CALL(*mock_device_handle_.get(), Close()).Times(AnyNumber()); | 278 EXPECT_CALL(*mock_device_handle_.get(), Close()).Times(AnyNumber()); |
276 ASSERT_TRUE(RunExtensionTest("usb/invalid_length_transfer")); | 279 ASSERT_TRUE(RunExtensionTest("usb/invalid_length_transfer")); |
277 } | 280 } |
OLD | NEW |