Refactor BluetoothAllowedDevicesMap

content/shell/browser/layout_test/blink_test_controller.cc

 // Copyright 2013 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/shell/browser/layout_test/blink_test_controller.h"
 
 #include <stddef.h>
 
 #include <iostream>
 #include <set>
 #include <utility>
 
 #include "base/base64.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/location.h"
 #include "base/logging.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/stl_util.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/stringprintf.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "build/build_config.h"
+#include "content/public/browser/bluetooth_allowed_devices_map_base.h"
+#include "content/public/browser/browser_context.h"
 #include "content/public/browser/devtools_agent_host.h"
 #include "content/public/browser/dom_storage_context.h"
 #include "content/public/browser/gpu_data_manager.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
@@ skipping 372 matching lines @@
 std::unique_ptr<BluetoothChooser> BlinkTestController::RunBluetoothChooser(
     RenderFrameHost* frame,
     const BluetoothChooser::EventHandler& event_handler) {
   if (bluetooth_chooser_factory_) {
     return bluetooth_chooser_factory_->RunBluetoothChooser(frame,
                                                            event_handler);
   }
   return base::MakeUnique<LayoutTestFirstDeviceBluetoothChooser>(event_handler);
 }
 
+BluetoothAllowedDevicesMapBase*
+BlinkTestController::GetBluetoothAllowedDevicesMap(
+    BrowserContext* browser_context) {
+  auto iter =
+      browser_contenxt_to_bluetooth_allowed_devices_map_.find(browser_context);
+  if (iter == browser_contenxt_to_bluetooth_allowed_devices_map_.end()) {
+    iter = browser_contenxt_to_bluetooth_allowed_devices_map_.insert(
+        iter,
+        std::make_pair(browser_context,
+                       base::MakeUnique<BluetoothAllowedDevicesMapBase>()));
+  }
+  return iter->second.get();
+}
+
 bool BlinkTestController::OnMessageReceived(const IPC::Message& message) {
   DCHECK(CalledOnValidThread());
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(BlinkTestController, message)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_PrintMessage, OnPrintMessage)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_TextDump, OnTextDump)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_InitiateLayoutDump,
                         OnInitiateLayoutDump)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_ImageDump, OnImageDump)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_AudioDump, OnAudioDump)
@@ skipping 14 matching lines @@
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_CloseRemainingWindows,
                         OnCloseRemainingWindows)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_ResetDone, OnResetDone)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_LeakDetectionDone, OnLeakDetectionDone)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_SetBluetoothManualChooser,
                         OnSetBluetoothManualChooser)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_GetBluetoothManualChooserEvents,
                         OnGetBluetoothManualChooserEvents)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_SendBluetoothManualChooserEvent,
                         OnSendBluetoothManualChooserEvent)
+    IPC_MESSAGE_HANDLER(ShellViewHostMsg_ResetBluetoothAllowedDevicesMap,
+                        OnResetBluetoothAllowedDevicesMap)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled;
 }
 
 bool BlinkTestController::OnMessageReceived(
     const IPC::Message& message,
     RenderFrameHost* render_frame_host) {
   bool handled = true;
@@ skipping 505 matching lines @@
 mojom::LayoutTestControl* BlinkTestController::GetLayoutTestControlPtr(
     RenderFrameHost* frame) {
   if (layout_test_control_map_.find(frame) == layout_test_control_map_.end()) {
     frame->GetRemoteAssociatedInterfaces()->GetInterface(
         &layout_test_control_map_[frame]);
   }
   DCHECK(layout_test_control_map_[frame].get());
   return layout_test_control_map_[frame].get();
 }
 
+void BlinkTestController::OnResetBluetoothAllowedDevicesMap() {
+  browser_contenxt_to_bluetooth_allowed_devices_map_.clear();
+}
+
 }  // namespace content