OLD | NEW |
| (Empty) |
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 #include <set> | |
6 #include <string> | |
7 #include <vector> | |
8 | |
9 #include "base/barrier_closure.h" | |
10 #include "base/bind.h" | |
11 #include "base/memory/scoped_ptr.h" | |
12 #include "base/message_loop/message_loop.h" | |
13 #include "base/run_loop.h" | |
14 #include "device/core/device_client.h" | |
15 #include "device/usb/device_impl.h" | |
16 #include "device/usb/device_manager_impl.h" | |
17 #include "device/usb/mock_usb_device.h" | |
18 #include "device/usb/mock_usb_service.h" | |
19 #include "device/usb/public/cpp/device_manager_delegate.h" | |
20 #include "device/usb/public/cpp/device_manager_factory.h" | |
21 #include "testing/gtest/include/gtest/gtest.h" | |
22 #include "third_party/mojo/src/mojo/public/cpp/bindings/interface_request.h" | |
23 | |
24 namespace device { | |
25 namespace usb { | |
26 | |
27 namespace { | |
28 | |
29 bool DefaultDelegateFilter(const DeviceInfo& device_info) { | |
30 return true; | |
31 } | |
32 | |
33 class TestDeviceManagerDelegate : public DeviceManagerDelegate { | |
34 public: | |
35 using Filter = base::Callback<bool(const DeviceInfo&)>; | |
36 | |
37 TestDeviceManagerDelegate(const Filter& filter) : filter_(filter) {} | |
38 ~TestDeviceManagerDelegate() override {} | |
39 | |
40 void set_filter(const Filter& filter) { filter_ = filter; } | |
41 | |
42 private: | |
43 // DeviceManagerDelegate implementation: | |
44 bool IsDeviceAllowed(const DeviceInfo& device_info) override { | |
45 return filter_.Run(device_info); | |
46 } | |
47 | |
48 Filter filter_; | |
49 }; | |
50 | |
51 class TestDeviceClient : public DeviceClient { | |
52 public: | |
53 TestDeviceClient() : delegate_filter_(base::Bind(&DefaultDelegateFilter)) {} | |
54 ~TestDeviceClient() override {} | |
55 | |
56 MockUsbService& mock_usb_service() { return mock_usb_service_; } | |
57 | |
58 void SetDelegateFilter(const TestDeviceManagerDelegate::Filter& filter) { | |
59 delegate_filter_ = filter; | |
60 } | |
61 | |
62 private: | |
63 // DeviceClient implementation: | |
64 UsbService* GetUsbService() override { return &mock_usb_service_; } | |
65 | |
66 void ConnectToUSBDeviceManager( | |
67 mojo::InterfaceRequest<DeviceManager> request) override { | |
68 new DeviceManagerImpl(request.Pass(), | |
69 scoped_ptr<DeviceManagerDelegate>( | |
70 new TestDeviceManagerDelegate(delegate_filter_))); | |
71 } | |
72 | |
73 TestDeviceManagerDelegate::Filter delegate_filter_; | |
74 MockUsbService mock_usb_service_; | |
75 }; | |
76 | |
77 class DeviceManagerImplTest : public testing::Test { | |
78 public: | |
79 DeviceManagerImplTest() | |
80 : message_loop_(new base::MessageLoop), | |
81 device_client_(new TestDeviceClient) {} | |
82 ~DeviceManagerImplTest() override {} | |
83 | |
84 protected: | |
85 MockUsbService& mock_usb_service() { | |
86 return device_client_->mock_usb_service(); | |
87 } | |
88 | |
89 void SetDelegateFilter(const TestDeviceManagerDelegate::Filter& filter) { | |
90 device_client_->SetDelegateFilter(filter); | |
91 } | |
92 | |
93 private: | |
94 scoped_ptr<base::MessageLoop> message_loop_; | |
95 scoped_ptr<TestDeviceClient> device_client_; | |
96 }; | |
97 | |
98 void VerifyDevicesAndThen(const std::set<std::string>& expected_serials, | |
99 scoped_ptr<std::set<std::string>> actual_serials, | |
100 const base::Closure& continuation) { | |
101 EXPECT_EQ(expected_serials, *actual_serials); | |
102 continuation.Run(); | |
103 } | |
104 | |
105 void OnGetDeviceInfo(std::set<std::string>* actual_serials, | |
106 const base::Closure& barrier, | |
107 DevicePtr device, | |
108 DeviceInfoPtr info) { | |
109 actual_serials->insert(info->serial_number.To<std::string>()); | |
110 barrier.Run(); | |
111 } | |
112 | |
113 void ExpectDevicesAndThen(const std::set<std::string>& serials, | |
114 const base::Closure& continuation, | |
115 mojo::Array<EnumerationResultPtr> results) { | |
116 EXPECT_EQ(serials.size(), results.size()); | |
117 scoped_ptr<std::set<std::string>> actual_serials(new std::set<std::string>); | |
118 std::set<std::string>* actual_serials_raw = actual_serials.get(); | |
119 base::Closure barrier = base::BarrierClosure( | |
120 static_cast<int>(results.size()), | |
121 base::Bind(&VerifyDevicesAndThen, serials, base::Passed(&actual_serials), | |
122 continuation)); | |
123 for (size_t i = 0; i < results.size(); ++i) { | |
124 DevicePtr device = results[i]->device.Pass(); | |
125 Device* raw_device = device.get(); | |
126 raw_device->GetDeviceInfo(base::Bind(&OnGetDeviceInfo, actual_serials_raw, | |
127 barrier, base::Passed(&device))); | |
128 } | |
129 } | |
130 | |
131 } // namespace | |
132 | |
133 // Test basic GetDevices functionality to ensure that all mock devices are | |
134 // returned by the service. | |
135 TEST_F(DeviceManagerImplTest, GetDevices) { | |
136 scoped_refptr<MockUsbDevice> device0 = | |
137 new MockUsbDevice(0x1234, 0x5678, "ACME", "Frobinator", "ABCDEF"); | |
138 scoped_refptr<MockUsbDevice> device1 = | |
139 new MockUsbDevice(0x1234, 0x5679, "ACME", "Frobinator+", "GHIJKL"); | |
140 scoped_refptr<MockUsbDevice> device2 = | |
141 new MockUsbDevice(0x1234, 0x567a, "ACME", "Frobinator Mk II", "MNOPQR"); | |
142 | |
143 mock_usb_service().AddDevice(device0); | |
144 mock_usb_service().AddDevice(device1); | |
145 mock_usb_service().AddDevice(device2); | |
146 | |
147 DeviceManagerPtr device_manager; | |
148 DeviceClient::Get()->ConnectToUSBDeviceManager( | |
149 mojo::GetProxy(&device_manager)); | |
150 | |
151 EnumerationOptionsPtr options = EnumerationOptions::New(); | |
152 options->filters = mojo::Array<DeviceFilterPtr>::New(1); | |
153 options->filters[0] = DeviceFilter::New(); | |
154 options->filters[0]->has_vendor_id = true; | |
155 options->filters[0]->vendor_id = 0x1234; | |
156 | |
157 std::set<std::string> serials; | |
158 serials.insert("ABCDEF"); | |
159 serials.insert("GHIJKL"); | |
160 serials.insert("MNOPQR"); | |
161 | |
162 EXPECT_CALL(*device0.get(), GetConfiguration()); | |
163 EXPECT_CALL(*device1.get(), GetConfiguration()); | |
164 EXPECT_CALL(*device2.get(), GetConfiguration()); | |
165 | |
166 base::RunLoop run_loop; | |
167 device_manager->GetDevices( | |
168 options.Pass(), | |
169 base::Bind(&ExpectDevicesAndThen, serials, run_loop.QuitClosure())); | |
170 run_loop.Run(); | |
171 } | |
172 | |
173 } // namespace usb | |
174 } // namespace device | |
OLD | NEW |