[Reland 1] Save pdf file instead of web page through context menu for embedded pdf file

chrome/browser/renderer_context_menu/render_view_context_menu_browsertest.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 "chrome/browser/renderer_context_menu/render_view_context_menu.h"
 
 #include <algorithm>
 #include <memory>
 #include <string>
 #include <utility>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
 #include "base/strings/string16.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/thread_restrictions.h"
 #include "build/build_config.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
+#include "chrome/browser/pdf/pdf_extension_test_util.h"
 #include "chrome/browser/profiles/profile_attributes_entry.h"
 #include "chrome/browser/profiles/profile_attributes_storage.h"
 #include "chrome/browser/profiles/profile_window.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu_browsertest_util.h"
 #include "chrome/browser/renderer_context_menu/render_view_context_menu_test_util.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/render_messages.h"
 #include "chrome/common/thumbnail_capturer.mojom.h"
+#include "chrome/grit/generated_resources.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/search_test_utils.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/data_reduction_proxy/core/common/data_reduction_proxy_switches.h"
+#include "components/guest_view/browser/guest_view_manager_delegate.h"
+#include "components/guest_view/browser/test_guest_view_manager.h"
 #include "components/search_engines/template_url_data.h"
 #include "components/search_engines/template_url_service.h"
 #include "content/public/browser/browser_message_filter.h"
+#include "content/public/browser/browser_plugin_guest_manager.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_frame_host.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 "content/public/test/test_utils.h"
+#include "extensions/browser/api/extensions_api_client.h"
+#include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h"
+#include "extensions/browser/guest_view/mime_handler_view/test_mime_handler_view_guest.h"
 #include "net/base/load_flags.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_filter.h"
 #include "net/url_request/url_request_interceptor.h"
 #include "services/service_manager/public/cpp/interface_provider.h"
 #include "third_party/WebKit/public/platform/WebInputEvent.h"
 #include "third_party/WebKit/public/web/WebContextMenuData.h"
 #include "ui/base/models/menu_model.h"
 
 using content::WebContents;
+using extensions::MimeHandlerViewGuest;
+using extensions::TestMimeHandlerViewGuest;
 
 namespace {
 
 class ContextMenuBrowserTest : public InProcessBrowserTest {
  public:
   ContextMenuBrowserTest() {}
 
  protected:
   std::unique_ptr<TestRenderViewContextMenu> CreateContextMenuMediaTypeNone(
       const GURL& unfiltered_url,
@@ skipping 41 matching lines @@
   Profile* CreateSecondaryProfile(int profile_num) {
     base::ThreadRestrictions::ScopedAllowIO allow_io;
     ProfileManager* profile_manager = g_browser_process->profile_manager();
     base::FilePath profile_path = profile_manager->user_data_dir();
     profile_path = profile_path.AppendASCII(
         base::StringPrintf("New Profile %d", profile_num));
     return profile_manager->GetProfile(profile_path);
   }
 };
 
+class PdfPluginContextMenuBrowserTest : public InProcessBrowserTest {
+ public:
+  PdfPluginContextMenuBrowserTest() {}
+
+  void SetUpOnMainThread() override {
+    guest_view::GuestViewManager::set_factory_for_testing(&factory_);
+    test_guest_view_manager_ = static_cast<guest_view::TestGuestViewManager*>(
+        guest_view::GuestViewManager::CreateWithDelegate(
+            browser()->profile(),
+            extensions::ExtensionsAPIClient::Get()
+                ->CreateGuestViewManagerDelegate(browser()->profile())));
+  }
+
+ protected:
+  guest_view::TestGuestViewManager* test_guest_view_manager() const {
+    return test_guest_view_manager_;
+  }
+
+  // Helper function for testing context menu of a pdf plugin inside a web page.
+  void TestContextMenuOfPdfInsideWebPage(
+      const base::FilePath::CharType* file_name) {
+    // Load a page with pdf file inside.
+    GURL page_url = ui_test_utils::GetTestUrl(
+        base::FilePath(FILE_PATH_LITERAL("pdf")), base::FilePath(file_name));
+    ui_test_utils::NavigateToURL(browser(), page_url);
+
+    WebContents* web_contents =
+        browser()->tab_strip_model()->GetActiveWebContents();
+    // Prepare to load a pdf plugin inside.
+    test_guest_view_manager_->RegisterTestGuestViewType<MimeHandlerViewGuest>(
+        base::Bind(&TestMimeHandlerViewGuest::Create));
+    ASSERT_TRUE(content::ExecuteScript(
+        web_contents,
+        "setTimeout(\"var l = document.getElementById('link1');"

[lazyboy, 2017/05/17 16:34:44]

Is this setTimeout necessary? Add a comment explaining the reason if it is so.
Generally avoid these if possible because they hint that we could still hit race
condition at times.

[Wei Li, 2017/05/18 23:25:28]

Since wait() functions are added, I think this can be removed. Done.

+        "l.click();\", 0)"));
+
+    // Wait until the guest contents of the PDF plugin is created,
+    // and then attached.
+    WebContents* guest_contents =
+        test_guest_view_manager_->WaitForSingleGuestCreated();
+    TestMimeHandlerViewGuest* guest = static_cast<TestMimeHandlerViewGuest*>(
+        extensions::MimeHandlerViewGuest::FromWebContents(guest_contents));
+    ASSERT_TRUE(guest);
+    guest->WaitForGuestAttached();
+    ASSERT_NE(web_contents, guest_contents);
+    // Get the pdf plugin's main frame.
+    content::RenderFrameHost* frame = guest_contents->GetMainFrame();
+    ASSERT_TRUE(frame);
+
+    content::ContextMenuParams params;
+    params.page_url = page_url;
+    params.frame_url = frame->GetLastCommittedURL();
+    params.frame_page_state =
+        content::PageState::CreateFromURL(params.frame_url);
+    params.media_type = blink::WebContextMenuData::kMediaTypePlugin;
+    TestRenderViewContextMenu menu(frame, params);
+    menu.Init();
+
+    // The full page related items such as 'reload' should not be displayed.
+    ASSERT_FALSE(menu.IsItemPresent(IDC_RELOAD));
+  }
+
+ private:
+  guest_view::TestGuestViewManagerFactory factory_;
+  guest_view::TestGuestViewManager* test_guest_view_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(PdfPluginContextMenuBrowserTest);
+};
+
 IN_PROC_BROWSER_TEST_F(ContextMenuBrowserTest,
                        OpenEntryPresentForNormalURLs) {
   std::unique_ptr<TestRenderViewContextMenu> menu =
       CreateContextMenuMediaTypeNone(GURL("http://www.google.com/"),
                                      GURL("http://www.google.com/"));
 
   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKNEWTAB));
   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKNEWWINDOW));
   ASSERT_TRUE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_COPYLINKLOCATION));
   ASSERT_FALSE(menu->IsItemPresent(IDC_CONTENT_CONTEXT_OPENLINKINPROFILE));
