Add a service to track devices selected by the user.

apps/saved_devices_service_unittest.cc

Index: apps/saved_devices_service_unittest.cc
diff --git a/apps/saved_devices_service_unittest.cc b/apps/saved_devices_service_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b2ed54451b006c6c7b0157cfea15ed172af3a983
--- /dev/null
+++ b/apps/saved_devices_service_unittest.cc
@@ -0,0 +1,213 @@
+// Copyright 2014 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 "apps/saved_devices_service.h"
+#include "base/run_loop.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/test/values_test_util.h"
+#include "chrome/browser/extensions/test_extension_environment.h"
+#include "chrome/test/base/testing_profile.h"
+#include "device/usb/usb_device.h"
+#include "device/usb/usb_device_handle.h"
+#include "extensions/browser/extension_prefs.h"
+#include "extensions/common/extension.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace apps {
+
+namespace {
+
+using device::UsbDevice;
+using device::UsbDeviceHandle;
+using device::UsbEndpointDirection;
+using device::UsbTransferCallback;
+using testing::Return;
+
+class MockUsbDeviceHandle : public UsbDeviceHandle {
+ public:
+  MockUsbDeviceHandle(const std::string& serial_number)
+      : UsbDeviceHandle(), serial_number_(serial_number) {}
+
+  MOCK_CONST_METHOD0(GetDevice, scoped_refptr<UsbDevice>());
+  MOCK_METHOD0(Close, void());
+
+  MOCK_METHOD10(ControlTransfer,
+                void(UsbEndpointDirection direction,
+                     TransferRequestType request_type,
+                     TransferRecipient recipient,
+                     uint8 request,
+                     uint16 value,
+                     uint16 index,
+                     net::IOBuffer* buffer,
+                     size_t length,
+                     unsigned int timeout,
+                     const UsbTransferCallback& callback));
+
+  MOCK_METHOD6(BulkTransfer,
+               void(UsbEndpointDirection direction,
+                    uint8 endpoint,
+                    net::IOBuffer* buffer,
+                    size_t length,
+                    unsigned int timeout,
+                    const UsbTransferCallback& callback));
+
+  MOCK_METHOD6(InterruptTransfer,
+               void(UsbEndpointDirection direction,
+                    uint8 endpoint,
+                    net::IOBuffer* buffer,
+                    size_t length,
+                    unsigned int timeout,
+                    const UsbTransferCallback& callback));
+
+  MOCK_METHOD8(IsochronousTransfer,
+               void(UsbEndpointDirection direction,
+                    uint8 endpoint,
+                    net::IOBuffer* buffer,
+                    size_t length,
+                    unsigned int packets,
+                    unsigned int packet_length,
+                    unsigned int timeout,
+                    const UsbTransferCallback& callback));
+
+  MOCK_METHOD0(ResetDevice, bool());
+  MOCK_METHOD1(ClaimInterface, bool(int interface_number));
+  MOCK_METHOD1(ReleaseInterface, bool(int interface_number));
+  MOCK_METHOD2(SetInterfaceAlternateSetting,
+               bool(int interface_number, int alternate_setting));
+  MOCK_METHOD1(GetManufacturer, bool(base::string16* manufacturer));
+  MOCK_METHOD1(GetProduct, bool(base::string16* product));
+
+  bool GetSerial(base::string16* serial) OVERRIDE {
+    if (serial_number_.empty()) {
+      return false;
+    }
+
+    *serial = base::UTF8ToUTF16(serial_number_);
+    return true;
+  }
+
+ private:
+  virtual ~MockUsbDeviceHandle() {}
+
+  const std::string serial_number_;
+};
+
+class MockUsbDevice : public UsbDevice {
+ public:
+  MockUsbDevice(const std::string& serial_number, uint32 unique_id)
+      : UsbDevice(0, 0, unique_id), serial_number_(serial_number) {}
+
+  MOCK_METHOD1(Close, bool(scoped_refptr<UsbDeviceHandle>));
+#if defined(OS_CHROMEOS)
+  MOCK_METHOD2(RequestUsbAccess, void(int, const base::Callback<void(bool)>&));
+#endif
+  MOCK_METHOD0(GetConfiguration, const device::UsbConfigDescriptor&());
+
+  scoped_refptr<UsbDeviceHandle> Open() OVERRIDE {
+    return new MockUsbDeviceHandle(serial_number_);
+  }
+
+  void NotifyDisconnect() { UsbDevice::NotifyDisconnect(); }
+
+ private:
+  virtual ~MockUsbDevice() {}
+
+  const std::string serial_number_;
+};
+}
+
+class SavedDevicesServiceTest : public testing::Test {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    testing::Test::SetUp();
+    env_.GetExtensionPrefs();  // Force creation before adding extensions.
+    extension_ = env_.MakeExtension(*base::test::ParseJson(
+                                        "{"
+                                        "  \"app\": {"
+                                        "    \"background\": {"
+                                        "      \"scripts\": [\"background.js\"]"
+                                        "    }"
+                                        "  },"
+                                        "  \"permissions\": ["
+                                        "    \"usb\""
+                                        "  ]"
+                                        "}"));
+    service_ = SavedDevicesService::Get(env_.profile());
+    device0 = new MockUsbDevice("ABCDE", 0);
+    device1 = new MockUsbDevice("", 1);
+    device2 = new MockUsbDevice("12345", 2);
+    device3 = new MockUsbDevice("", 3);
+  }
+
+  extensions::TestExtensionEnvironment env_;
+  const extensions::Extension* extension_;
+  SavedDevicesService* service_;
+  scoped_refptr<MockUsbDevice> device0;
+  scoped_refptr<MockUsbDevice> device1;
+  scoped_refptr<MockUsbDevice> device2;
+  scoped_refptr<MockUsbDevice> device3;
+};
+
+TEST_F(SavedDevicesServiceTest, RegisterDevices) {
+  SavedDevicesService::SavedDevices* saved_devices =
+      service_->GetOrInsert(extension_->id());
+
+  base::string16 serial_number(base::ASCIIToUTF16("ABCDE"));
+  saved_devices->RegisterDevice(device0, &serial_number);
+  saved_devices->RegisterDevice(device1, NULL);
+
+  // This is necessary as writing out registered devices happens in a task on
+  // the UI thread.
+  base::RunLoop run_loop;
+  run_loop.RunUntilIdle();
+
+  ASSERT_TRUE(saved_devices->IsRegistered(device0));
+  ASSERT_TRUE(saved_devices->IsRegistered(device1));
+  ASSERT_FALSE(saved_devices->IsRegistered(device2));
+  ASSERT_FALSE(saved_devices->IsRegistered(device3));
+
+  std::vector<SavedDeviceEntry> device_entries =
+      service_->GetAllDevices(extension_->id());
+  ASSERT_EQ(1U, device_entries.size());
+  ASSERT_EQ(base::ASCIIToUTF16("ABCDE"), device_entries[0].serial_number);
+
+  device1->NotifyDisconnect();
+
+  ASSERT_TRUE(saved_devices->IsRegistered(device0));
+  ASSERT_FALSE(saved_devices->IsRegistered(device1));
+  ASSERT_FALSE(saved_devices->IsRegistered(device2));
+  ASSERT_FALSE(saved_devices->IsRegistered(device3));
+
+  service_->Clear(extension_->id());
+
+  // App is normally restarted, clearing its reference to the SavedDevices.
+  saved_devices = service_->GetOrInsert(extension_->id());
+  ASSERT_FALSE(saved_devices->IsRegistered(device0));
+  device_entries = service_->GetAllDevices(extension_->id());
+  ASSERT_EQ(0U, device_entries.size());
+}
+
+TEST_F(SavedDevicesServiceTest, LoadPrefs) {
+  scoped_ptr<base::Value> prefs_value = base::test::ParseJson(
+      "["
+      "  {"
+      "    \"product_id\": 0,"
+      "    \"serial_number\": \"ABCDE\","
+      "    \"type\": \"usb\","
+      "    \"vendor_id\": 0"
+      "  }"
+      "]");
+  env_.GetExtensionPrefs()->UpdateExtensionPref(
+      extension_->id(), "devices", prefs_value.release());
+
+  SavedDevicesService::SavedDevices* saved_devices =
+      service_->GetOrInsert(extension_->id());
+  ASSERT_TRUE(saved_devices->IsRegistered(device0));
+  ASSERT_FALSE(saved_devices->IsRegistered(device1));
+  ASSERT_FALSE(saved_devices->IsRegistered(device2));
+  ASSERT_FALSE(saved_devices->IsRegistered(device3));
+}
+
+}  // namespace apps