Refactor BluetoothAllowedDevicesMap

content/browser/bluetooth/frame_connected_bluetooth_devices_unittest.cc

 // Copyright 2016 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 "content/browser/bluetooth/frame_connected_bluetooth_devices.h"
 
+#include <memory>
+
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "content/browser/bluetooth/web_bluetooth_service_impl.h"
+#include "content/public/browser/bluetooth_allowed_devices_map_base.h"
+#include "content/public/browser/web_contents_delegate.h"
 #include "content/test/test_render_view_host.h"
 #include "content/test/test_web_contents.h"
 #include "device/bluetooth/bluetooth_gatt_connection.h"
 #include "device/bluetooth/test/mock_bluetooth_adapter.h"
 #include "device/bluetooth/test/mock_bluetooth_device.h"
 #include "device/bluetooth/test/mock_bluetooth_gatt_connection.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace content {
@@ skipping 11 matching lines @@
 namespace {
 
 const WebBluetoothDeviceId kDeviceId0("000000000000000000000A==");
 constexpr char kDeviceAddress0[] = "0";
 constexpr char kDeviceName0[] = "Device0";
 
 const WebBluetoothDeviceId kDeviceId1("111111111111111111111A==");
 constexpr char kDeviceAddress1[] = "1";
 constexpr char kDeviceName1[] = "Device1";
 
+class MockWebContentsDelegate : public WebContentsDelegate {
+ public:
+  BluetoothAllowedDevicesMapBase* GetBluetoothDevicesMap(
+      RenderFrameHost* frame) override {
+    if (!bluetooth_allowed_devices_map_) {
+      bluetooth_allowed_devices_map_.reset(
+          new BluetoothAllowedDevicesMapBase());
+    }
+    return bluetooth_allowed_devices_map_.get();
+  }
+
+  std::unique_ptr<BluetoothAllowedDevicesMapBase>
+      bluetooth_allowed_devices_map_;
+};
+
 }  // namespace
 
 class FrameConnectedBluetoothDevicesTest
     : public RenderViewHostImplTestHarness {
  public:
   FrameConnectedBluetoothDevicesTest()
       : adapter_(new NiceMockBluetoothAdapter()),
         device0_(adapter_.get(),
                  0 /* class */,
                  kDeviceName0,
@@ skipping 11 matching lines @@
         .WillByDefault(Return(&device0_));
     ON_CALL(*adapter_, GetDevice(StrEq(kDeviceAddress1)))
         .WillByDefault(Return(&device1_));
   }
 
   ~FrameConnectedBluetoothDevicesTest() override {}
 
   void SetUp() override {
     RenderViewHostImplTestHarness::SetUp();
 
+    contents()->SetDelegate(&mock_web_contents_delegate_);
+
     // Create subframe to simulate two maps on the same WebContents.
     contents()->GetMainFrame()->InitializeRenderFrameIfNeeded();
     TestRenderFrameHost* subframe =
         contents()->GetMainFrame()->AppendChild("bluetooth_frame");
     subframe->InitializeRenderFrameIfNeeded();
 
     // Simulate two frames each connected to a bluetooth service.
     service0_ =
         contents()->GetMainFrame()->CreateWebBluetoothServiceForTesting();
     map0_ = service0_->connected_devices_.get();
@@ skipping 30 matching lines @@
   FrameConnectedBluetoothDevices* map0_;
   WebBluetoothServiceImpl* service0_;
 
   FrameConnectedBluetoothDevices* map1_;
   WebBluetoothServiceImpl* service1_;
 
  private:
   scoped_refptr<NiceMockBluetoothAdapter> adapter_;
   NiceMockBluetoothDevice device0_;
   NiceMockBluetoothDevice device1_;
+  MockWebContentsDelegate mock_web_contents_delegate_;
 };
 
 TEST_F(FrameConnectedBluetoothDevicesTest, Insert_Once) {
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
 
   EXPECT_TRUE(contents()->IsConnectedToBluetoothDevice());
   EXPECT_TRUE(map0_->IsConnectedToDeviceWithId(kDeviceId0));
 }
 
 TEST_F(FrameConnectedBluetoothDevicesTest, Insert_Twice) {
@@ skipping 253 matching lines @@
 TEST_F(FrameConnectedBluetoothDevicesTest,
        DestroyedByWebContentsImplDestruction) {
   // Tests that we don't crash when FrameConnectedBluetoothDevices contains
   // at least one device, and it is destroyed while WebContentsImpl is being
   // destroyed.
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
   DeleteContents();
 }
 
 }  // namespace content