Store USB device permissions in website settings.

chrome/browser/usb/web_usb_permission_store_unittest.cc

Index: chrome/browser/usb/web_usb_permission_store_unittest.cc
diff --git a/chrome/browser/usb/web_usb_permission_store_unittest.cc b/chrome/browser/usb/web_usb_permission_store_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..7e3ff38a27a957c32433c4b5118da8070573e0e9
--- /dev/null
+++ b/chrome/browser/usb/web_usb_permission_store_unittest.cc
@@ -0,0 +1,113 @@
+// 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 "chrome/browser/usb/web_usb_permission_store.h"
+#include "chrome/test/base/testing_profile.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "device/core/device_client.h"
+#include "device/usb/mock_usb_device.h"
+#include "device/usb/mock_usb_service.h"
+
+using device::MockUsbDevice;
+using device::UsbDevice;
+
+namespace {
+
+class TestDeviceClient : public device::DeviceClient {
+ public:
+  TestDeviceClient() {}
+  ~TestDeviceClient() override {}
+
+  device::MockUsbService& usb_service() { return usb_service_; }
+
+ private:
+  device::UsbService* GetUsbService() override { return &usb_service_; }
+
+  device::MockUsbService usb_service_;
+};
+
+}  // namespace
+
+class WebUSBPermissionStoreTest : public testing::Test {
+ public:
+  WebUSBPermissionStoreTest() {}
+  ~WebUSBPermissionStoreTest() override {}
+
+ protected:
+  Profile* profile() { return &profile_; }
+  device::MockUsbService& usb_service() { return device_client_.usb_service(); }
+
+ private:
+  content::TestBrowserThreadBundle thread_bundle_;
+  TestDeviceClient device_client_;
+  TestingProfile profile_;
+};
+
+TEST_F(WebUSBPermissionStoreTest, CheckGrantAndRevokePermission) {
+  GURL origin("https://www.google.com");
+  scoped_refptr<UsbDevice> device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "123ABC");
+  usb_service().AddDevice(device);
+  WebUSBPermissionStore* store = WebUSBPermissionStore::Get(profile());
+
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+  store->GrantDevicePermission(origin, device->guid());
+  EXPECT_TRUE(store->HasDevicePermission(origin, device->guid()));
+  store->RevokeDevicePermission(origin, device->guid());
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+}
+
+TEST_F(WebUSBPermissionStoreTest, CheckGrantAndRevokeEphemeralPermission) {
+  GURL origin("https://www.google.com");
+  scoped_refptr<UsbDevice> device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "");
+  scoped_refptr<UsbDevice> other_device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "");
+  usb_service().AddDevice(device);
+  WebUSBPermissionStore* store = WebUSBPermissionStore::Get(profile());
+
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+  store->GrantDevicePermission(origin, device->guid());
+  EXPECT_TRUE(store->HasDevicePermission(origin, device->guid()));
+  store->RevokeDevicePermission(origin, device->guid());
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+}
+
+TEST_F(WebUSBPermissionStoreTest, DisconnectDeviceWithPermission) {
+  GURL origin("https://www.google.com");
+  scoped_refptr<UsbDevice> device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "123ABC");
+  usb_service().AddDevice(device);
+  WebUSBPermissionStore* store = WebUSBPermissionStore::Get(profile());
+
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+  store->GrantDevicePermission(origin, device->guid());
+  EXPECT_TRUE(store->HasDevicePermission(origin, device->guid()));
+  usb_service().RemoveDevice(device);
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+
+  scoped_refptr<UsbDevice> reconnected_device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "123ABC");
+  usb_service().AddDevice(reconnected_device);
+  EXPECT_TRUE(store->HasDevicePermission(origin, reconnected_device->guid()));
+}
+
+TEST_F(WebUSBPermissionStoreTest, DisconnectDeviceWithEphemeralPermission) {
+  GURL origin("https://www.google.com");
+  scoped_refptr<UsbDevice> device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "");
+  usb_service().AddDevice(device);
+  WebUSBPermissionStore* store = WebUSBPermissionStore::Get(profile());
+
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+  store->GrantDevicePermission(origin, device->guid());
+  EXPECT_TRUE(store->HasDevicePermission(origin, device->guid()));
+  usb_service().RemoveDevice(device);
+  EXPECT_FALSE(store->HasDevicePermission(origin, device->guid()));
+
+  scoped_refptr<UsbDevice> reconnected_device =
+      new MockUsbDevice(0, 0, "Google", "Gizmo", "");
+  usb_service().AddDevice(reconnected_device);
+  EXPECT_FALSE(store->HasDevicePermission(origin, reconnected_device->guid()));
+}