@@ skipping 527 matching lines @@
   thumbnail_capturer->RequestThumbnailForContextNode(
       0, gfx::Size(2048, 2048),
       base::Bind(callback, &response_received, run_loop.QuitClosure()));
   run_loop.Run();
 
   // The browser should receive a response from the renderer, because the
   // renderer should not crash.
   ASSERT_TRUE(response_received);
 }
 
+IN_PROC_BROWSER_TEST_F(PdfPluginContextMenuBrowserTest,
+                       FullPagePdfHasPageItems) {
+  // Load a pdf page.
+  GURL page_url =
+      ui_test_utils::GetTestUrl(base::FilePath(FILE_PATH_LITERAL("pdf")),
+                                base::FilePath(FILE_PATH_LITERAL("test.pdf")));
+  ui_test_utils::NavigateToURL(browser(), page_url);
+
+  // Get the pdf plugin's WebContents and its main frame.
+  WebContents* web_contents =
+      browser()->tab_strip_model()->GetActiveWebContents();
+  pdf_extension_test_util::EnsurePDFHasLoaded(web_contents);
+  content::BrowserPluginGuestManager* guest_manager =
+      web_contents->GetBrowserContext()->GetGuestManager();
+  WebContents* guest_contents = guest_manager->GetFullPageGuest(web_contents);
+  ASSERT_TRUE(guest_contents);
+  content::RenderFrameHost* frame = guest_contents->GetMainFrame();
+  ASSERT_TRUE(frame);
+  ASSERT_NE(frame, web_contents->GetMainFrame());
+
+  content::ContextMenuParams params;
+  params.page_url = page_url;
+  params.frame_url = frame->GetLastCommittedURL();
+  params.frame_page_state = content::PageState::CreateFromURL(params.frame_url);
+  params.media_type = blink::WebContextMenuData::kMediaTypePlugin;
+  TestRenderViewContextMenu menu(frame, params);
+  menu.Init();
+
+  // The full page related items such as 'reload' should be there.
+  ASSERT_TRUE(menu.IsItemPresent(IDC_RELOAD));
+}
+
+IN_PROC_BROWSER_TEST_F(PdfPluginContextMenuBrowserTest,
+                       EmbeddedPdfHasNoPageItems) {
+  TestContextMenuOfPdfInsideWebPage(FILE_PATH_LITERAL("test-embed-pdf.html"));
+}
+
+IN_PROC_BROWSER_TEST_F(PdfPluginContextMenuBrowserTest,
+                       IframedPdfHasNoPageItems) {
+  TestContextMenuOfPdfInsideWebPage(FILE_PATH_LITERAL("test-iframe-pdf.html"));
+}
+
 class LoadImageRequestInterceptor : public net::URLRequestInterceptor {
  public:
   LoadImageRequestInterceptor() : num_requests_(0),
                                   requests_to_wait_for_(-1),
                                   weak_factory_(this) {
   }
 
   ~LoadImageRequestInterceptor() override {}
 
   // net::URLRequestInterceptor implementation
@@ skipping 107 matching lines @@
 IN_PROC_BROWSER_TEST_F(LoadImageBrowserTest, LoadImage) {
   static const char kValidImage[] = "/load_image/image.png";
   SetupAndLoadImagePage(kValidImage);
   AddLoadImageInterceptor(kValidImage);
   AttemptLoadImage();
   interceptor_->WaitForRequests(1);
   EXPECT_EQ(1, interceptor_->num_requests());
 }
 
 }  // namespace