Add PDFExtensionTest.DownloadPDFButton

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"
 #include "base/files/file_util.h"
 #include "base/hash.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/path_service.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/threading/platform_thread.h"
 #include "build/build_config.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/component_loader.h"
 #include "chrome/browser/extensions/extension_apitest.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/pdf/pdf_extension_test_util.h"
-#include "chrome/browser/pdf/pdf_extension_util.h"
 #include "chrome/browser/plugins/plugin_prefs.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_content_client.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "components/pdf/browser/pdf_web_contents_helper.h"
+#include "components/pdf/browser/pdf_web_contents_helper_client.h"
 #include "components/zoom/page_zoom.h"
 #include "components/zoom/test/zoom_test_utils.h"
 #include "components/zoom/zoom_controller.h"
 #include "content/public/browser/browser_accessibility_state.h"
 #include "content/public/browser/browser_plugin_guest_manager.h"
 #include "content/public/browser/download_item.h"
 #include "content/public/browser/download_manager.h"
 #include "content/public/browser/notification_observer.h"
 #include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/notification_service.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
 
+namespace {
+
 const int kNumberLoadTestParts = 10;
 
 #if defined(OS_MACOSX)
 const int kDefaultKeyModifier = blink::WebInputEvent::MetaKey;
 #else
 const int kDefaultKeyModifier = blink::WebInputEvent::ControlKey;
 #endif
 
 // 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 TestPDFWebContentsHelperClient : public pdf::PDFWebContentsHelperClient {
+ public:
+  explicit TestPDFWebContentsHelperClient(content::WebContents* contents)
+      : contents_(contents) {}
+  ~TestPDFWebContentsHelperClient() override {}
+
+  int save_url_count() const { return save_url_count_; }
+
+ private:
+  // PDFWebContentsHelperClient:
+  void UpdateLocationBar(content::WebContents* contents) override {}
+  void UpdateContentRestrictions(content::WebContents* contents,
+                                 int content_restrictions) override {}
+  void OnPDFHasUnsupportedFeature(content::WebContents* contents) override {}
+  void OnSaveURL(content::WebContents* contents) override {
+    EXPECT_EQ(contents_, contents);
+    ++save_url_count_;
+  }
+
+  int save_url_count_ = 0;
+  content::WebContents* const contents_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestPDFWebContentsHelperClient);
+};
+
+}  // namespace
+
 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 {
@@ skipping 149 matching lines @@
         "    window.domAutomationController.send(false);"
         "  if (event.data.type == 'getSelectedTextReply')"
         "    window.domAutomationController.send(true);"
         "});"
         "document.getElementsByTagName('embed')[0].postMessage("
         "    {type: 'getSelectedText'});",
         &success));
     ASSERT_EQ(expect_success, success);
   }
 
+  content::WebContents* SetUpForLinkClickTest() {
+    host_resolver()->AddRule("www.example.com", "127.0.0.1");
+    GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-link.pdf"));
+    return LoadPdfGetGuestContents(test_pdf_url);
+  }
+
   void ConvertPageCoordToScreenCoord(content::WebContents* contents,
                                      gfx::Point* point) {
     ASSERT_TRUE(contents);
     ASSERT_TRUE(content::ExecuteScript(contents,
         "var visiblePage = viewer.viewport.getMostVisiblePage();"
         "var visiblePageDimensions ="
         "    viewer.viewport.getPageScreenRect(visiblePage);"
         "var viewportPosition = viewer.viewport.position;"
         "var screenOffsetX = visiblePageDimensions.x - viewportPosition.x;"
         "var screenOffsetY = visiblePageDimensions.y - viewportPosition.y;"
@@ skipping 10 matching lines @@
         &x));
 
     int y;
     ASSERT_TRUE(content::ExecuteScriptAndExtractInt(
         contents,
         "window.domAutomationController.send(linkScreenPositionY);",
         &y));
 
     point->SetPoint(x, y);
   }
-
 };
 
 IN_PROC_BROWSER_TEST_P(PDFExtensionTest, Load) {
 #if defined(GOOGLE_CHROME_BUILD)
   // Load private PDFs.
   LoadAllPdfsTest("pdf_private", GetParam());
 #endif
   // Load public PDFs.
   LoadAllPdfsTest("pdf", GetParam());
 }
