content/test/accessibility_browser_test_utils.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/test/accessibility_browser_test_utils.h"
   6
   7 #include "base/bind.h"
   8 #include "base/logging.h"
   9 #include "base/strings/string_util.h"
  10 #include "base/strings/utf_string_conversions.h"
  11 #include "content/browser/renderer_host/render_view_host_impl.h"
  12 #include "content/browser/renderer_host/render_widget_host_view_base.h"
  13 #include "content/common/view_message_enums.h"
  14 #include "content/public/browser/web_contents.h"
  15 #include "content/public/common/url_constants.h"
  16 #include "content/public/test/test_utils.h"
  17 #include "content/shell/browser/shell.h"
  18 #include "ui/accessibility/ax_node.h"
  19
  20 namespace content {
  21
  22 AccessibilityNotificationWaiter::AccessibilityNotificationWaiter(Shell* shell)
  23     : shell_(shell),
  24       event_to_wait_for_(ui::AX_EVENT_NONE),
  25       loop_runner_(new MessageLoopRunner()),
  26       weak_factory_(this) {
  27   WebContents* web_contents = shell_->web_contents();
  28   view_host_ = static_cast<RenderViewHostImpl*>(
  29       web_contents->GetRenderViewHost());
  30   view_host_->SetAccessibilityCallbackForTesting(
  31       base::Bind(&AccessibilityNotificationWaiter::OnAccessibilityEvent,
  32                  weak_factory_.GetWeakPtr()));
  33 }
  34
  35 AccessibilityNotificationWaiter::AccessibilityNotificationWaiter(
  36     Shell* shell,
  37     AccessibilityMode accessibility_mode,
  38     ui::AXEvent event_type)
  39     : shell_(shell),
  40       event_to_wait_for_(event_type),
  41       loop_runner_(new MessageLoopRunner()),
  42       weak_factory_(this) {
  43   WebContents* web_contents = shell_->web_contents();
  44   view_host_ = static_cast<RenderViewHostImpl*>(
  45       web_contents->GetRenderViewHost());
  46   view_host_->SetAccessibilityCallbackForTesting(
  47       base::Bind(&AccessibilityNotificationWaiter::OnAccessibilityEvent,
  48                  weak_factory_.GetWeakPtr()));
  49   view_host_->AddAccessibilityMode(accessibility_mode);
  50 }
  51
  52 AccessibilityNotificationWaiter::~AccessibilityNotificationWaiter() {
  53 }
  54
  55 void AccessibilityNotificationWaiter::WaitForNotification() {
  56   loop_runner_->Run();
  57 }
  58
  59 const ui::AXTree& AccessibilityNotificationWaiter::GetAXTree() const {
  60   return view_host_->ax_tree_for_testing();
  61 }
  62
  63 void AccessibilityNotificationWaiter::OnAccessibilityEvent(
  64     ui::AXEvent event_type) {
  65   if (!IsAboutBlank() && (event_to_wait_for_ == ui::AX_EVENT_NONE ||
  66                           event_to_wait_for_ == event_type)) {
  67     loop_runner_->Quit();
  68   }
  69 }
  70
  71 bool AccessibilityNotificationWaiter::IsAboutBlank() {
  72   // Skip any accessibility notifications related to "about:blank",
  73   // to avoid a possible race condition between the test beginning
  74   // listening for accessibility events and "about:blank" loading.
  75   const ui::AXNodeData& root = GetAXTree().GetRoot()->data();
  76   for (size_t i = 0; i < root.string_attributes.size(); ++i) {
  77     if (root.string_attributes[i].first != ui::AX_ATTR_DOC_URL)
  78       continue;
  79     const std::string& doc_url = root.string_attributes[i].second;
  80     return doc_url == kAboutBlankURL;
  81   }
  82   return false;
  83 }
  84
  85 }  // namespace content