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
AccessibilityNotificationWaiter(Shell * shell)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 event_target_id_(0) {
28 WebContents* web_contents = shell_->web_contents();
29 view_host_ = static_cast<RenderViewHostImpl*>(
30 web_contents->GetRenderViewHost());
31 view_host_->SetAccessibilityCallbackForTesting(
32 base::Bind(&AccessibilityNotificationWaiter::OnAccessibilityEvent,
33 weak_factory_.GetWeakPtr()));
34 }
35
AccessibilityNotificationWaiter(Shell * shell,AccessibilityMode accessibility_mode,ui::AXEvent event_type)36 AccessibilityNotificationWaiter::AccessibilityNotificationWaiter(
37 Shell* shell,
38 AccessibilityMode accessibility_mode,
39 ui::AXEvent event_type)
40 : shell_(shell),
41 event_to_wait_for_(event_type),
42 loop_runner_(new MessageLoopRunner()),
43 weak_factory_(this),
44 event_target_id_(0) {
45 WebContents* web_contents = shell_->web_contents();
46 view_host_ = static_cast<RenderViewHostImpl*>(
47 web_contents->GetRenderViewHost());
48 view_host_->SetAccessibilityCallbackForTesting(
49 base::Bind(&AccessibilityNotificationWaiter::OnAccessibilityEvent,
50 weak_factory_.GetWeakPtr()));
51 view_host_->AddAccessibilityMode(accessibility_mode);
52 }
53
~AccessibilityNotificationWaiter()54 AccessibilityNotificationWaiter::~AccessibilityNotificationWaiter() {
55 }
56
WaitForNotification()57 void AccessibilityNotificationWaiter::WaitForNotification() {
58 loop_runner_->Run();
59
60 // Each loop runner can only be called once. Create a new one in case
61 // the caller wants to call this again to wait for the next notification.
62 loop_runner_ = new MessageLoopRunner();
63 }
64
GetAXTree() const65 const ui::AXTree& AccessibilityNotificationWaiter::GetAXTree() const {
66 return view_host_->ax_tree_for_testing();
67 }
68
OnAccessibilityEvent(ui::AXEvent event_type,int event_target_id)69 void AccessibilityNotificationWaiter::OnAccessibilityEvent(
70 ui::AXEvent event_type, int event_target_id) {
71 if (!IsAboutBlank() && (event_to_wait_for_ == ui::AX_EVENT_NONE ||
72 event_to_wait_for_ == event_type)) {
73 event_target_id_ = event_target_id;
74 loop_runner_->Quit();
75 }
76 }
77
IsAboutBlank()78 bool AccessibilityNotificationWaiter::IsAboutBlank() {
79 // Skip any accessibility notifications related to "about:blank",
80 // to avoid a possible race condition between the test beginning
81 // listening for accessibility events and "about:blank" loading.
82 const ui::AXNodeData& root = GetAXTree().GetRoot()->data();
83 for (size_t i = 0; i < root.string_attributes.size(); ++i) {
84 if (root.string_attributes[i].first != ui::AX_ATTR_DOC_URL)
85 continue;
86 const std::string& doc_url = root.string_attributes[i].second;
87 return doc_url == url::kAboutBlankURL;
88 }
89 return false;
90 }
91
92 } // namespace content
93