Take a snapshot of the AXTree

android_webview/javatests/src/org/chromium/android_webview/test/AwContentsTest.java

Index: android_webview/javatests/src/org/chromium/android_webview/test/AwContentsTest.java
diff --git a/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsTest.java b/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsTest.java
index fa4bfd03cde384b244ee33330163d9e20f9d634b..9fdb0612c50959667074b9d96e2d0c45605dc84a 100644
--- a/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsTest.java
+++ b/android_webview/javatests/src/org/chromium/android_webview/test/AwContentsTest.java
@@ -33,6 +33,8 @@ import org.chromium.base.test.util.MinAndroidSdkLevel;
 import org.chromium.content.browser.test.util.CallbackHelper;
 import org.chromium.content.browser.test.util.Criteria;
 import org.chromium.content.browser.test.util.CriteriaHelper;
+import org.chromium.content_public.browser.AccessibilitySnapshotCallback;
+import org.chromium.content_public.browser.AccessibilitySnapshotNode;
 import org.chromium.content_public.browser.LoadUrlParams;
 import org.chromium.net.test.util.TestWebServer;
 
@@ -626,4 +628,40 @@ public class AwContentsTest extends AwTestBase {
         assertEquals("null", executeJavaScriptAndWaitForResult(awContents, mContentsClient,
                 script));
     }
+
+    /**
+     * Verifies that AX tree is returned.
+     */
+    @Feature({"AndroidWebView"})
+    @SmallTest
+    public void testRequestAccessibilitySnapshot() throws Throwable {
+        final AwTestContainerView testView = createAwTestContainerViewOnMainSync(mContentsClient);
+        final AwContents awContents = testView.getAwContents();
+        final CallbackHelper loadHelper = mContentsClient.getOnPageFinishedHelper();
+        loadDataSync(awContents, loadHelper, "<button>Click</button>", "text/html", false);
+
+        final CallbackHelper callbackHelper = new CallbackHelper();
+        final AccessibilitySnapshotCallback callback = new AccessibilitySnapshotCallback() {
+            @Override
+            public void onAccessibilitySnapshot(AccessibilitySnapshotNode root) {
+                assertEquals(1, root.children.size());
+                assertEquals("", root.text);
+                AccessibilitySnapshotNode child = root.children.get(0);
+                assertEquals(1, child.children.size());
+                assertEquals("", child.text);
+                AccessibilitySnapshotNode grandChild = child.children.get(0);
+                assertEquals(0, grandChild.children.size());
+                assertEquals("Click", grandChild.text);
+                callbackHelper.notifyCalled();
+            }
+        };
+
+        runTestOnUiThread(new Runnable() {
+            @Override
+            public void run() {
+                awContents.requestAccessibilitySnapshot(callback);
+            }
+        });
+        callbackHelper.waitForCallback(callbackHelper.getCallCount());
+    }
 }