Better way to implement FrameConnectedBluetoothDevicesTest

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 "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
+#include "content/browser/bluetooth/web_bluetooth_service_impl.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 {
 
 constexpr char kDeviceId0[] = "0";
 constexpr char kDeviceAddress0[] = "0";
 constexpr char kDeviceName0[] = "Device0";
 
 constexpr char kDeviceId1[] = "1";
 constexpr char kDeviceAddress1[] = "1";
 constexpr char kDeviceName1[] = "Device1";
 
+}  // namespace
+
 class FrameConnectedBluetoothDevicesTest
     : public RenderViewHostImplTestHarness {
  public:
   FrameConnectedBluetoothDevicesTest()
       : adapter_(new NiceMockBluetoothAdapter()),
         device0_(adapter_.get(),
                  0 /* class */,
                  kDeviceName0,
                  kDeviceAddress0,
                  false /* paired */,
                  false /* connected */),
         device1_(adapter_.get(),
                  0 /* class */,
                  kDeviceName1,
                  kDeviceAddress1,
                  false /* paired */,
                  false /* connected */) {
     ON_CALL(*adapter_, GetDevice(_)).WillByDefault(Return(nullptr));
     ON_CALL(*adapter_, GetDevice(StrEq(kDeviceAddress0)))
         .WillByDefault(Return(&device0_));
     ON_CALL(*adapter_, GetDevice(StrEq(kDeviceAddress1)))
         .WillByDefault(Return(&device1_));
   }
 
   ~FrameConnectedBluetoothDevicesTest() override {}
 
   void SetUp() override {
     RenderViewHostImplTestHarness::SetUp();
-    map0_.reset(new FrameConnectedBluetoothDevices(contents()->GetMainFrame()));
-    map1_.reset(new FrameConnectedBluetoothDevices(contents()->GetMainFrame()));
+
+    // TODO(nick): This test used to work by creating two
+    // FrameConnectedBluetoothDevices instances on the same RenderFrameHost. But
+    // that would never possibly happen in reality! To get two
+    // FrameConnectedBluetoothDevices on the same WebContents, we should create
+    // a child frame.
+    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();
+
+    service1_ = subframe->CreateWebBluetoothServiceForTesting();
+    map1_ = service1_->connected_devices_.get();
   }
 
   void TearDown() override {
-    map1_.reset();
-    map0_.reset();
     RenderViewHostImplTestHarness::TearDown();
   }
 
   std::unique_ptr<NiceMockBluetoothGattConnection> GetConnection(
       const std::string& address) {
     return base::WrapUnique(
         new NiceMockBluetoothGattConnection(adapter_.get(), address));
   }
 
+  void ResetService0() {
+    service0_->ClearState();
+    map0_ = nullptr;
+  }
+
+  void ResetService1() {
+    service1_->ClearState();
+    map1_ = nullptr;
+  }
+
  protected:
-  std::unique_ptr<FrameConnectedBluetoothDevices> map0_;
-  std::unique_ptr<FrameConnectedBluetoothDevices> map1_;
+  FrameConnectedBluetoothDevices* map0_;
+  WebBluetoothServiceImpl* service0_;
+
+  FrameConnectedBluetoothDevices* map1_;
+  WebBluetoothServiceImpl* service1_;
 
  private:
   scoped_refptr<NiceMockBluetoothAdapter> adapter_;
   NiceMockBluetoothDevice device0_;
   NiceMockBluetoothDevice device1_;
 };
 
-}  // namespace
-
 TEST_F(FrameConnectedBluetoothDevicesTest, Insert_Once) {
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
 
   EXPECT_TRUE(contents()->IsConnectedToBluetoothDevice());
   EXPECT_TRUE(map0_->IsConnectedToDeviceWithId(kDeviceId0));
 }
 
 TEST_F(FrameConnectedBluetoothDevicesTest, Insert_Twice) {
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
@@ skipping 217 matching lines @@
   EXPECT_FALSE(contents()->IsConnectedToBluetoothDevice());
   EXPECT_FALSE(map1_->IsConnectedToDeviceWithId(kDeviceId1));
 }
 
 TEST_F(FrameConnectedBluetoothDevicesTest, Destruction_MultipleDevices) {
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
   map0_->Insert(kDeviceId1, GetConnection(kDeviceAddress1));
 
   EXPECT_TRUE(contents()->IsConnectedToBluetoothDevice());
 
-  map0_.reset();
+  ResetService0();
 
   EXPECT_FALSE(contents()->IsConnectedToBluetoothDevice());
 }
 
 TEST_F(FrameConnectedBluetoothDevicesTest, Destruction_MultipleMaps) {
   map0_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
   map0_->Insert(kDeviceId1, GetConnection(kDeviceAddress1));
 
   map1_->Insert(kDeviceId0, GetConnection(kDeviceAddress0));
   map1_->Insert(kDeviceId1, GetConnection(kDeviceAddress1));
 
   EXPECT_TRUE(contents()->IsConnectedToBluetoothDevice());
 
-  map0_.reset();
+  ResetService0();
 
   // WebContents should still be connected because of map1_.
   EXPECT_TRUE(contents()->IsConnectedToBluetoothDevice());
 
-  map1_.reset();
+  ResetService1();
 
   EXPECT_FALSE(contents()->IsConnectedToBluetoothDevice());
 }
 
 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));
-  // A FrameConnectedBluetoothDevices instance is usually owned by a
-  // WebBluetoothServiceImpl instance which is owned by a RenderFrameHost which
-  // is owned by WebContentsImpl. In order to avoid adding unnecessary
-  // complexity to WebBluetoothServiceImpl just so we can perform this test we
-  // add the map directly to a frame and then delete WebContents.
-  contents()->GetMainFrame()->SetFrameConnectedBluetoothDevices(
-      std::move(map0_));
   DeleteContents();
 }
 
 }  // namespace content