Implement UIElementsForSearchPredicate accessibility APIs for OS X.

content/browser/accessibility/one_shot_accessibility_tree_search_unittest.cc

Index: content/browser/accessibility/one_shot_accessibility_tree_search_unittest.cc
diff --git a/content/browser/accessibility/one_shot_accessibility_tree_search_unittest.cc b/content/browser/accessibility/one_shot_accessibility_tree_search_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..abc5112bea866cab3e9a8a451349fd78a78f0c13
--- /dev/null
+++ b/content/browser/accessibility/one_shot_accessibility_tree_search_unittest.cc
@@ -0,0 +1,174 @@
+// Copyright 2015 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 "base/memory/scoped_ptr.h"
+#include "content/browser/accessibility/browser_accessibility.h"
+#include "content/browser/accessibility/browser_accessibility_manager.h"
+#include "content/browser/accessibility/one_shot_accessibility_tree_search.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace content {
+
+namespace {
+
+class TestBrowserAccessibilityManager : public BrowserAccessibilityManager {
+ public:
+  TestBrowserAccessibilityManager(const ui::AXTreeUpdate& initial_tree)
+      : BrowserAccessibilityManager(initial_tree,
+                                    nullptr,
+                                    new BrowserAccessibilityFactory()) {}
+};
+
+}  // namespace
+
+class OneShotAccessibilityTreeSearchTest : public testing::Test {
+ public:
+  OneShotAccessibilityTreeSearchTest() {}
+  ~OneShotAccessibilityTreeSearchTest() override {}
+
+ protected:
+  void SetUp() override;
+
+  scoped_ptr<BrowserAccessibilityManager> tree_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(OneShotAccessibilityTreeSearchTest);
+};
+
+void OneShotAccessibilityTreeSearchTest::SetUp() {
+  ui::AXNodeData root;
+  root.id = 1;
+  root.SetName("Document");
+  root.role = ui::AX_ROLE_ROOT_WEB_AREA;
+  root.state = 0;
+  root.child_ids.push_back(2);
+  root.child_ids.push_back(3);
+  root.child_ids.push_back(6);
+
+  ui::AXNodeData heading;
+  heading.id = 2;
+  heading.SetName("Heading");
+  heading.role = ui::AX_ROLE_HEADING;
+  heading.state = 0;
+
+  ui::AXNodeData list;
+  list.id = 3;
+  list.role = ui::AX_ROLE_LIST;
+  list.state = 0;
+  list.child_ids.push_back(4);
+  list.child_ids.push_back(5);
+
+  ui::AXNodeData list_item_1;
+  list_item_1.id = 4;
+  list_item_1.SetName("Autobots");
+  list_item_1.role = ui::AX_ROLE_LIST_ITEM;
+  list_item_1.state = 0;
+
+  ui::AXNodeData list_item_2;
+  list_item_2.id = 5;
+  list_item_2.SetName("Decepticons");
+  list_item_2.role = ui::AX_ROLE_LIST_ITEM;
+  list_item_2.state = 0;
+
+  ui::AXNodeData footer;
+  footer.id = 6;
+  footer.SetName("Footer");
+  footer.role = ui::AX_ROLE_FOOTER;
+  footer.state = 1 << ui::AX_STATE_OFFSCREEN;
+
+  tree_.reset(new TestBrowserAccessibilityManager(
+      MakeAXTreeUpdate(root, heading, list, list_item_1, list_item_2, footer)));
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, GetAll) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  ASSERT_EQ(6U, search.CountMatches());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, ForwardsWithStartNode) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetStartNode(tree_->GetFromID(4));
+  ASSERT_EQ(2U, search.CountMatches());
+  EXPECT_EQ(5, search.GetMatchAtIndex(0)->GetId());
+  EXPECT_EQ(6, search.GetMatchAtIndex(1)->GetId());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, BackwardsWithStartNode) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetStartNode(tree_->GetFromID(4));
+  search.SetDirection(OneShotAccessibilityTreeSearch::BACKWARDS);
+  ASSERT_EQ(3U, search.CountMatches());
+  EXPECT_EQ(3, search.GetMatchAtIndex(0)->GetId());
+  EXPECT_EQ(2, search.GetMatchAtIndex(1)->GetId());
+  EXPECT_EQ(1, search.GetMatchAtIndex(2)->GetId());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, ResultLimitZero) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetResultLimit(0);
+  ASSERT_EQ(0U, search.CountMatches());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, ResultLimitFive) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetResultLimit(5);
+  ASSERT_EQ(5U, search.CountMatches());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, DescendantsOnly) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetStartNode(tree_->GetFromID(1));
+  search.SetImmediateDescendantsOnly(true);
+  ASSERT_EQ(3U, search.CountMatches());
+  EXPECT_EQ(2, search.GetMatchAtIndex(0)->GetId());
+  EXPECT_EQ(3, search.GetMatchAtIndex(1)->GetId());
+  EXPECT_EQ(6, search.GetMatchAtIndex(2)->GetId());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, VisibleOnly) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetVisibleOnly(true);
+  ASSERT_EQ(5U, search.CountMatches());
+  EXPECT_EQ(1, search.GetMatchAtIndex(0)->GetId());
+  EXPECT_EQ(2, search.GetMatchAtIndex(1)->GetId());
+  EXPECT_EQ(3, search.GetMatchAtIndex(2)->GetId());
+  EXPECT_EQ(4, search.GetMatchAtIndex(3)->GetId());
+  EXPECT_EQ(5, search.GetMatchAtIndex(4)->GetId());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, CaseInsensitiveStringMatch) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.SetSearchText("eCEptiCOn");
+  ASSERT_EQ(1U, search.CountMatches());
+  EXPECT_EQ(5, search.GetMatchAtIndex(0)->GetId());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, OnePredicate) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.AddPredicate([](BrowserAccessibility* start,
+                         BrowserAccessibility* current) {
+    return current->GetRole() == ui::AX_ROLE_LIST_ITEM;
+  });
+  ASSERT_EQ(2U, search.CountMatches());
+  EXPECT_EQ(4, search.GetMatchAtIndex(0)->GetId());
+  EXPECT_EQ(5, search.GetMatchAtIndex(1)->GetId());
+}
+
+TEST_F(OneShotAccessibilityTreeSearchTest, TwoPredicates) {
+  OneShotAccessibilityTreeSearch search(tree_.get());
+  search.AddPredicate([](BrowserAccessibility* start,
+                         BrowserAccessibility* current) {
+    return (current->GetRole() == ui::AX_ROLE_LIST ||
+            current->GetRole() == ui::AX_ROLE_LIST_ITEM);
+  });
+  search.AddPredicate([](BrowserAccessibility* start,
+                         BrowserAccessibility* current) {
+    return (current->GetId() % 2 == 1);
+  });
+  ASSERT_EQ(2U, search.CountMatches());
+  EXPECT_EQ(3, search.GetMatchAtIndex(0)->GetId());
+  EXPECT_EQ(5, search.GetMatchAtIndex(1)->GetId());
+}
+
+}  // namespace content