Add functions to let tests read and control the Bluetooth chooser state.

content/shell/browser/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/blink_test_controller.h"
 
 #include <iostream>
 
 #include "base/base64.h"
 #include "base/callback.h"
@@ skipping 12 matching lines @@
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/browser/service_worker_context.h"
 #include "content/public/browser/storage_partition.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/url_constants.h"
+#include "content/shell/browser/layout_test/layout_test_bluetooth_chooser_factory.h"
 #include "content/shell/browser/layout_test/layout_test_devtools_frontend.h"
 #include "content/shell/browser/shell.h"
 #include "content/shell/browser/shell_browser_context.h"
 #include "content/shell/browser/shell_content_browser_client.h"
 #include "content/shell/browser/shell_devtools_frontend.h"
 #include "content/shell/common/shell_messages.h"
 #include "content/shell/common/shell_switches.h"
 #include "content/shell/renderer/layout_test/blink_test_helpers.h"
 #include "ui/gfx/codec/png_codec.h"
 
@@ skipping 313 matching lines @@
   RenderViewHost* render_view_host =
       main_window_->web_contents()->GetRenderViewHost();
   render_view_host->Send(
       new ShellViewMsg_NotifyDone(render_view_host->GetRoutingID()));
 }
 
 bool BlinkTestController::IsMainWindow(WebContents* web_contents) const {
   return main_window_ && web_contents == main_window_->web_contents();
 }
 
+scoped_ptr<BluetoothChooser> BlinkTestController::RunBluetoothChooser(
+    WebContents* web_contents,
+    const BluetoothChooser::EventHandler& event_handler,
+    const GURL& origin) {
+  if (bluetooth_chooser_factory_) {
+    return bluetooth_chooser_factory_->RunBluetoothChooser(
+        web_contents, event_handler, origin);
+  }
+  return nullptr;
+}
+
 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_ImageDump, OnImageDump)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_AudioDump, OnAudioDump)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_OverridePreferences,
                         OnOverridePreferences)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_TestFinished, OnTestFinished)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_ClearDevToolsLocalStorage,
                         OnClearDevToolsLocalStorage)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_ShowDevTools, OnShowDevTools)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_CloseDevTools, OnCloseDevTools)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_GoToOffset, OnGoToOffset)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_Reload, OnReload)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_LoadURLForFrame, OnLoadURLForFrame)
     IPC_MESSAGE_HANDLER(ShellViewHostMsg_CaptureSessionHistory,
                         OnCaptureSessionHistory)
     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_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled;
 }
 
 void BlinkTestController::PluginCrashed(const base::FilePath& plugin_path,
                                         base::ProcessId plugin_pid) {
   DCHECK(CalledOnValidThread());
   printer_->AddErrorMessage(
@@ skipping 295 matching lines @@
   }
 
   printer_->AddErrorMessage(
       base::StringPrintf("#LEAK - renderer pid %d (%s)", current_pid_,
                          result.detail.c_str()));
   CHECK(!crash_when_leak_found_);
 
   DiscardMainWindow();
 }
 
+void BlinkTestController::OnSetBluetoothManualChooser(bool enable) {
+  bluetooth_chooser_factory_.reset();
+  if (enable) {
+    bluetooth_chooser_factory_.reset(new LayoutTestBluetoothChooserFactory());
+  }
+}
+
+void BlinkTestController::OnGetBluetoothManualChooserEvents(
+    std::vector<std::string>* events) {
+  if (!bluetooth_chooser_factory_) {
+    printer_->AddErrorMessage(
+        "FAIL: Must call setBluetoothManualChooser before "
+        "getBluetoothManualChooserEvents.");
+    return;
+  }
+  *events = bluetooth_chooser_factory_->GetAndResetEvents();
+}
+
+void BlinkTestController::OnSendBluetoothManualChooserEvent(
+    const std::string& event_name,
+    const std::string& argument) {
+  if (!bluetooth_chooser_factory_) {
+    printer_->AddErrorMessage(
+        "FAIL: Must call setBluetoothManualChooser before "
+        "sendBluetoothManualChooserEvent.");
+    return;
+  }
+  BluetoothChooser::Event event;
+  if (event_name == "cancelled") {
+    event = BluetoothChooser::Event::CANCELLED;
+  } else if (event_name == "selected") {
+    event = BluetoothChooser::Event::SELECTED;
+  } else {
+    printer_->AddErrorMessage(base::StringPrintf(
+        "FAIL: Unexpected sendBluetoothManualChooserEvent() event name '%s'.",
+        event_name.c_str()));
+    return;
+  }
+  bluetooth_chooser_factory_->SendEvent(event, argument);
+}
+
 }  // namespace content