Forward BrowserAccessibilityWin table APIs to AXPlatformNodeWin.

ui/accessibility/platform/test_ax_node_wrapper.cc

 // 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/containers/hash_tables.h"
 #include "ui/accessibility/ax_action_data.h"
 #include "ui/accessibility/platform/test_ax_node_wrapper.h"
 #include "ui/gfx/geometry/rect_conversions.h"
 
 namespace ui {
@@ skipping 123 matching lines @@
 gfx::NativeViewAccessible TestAXNodeWrapper::HitTestSync(int x, int y) {
   TestAXNodeWrapper* wrapper = HitTestSyncInternal(x, y);
   return wrapper ? wrapper->ax_platform_node()->GetNativeViewAccessible()
                  : nullptr;
 }
 
 gfx::NativeViewAccessible TestAXNodeWrapper::GetFocus() {
   return nullptr;
 }
 
+// Walk the AXTree and ensure that all wrappers are created
+void TestAXNodeWrapper::BuildAllWrappers(AXTree* tree, AXNode* node) {
+  for (int i = 0; i < node->child_count(); i++) {
+    auto* child = node->children()[i];
+    TestAXNodeWrapper::GetOrCreate(tree, child);
+
+    BuildAllWrappers(tree, child);
+  }
+}
+
 AXPlatformNode* TestAXNodeWrapper::GetFromNodeID(int32_t id) {
+  // Force creating all of the wrappers for this tree.
+  BuildAllWrappers(tree_, node_);
+
+  for (auto it = g_node_to_wrapper_map.begin();
+       it != g_node_to_wrapper_map.end(); ++it) {
+    AXNode* node = it->first;
+    if (node->id() == id) {
+      TestAXNodeWrapper* wrapper = it->second;
+      return wrapper->ax_platform_node();
+    }
+  }
   return nullptr;
 }
 
 gfx::AcceleratedWidget
 TestAXNodeWrapper::GetTargetForNativeAccessibilityEvent() {
   return gfx::kNullAcceleratedWidget;
 }
 
 bool TestAXNodeWrapper::AccessibilityPerformAction(
     const ui::AXActionData& data) {
   return true;
 }
 
 bool TestAXNodeWrapper::ShouldIgnoreHoveredStateForTesting() {
   return true;
 }
 
 TestAXNodeWrapper::TestAXNodeWrapper(AXTree* tree, AXNode* node)
     : tree_(tree),
       node_(node),
       platform_node_(AXPlatformNode::Create(this)) {
 }
 
 }  // namespace ui