@@ skipping 418 matching lines @@
                               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, LinkCtrlLeftClick) {
-  host_resolver()->AddRule("www.example.com", "127.0.0.1");
-  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-link.pdf"));
-  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  content::WebContents* guest_contents = SetUpForLinkClickTest();
   ASSERT_TRUE(guest_contents);
 
   // The link position of the test-link.pdf in page coordinates is (110, 110).
   // Convert the link position from page coordinates to screen coordinates.
   gfx::Point link_position(110, 110);
   ConvertPageCoordToScreenCoord(guest_contents, &link_position);
 
   content::WebContents* web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
 
@@ skipping 14 matching lines @@
   content::WebContents* new_web_contents =
       browser()->tab_strip_model()->GetWebContentsAt(1);
   ASSERT_TRUE(new_web_contents);
   ASSERT_NE(web_contents, new_web_contents);
 
   const GURL& url = new_web_contents->GetURL();
   ASSERT_EQ(std::string("http://www.example.com/"), url.spec());
 }
 
 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, LinkMiddleClick) {
-  host_resolver()->AddRule("www.example.com", "127.0.0.1");
-  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-link.pdf"));
-  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  content::WebContents* guest_contents = SetUpForLinkClickTest();
   ASSERT_TRUE(guest_contents);
 
   // The link position of the test-link.pdf in page coordinates is (110, 110).
   // Convert the link position from page coordinates to screen coordinates.
   gfx::Point link_position(110, 110);
   ConvertPageCoordToScreenCoord(guest_contents, &link_position);
 
   content::WebContents* web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
 
@@ skipping 14 matching lines @@
   content::WebContents* new_web_contents =
       browser()->tab_strip_model()->GetWebContentsAt(1);
   ASSERT_TRUE(new_web_contents);
   ASSERT_NE(web_contents, new_web_contents);
 
   const GURL& url = new_web_contents->GetURL();
   ASSERT_EQ(std::string("http://www.example.com/"), url.spec());
 }
 
 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, LinkCtrlShiftLeftClick) {
-  host_resolver()->AddRule("www.example.com", "127.0.0.1");
-  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-link.pdf"));
-  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  content::WebContents* guest_contents = SetUpForLinkClickTest();
   ASSERT_TRUE(guest_contents);
 
   // The link position of the test-link.pdf in page coordinates is (110, 110).
   // Convert the link position from page coordinates to screen coordinates.
   gfx::Point link_position(110, 110);
   ConvertPageCoordToScreenCoord(guest_contents, &link_position);
 
   content::WebContents* web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
 
@@ skipping 11 matching lines @@
 
   content::WebContents* active_web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
   ASSERT_NE(web_contents, active_web_contents);
 
   const GURL& url = active_web_contents->GetURL();
   ASSERT_EQ(std::string("http://www.example.com/"), url.spec());
 }
 
 IN_PROC_BROWSER_TEST_F(PDFExtensionTest, LinkShiftMiddleClick) {
-  host_resolver()->AddRule("www.example.com", "127.0.0.1");
-  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test-link.pdf"));
-  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  content::WebContents* guest_contents = SetUpForLinkClickTest();
   ASSERT_TRUE(guest_contents);
 
   // The link position of the test-link.pdf in page coordinates is (110, 110).
   // Convert the link position from page coordinates to screen coordinates.
   gfx::Point link_position(110, 110);
   ConvertPageCoordToScreenCoord(guest_contents, &link_position);
 
   content::WebContents* web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
 
   content::WindowedNotificationObserver observer(
       chrome::NOTIFICATION_TAB_ADDED,
       content::NotificationService::AllSources());
   content::SimulateMouseClickAt(web_contents, blink::WebInputEvent::ShiftKey,
       blink::WebMouseEvent::ButtonMiddle, link_position);
   observer.Wait();
 
   int tab_count = browser()->tab_strip_model()->count();
   ASSERT_EQ(2, tab_count);
 
   content::WebContents* active_web_contents =
       browser()->tab_strip_model()->GetActiveWebContents();
   ASSERT_NE(web_contents, active_web_contents);
 
   const GURL& url = active_web_contents->GetURL();
   ASSERT_EQ(std::string("http://www.example.com/"), url.spec());
 }
+
+IN_PROC_BROWSER_TEST_F(PDFExtensionTest, DownloadPDFButton) {
+  GURL test_pdf_url(embedded_test_server()->GetURL("/pdf/test.pdf"));
+  content::WebContents* guest_contents = LoadPdfGetGuestContents(test_pdf_url);
+  ASSERT_TRUE(guest_contents);
+
+  auto* helper = pdf::PDFWebContentsHelper::FromWebContents(guest_contents);
+  ASSERT_TRUE(helper);
+
+  // Pass ownership to |helper| but retain a raw pointer to check test results.
+  // |client| is alive for the duration of this test.
+  auto* client = new TestPDFWebContentsHelperClient(guest_contents);
+  helper->SetClientForTest(base::WrapUnique(client));
+
+  const char kScript[] = "$('toolbar').fire('save');";
+  ASSERT_TRUE(content::ExecuteScript(guest_contents, kScript));
+
+  // Wait to get the save count. If it never arrives and the test hangs, then
+  // the test has failed. Unfortunately with multiple processes in the PDF
+  // Viewer, it is too hard for the test to be event driven.
+  while (!client->save_url_count())
+    base::PlatformThread::Sleep(base::TimeDelta::FromMilliseconds(100));
+  EXPECT_EQ(1, client->save_url_count());
+}