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 |