Build a basic Mojo service framework for device/usb

device/usb/device_manager_impl_unittest.cc

+// Copyright 2015 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <set>
+#include <string>
+#include <vector>
+
+#include "base/barrier_closure.h"
+#include "base/bind.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/message_loop/message_loop.h"
+#include "base/run_loop.h"
+#include "device/core/device_client.h"
+#include "device/usb/device_impl.h"
+#include "device/usb/device_manager_impl.h"
+#include "device/usb/mock_usb_device.h"
+#include "device/usb/mock_usb_service.h"
+#include "device/usb/public/cpp/device_manager_delegate.h"
+#include "device/usb/public/cpp/device_manager_factory.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/mojo/src/mojo/public/cpp/bindings/interface_request.h"
+
+namespace device {
+namespace usb {
+
+namespace {
+
+bool DefaultDelegateFilter(const DeviceInfo& device_info) {
+  return true;
+}
+
+class TestDeviceManagerDelegate : public DeviceManagerDelegate {
+ public:
+  using Filter = base::Callback<bool(const DeviceInfo&)>;
+
+  TestDeviceManagerDelegate(const Filter& filter) : filter_(filter) {}
+  ~TestDeviceManagerDelegate() override {}
+
+  void set_filter(const Filter& filter) { filter_ = filter; }
+
+ private:
+  // DeviceManagerDelegate implementation:
+  bool IsDeviceAllowed(const DeviceInfo& device_info) override {
+    return filter_.Run(device_info);
+  }
+
+  Filter filter_;
+};
+
+class TestDeviceClient : public DeviceClient {
+ public:
+  TestDeviceClient() : delegate_filter_(base::Bind(&DefaultDelegateFilter)) {}
+  ~TestDeviceClient() override {}
+
+  MockUsbService& mock_usb_service() { return mock_usb_service_; }
+
+  void SetDelegateFilter(const TestDeviceManagerDelegate::Filter& filter) {
+    delegate_filter_ = filter;
+  }
+
+ private:
+  // DeviceClient implementation:
+  UsbService* GetUsbService() override { return &mock_usb_service_; }
+
+  void ConnectToUSBDeviceManager(
+      mojo::InterfaceRequest<DeviceManager> request) override {
+    new DeviceManagerImpl(request.Pass(),
+                          scoped_ptr<DeviceManagerDelegate>(
+                              new TestDeviceManagerDelegate(delegate_filter_)));
+  }
+
+  TestDeviceManagerDelegate::Filter delegate_filter_;
+  MockUsbService mock_usb_service_;
+};
+
+class DeviceManagerImplTest : public testing::Test {
+ public:
+  DeviceManagerImplTest()
+      : message_loop_(new base::MessageLoop),
+        device_client_(new TestDeviceClient) {}
+  ~DeviceManagerImplTest() override {}
+
+ protected:
+  MockUsbService& mock_usb_service() {
+    return device_client_->mock_usb_service();
+  }
+
+  void SetDelegateFilter(const TestDeviceManagerDelegate::Filter& filter) {
+    device_client_->SetDelegateFilter(filter);
+  }
+
+ private:
+  scoped_ptr<base::MessageLoop> message_loop_;
+  scoped_ptr<TestDeviceClient> device_client_;
+};
+
+void VerifyDevicesAndThen(const std::set<std::string>& expected_serials,
+                          scoped_ptr<std::set<std::string>> actual_serials,
+                          const base::Closure& continuation) {
+  EXPECT_EQ(expected_serials, *actual_serials);
+  continuation.Run();
+}
+
+void OnGetDeviceInfo(std::set<std::string>* actual_serials,
+                     const base::Closure& barrier,
+                     DevicePtr device,
+                     DeviceInfoPtr info) {
+  actual_serials->insert(info->serial_number.To<std::string>());
+  barrier.Run();
+}
+
+void ExpectDevicesAndThen(const std::set<std::string>& serials,
+                          const base::Closure& continuation,
+                          mojo::Array<EnumerationResultPtr> results) {
+  EXPECT_EQ(serials.size(), results.size());
+  scoped_ptr<std::set<std::string>> actual_serials(new std::set<std::string>);
+  std::set<std::string>* actual_serials_raw = actual_serials.get();
+  base::Closure barrier = base::BarrierClosure(
+      static_cast<int>(results.size()),
+      base::Bind(&VerifyDevicesAndThen, serials, base::Passed(&actual_serials),
+                 continuation));
+  for (size_t i = 0; i < results.size(); ++i) {
+    DevicePtr device = results[i]->device.Pass();
+    Device* raw_device = device.get();
+    raw_device->GetDeviceInfo(base::Bind(&OnGetDeviceInfo, actual_serials_raw,
+                                         barrier, base::Passed(&device)));
+  }
+}
+
+}  // namespace
+
+// Test basic GetDevices functionality to ensure that all mock devices are
+// returned by the service.
+TEST_F(DeviceManagerImplTest, GetDevices) {
+  scoped_refptr<MockUsbDevice> device0 =
+      new MockUsbDevice(0x1234, 0x5678, "ACME", "Frobinator", "ABCDEF");
+  scoped_refptr<MockUsbDevice> device1 =
+      new MockUsbDevice(0x1234, 0x5679, "ACME", "Frobinator+", "GHIJKL");
+  scoped_refptr<MockUsbDevice> device2 =
+      new MockUsbDevice(0x1234, 0x567a, "ACME", "Frobinator Mk II", "MNOPQR");
+
+  mock_usb_service().AddDevice(device0);
+  mock_usb_service().AddDevice(device1);
+  mock_usb_service().AddDevice(device2);
+
+  DeviceManagerPtr device_manager;
+  DeviceClient::Get()->ConnectToUSBDeviceManager(
+      mojo::GetProxy(&device_manager));
+
+  EnumerationOptionsPtr options = EnumerationOptions::New();
+  options->filters = mojo::Array<DeviceFilterPtr>::New(1);
+  options->filters[0] = DeviceFilter::New();
+  options->filters[0]->has_vendor_id = true;
+  options->filters[0]->vendor_id = 0x1234;
+
+  std::set<std::string> serials;
+  serials.insert("ABCDEF");
+  serials.insert("GHIJKL");
+  serials.insert("MNOPQR");
+
+  EXPECT_CALL(*device0.get(), GetConfiguration());
+  EXPECT_CALL(*device1.get(), GetConfiguration());
+  EXPECT_CALL(*device2.get(), GetConfiguration());
+
+  base::RunLoop run_loop;
+  device_manager->GetDevices(
+      options.Pass(),
+      base::Bind(&ExpectDevicesAndThen, serials, run_loop.QuitClosure()));
+  run_loop.Run();
+}
+
+}  // namespace usb
+}  // namespace device