Enable PDF accessibility when RenderFrame's accessibility mode changes.

chrome/browser/pdf/pdf_extension_test.cc

 // Copyright 2014 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 <stddef.h>
 
 #include <vector>
 
 #include "base/base_paths.h"
 #include "base/files/file_enumerator.h"
@@ skipping 30 matching lines @@
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/plugin_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/render_view_host.h"
 #include "content/public/browser/render_widget_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/common/manifest_handlers/mime_types_handler.h"
 #include "extensions/test/result_catcher.h"
+#include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "url/gurl.h"
 
 #if defined(TOOLKIT_VIEWS) && !defined(OS_MACOSX)
 #include "chrome/browser/ui/views/location_bar/zoom_bubble_view.h"
 #endif
 
 const int kNumberLoadTestParts = 10;
 
+// Using ASSERT_TRUE deliberately instead of ASSERT_EQ or ASSERT_STREQ
+// in order to print a more readable message if the strings differ.
+#define ASSERT_MULTILINE_STREQ(expected, actual) \
+    ASSERT_TRUE(expected == actual) \
+        << "Expected:\n" << expected \
+        << "\n\nActual:\n" << actual
+
 bool GetGuestCallback(content::WebContents** guest_out,
                       content::WebContents* guest) {
   EXPECT_FALSE(*guest_out);
   *guest_out = guest;
   // Return false so that we iterate through all the guests and verify there is
   // only one.
   return false;
 }
 
 class PDFExtensionTest : public ExtensionApiTest,
                          public testing::WithParamInterface<int> {
  public:
   ~PDFExtensionTest() override {}
 
+  void SetUpCommandLine(base::CommandLine* command_line) override {
+    content::IsolateAllSitesForTesting(command_line);
+  }
+
   void SetUpOnMainThread() override {
     ExtensionApiTest::SetUpOnMainThread();
+    host_resolver()->AddRule("*", "127.0.0.1");
     ASSERT_TRUE(embedded_test_server()->Start());
+    content::SetupCrossSiteRedirector(embedded_test_server());
   }
 
   void TearDownOnMainThread() override {
     ASSERT_TRUE(embedded_test_server()->ShutdownAndWaitUntilComplete());
     ExtensionApiTest::TearDownOnMainThread();
   }
 
   bool PdfIsExpectedToFailLoad(const std::string& pdf_file) {
     const char* const kFailingPdfs[] = {
         // TODO(thestig): Investigate why this file doesn't fail when served by
@@ skipping 418 matching lines @@
 #endif
   ASSERT_TRUE(
       content::ExecuteScript(guest_contents, "viewer.viewport.zoomIn();"));
 
   watcher.Wait();
 #if defined(TOOLKIT_VIEWS) && !defined(OS_MACOSX)
   EXPECT_EQ(nullptr, ZoomBubbleView::GetZoomBubble());
 #endif
 }
 
-IN_PROC_BROWSER_TEST_F(PDFExtensionTest, PdfAccessibility) {
-  content::BrowserAccessibilityState::GetInstance()->EnableAccessibility();
-
-  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-bookmarks.pdf"));
-  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
-  ASSERT_TRUE(guest_contents);
-
-  WaitForAccessibilityTreeToContainNodeWithName(guest_contents,
-                                                "1 First Section\r\n");
-  ui::AXTreeUpdate ax_tree = GetAccessibilityTreeSnapshot(guest_contents);
-
+static std::string DumpPdfAccessibilityTree(const ui::AXTreeUpdate& ax_tree) {
   // Create a string representation of the tree starting with the embedded
   // object.
   std::string ax_tree_dump;
   base::hash_map<int32_t, int> id_to_indentation;
   bool found_embedded_object = false;
   for (auto& node : ax_tree.nodes) {
     if (node.role == ui::AX_ROLE_EMBEDDED_OBJECT)
       found_embedded_object = true;
     if (!found_embedded_object)
       continue;
 
     int indent = id_to_indentation[node.id];
     ax_tree_dump += std::string(2 * indent, ' ');
     ax_tree_dump += ui::ToString(node.role);
 
     std::string name = node.GetStringAttribute(ui::AX_ATTR_NAME);
     base::ReplaceChars(name, "\r", "\\r", &name);
     base::ReplaceChars(name, "\n", "\\n", &name);
     if (!name.empty())
       ax_tree_dump += " '" + name + "'";
     ax_tree_dump += "\n";
     for (size_t j = 0; j < node.child_ids.size(); ++j)
       id_to_indentation[node.child_ids[j]] = indent + 1;
   }
 
-  const char* expected_ax_tree =
-      "embeddedObject\n"
-      "  group\n"
-      "    region 'Page 1'\n"
-      "      paragraph\n"
-      "        staticText '1 First Section\\r\\n'\n"
-      "          inlineTextBox '1 '\n"
-      "          inlineTextBox 'First Section\\r\\n'\n"
-      "      paragraph\n"
-      "        staticText 'This is the first section.\\r\\n1'\n"
-      "          inlineTextBox 'This is the first section.\\r\\n'\n"
-      "          inlineTextBox '1'\n"
-      "    region 'Page 2'\n"
-      "      paragraph\n"
-      "        staticText '1.1 First Subsection\\r\\n'\n"
-      "          inlineTextBox '1.1 '\n"
-      "          inlineTextBox 'First Subsection\\r\\n'\n"
-      "      paragraph\n"
-      "        staticText 'This is the first subsection.\\r\\n2'\n"
-      "          inlineTextBox 'This is the first subsection.\\r\\n'\n"
-      "          inlineTextBox '2'\n"
-      "    region 'Page 3'\n"
-      "      paragraph\n"
-      "        staticText '2 Second Section\\r\\n'\n"
-      "          inlineTextBox '2 '\n"
-      "          inlineTextBox 'Second Section\\r\\n'\n"
-      "      paragraph\n"
-      "        staticText '3'\n"
-      "          inlineTextBox '3'\n";
+  return ax_tree_dump;
+}
 
-  // Using ASSERT_TRUE deliberately instead of ASSERT_EQ or ASSERT_STREQ
-  // in order to print a more readable message if the strings differ.
-  ASSERT_TRUE(expected_ax_tree == ax_tree_dump)
-      << "Expected:\n" << expected_ax_tree
-      << "\n\nActual:\n" << ax_tree_dump;
+static const char kExpectedPDFAXTree[] =
+    "embeddedObject\n"
+    "  group\n"
+    "    region 'Page 1'\n"
+    "      paragraph\n"
+    "        staticText '1 First Section\\r\\n'\n"
+    "          inlineTextBox '1 '\n"
+    "          inlineTextBox 'First Section\\r\\n'\n"
+    "      paragraph\n"
+    "        staticText 'This is the first section.\\r\\n1'\n"
+    "          inlineTextBox 'This is the first section.\\r\\n'\n"
+    "          inlineTextBox '1'\n"
+    "    region 'Page 2'\n"
+    "      paragraph\n"
+    "        staticText '1.1 First Subsection\\r\\n'\n"
+    "          inlineTextBox '1.1 '\n"
+    "          inlineTextBox 'First Subsection\\r\\n'\n"
+    "      paragraph\n"
+    "        staticText 'This is the first subsection.\\r\\n2'\n"
+    "          inlineTextBox 'This is the first subsection.\\r\\n'\n"
+    "          inlineTextBox '2'\n"
+    "    region 'Page 3'\n"
+    "      paragraph\n"
+    "        staticText '2 Second Section\\r\\n'\n"
+    "          inlineTextBox '2 '\n"
+    "          inlineTextBox 'Second Section\\r\\n'\n"
+    "      paragraph\n"
+    "        staticText '3'\n"
+    "          inlineTextBox '3'\n";
+
+IN_PROC_BROWSER_TEST_F(PDFExtensionTest, PdfAccessibility) {
+  content::BrowserAccessibilityState::GetInstance()->EnableAccessibility();
+
+  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-bookmarks.pdf"));
+  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  ASSERT_TRUE(guest_contents);
+
+  WaitForAccessibilityTreeToContainNodeWithName(guest_contents,
+                                                "1 First Section\r\n");
+  ui::AXTreeUpdate ax_tree = GetAccessibilityTreeSnapshot(guest_contents);
+  std::string ax_tree_dump = DumpPdfAccessibilityTree(ax_tree);
+  ASSERT_MULTILINE_STREQ(kExpectedPDFAXTree, ax_tree_dump);
 }
+
+IN_PROC_BROWSER_TEST_F(PDFExtensionTest, PdfAccessibilityEnableLater) {
+  // In this test, load the PDF file first, with accessibility off.
+  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-bookmarks.pdf"));
+  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  ASSERT_TRUE(guest_contents);
+
+  // Now enable accessibility globally, and assert that the PDF accessibility
+  // tree loads.
+  EnableAccessibilityForWebContents(guest_contents);
+  WaitForAccessibilityTreeToContainNodeWithName(guest_contents,
+                                                "1 First Section\r\n");
+  ui::AXTreeUpdate ax_tree = GetAccessibilityTreeSnapshot(guest_contents);
+  std::string ax_tree_dump = DumpPdfAccessibilityTree(ax_tree);
+  ASSERT_MULTILINE_STREQ(kExpectedPDFAXTree, ax_tree_dump);
+}
+
+bool RetrieveGuestContents(
+    content::WebContents** out_guest_contents,
+    content::WebContents* in_guest_contents) {
+  *out_guest_contents = in_guest_contents;
+  return true;
+}
+
+IN_PROC_BROWSER_TEST_F(PDFExtensionTest, PdfAccessibilityInIframe) {
+  content::BrowserAccessibilityState::GetInstance()->EnableAccessibility();
+  GURL test_iframe_url(embedded_test_server()->GetURL("/pdf/test-iframe.html"));
+  ui_test_utils::NavigateToURL(browser(), test_iframe_url);
+  content::WebContents* contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  WaitForAccessibilityTreeToContainNodeWithName(contents,
+                                                "1 First Section\r\n");
+
+  content::WebContents* guest_contents = nullptr;
+  content::BrowserPluginGuestManager* guest_manager =
+        contents->GetBrowserContext()->GetGuestManager();
+  guest_manager->ForEachGuest(contents,
+                              base::Bind(&RetrieveGuestContents,
+                                         &guest_contents));
+  ASSERT_TRUE(guest_contents);
+
+  ui::AXTreeUpdate ax_tree = GetAccessibilityTreeSnapshot(guest_contents);
+  std::string ax_tree_dump = DumpPdfAccessibilityTree(ax_tree);
+  ASSERT_MULTILINE_STREQ(kExpectedPDFAXTree, ax_tree_dump);
+}
+
+IN_PROC_BROWSER_TEST_F(PDFExtensionTest, PdfAccessibilityInOOPIF) {
+  content::BrowserAccessibilityState::GetInstance()->EnableAccessibility();
+  GURL test_iframe_url(embedded_test_server()->GetURL(
+      "/pdf/test-cross-site-iframe.html"));
+  ui_test_utils::NavigateToURL(browser(), test_iframe_url);
+  content::WebContents* contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  WaitForAccessibilityTreeToContainNodeWithName(contents,
+                                                "1 First Section\r\n");
+
+  content::WebContents* guest_contents = nullptr;
+  content::BrowserPluginGuestManager* guest_manager =
+        contents->GetBrowserContext()->GetGuestManager();
+  guest_manager->ForEachGuest(contents,
+                              base::Bind(&RetrieveGuestContents,
+                                         &guest_contents));
+  ASSERT_TRUE(guest_contents);
+
+  ui::AXTreeUpdate ax_tree = GetAccessibilityTreeSnapshot(guest_contents);
+  std::string ax_tree_dump = DumpPdfAccessibilityTree(ax_tree);
+  ASSERT_MULTILINE_STREQ(kExpectedPDFAXTree, ax_tree_dump);
+}