Add PDFExtensionTest.DownloadPDFButton

components/pdf/browser/pdf_web_contents_helper.cc

 // Copyright (c) 2012 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 "components/pdf/browser/pdf_web_contents_helper.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/strings/utf_string_conversions.h"
@@ skipping 23 matching lines @@
 
 PDFWebContentsHelper::~PDFWebContentsHelper() {
 }
 
 void PDFWebContentsHelper::ShowOpenInReaderPrompt(
     std::unique_ptr<OpenPDFInReaderPromptClient> prompt) {
   open_in_reader_prompt_ = std::move(prompt);
   UpdateLocationBar();
 }
 
+void PDFWebContentsHelper::SetClientForTest(
+    std::unique_ptr<PDFWebContentsHelperClient> client) {
+  client_ = std::move(client);
+}
+
 bool PDFWebContentsHelper::OnMessageReceived(
     const IPC::Message& message,
     content::RenderFrameHost* render_frame_host) {
   bool handled = true;
   IPC_BEGIN_MESSAGE_MAP(PDFWebContentsHelper, message)
     IPC_MESSAGE_HANDLER(PDFHostMsg_PDFHasUnsupportedFeature,
                         OnHasUnsupportedFeature)
     IPC_MESSAGE_HANDLER(PDFHostMsg_PDFSaveURLAs, OnSaveURLAs)
     IPC_MESSAGE_HANDLER(PDFHostMsg_PDFUpdateContentRestrictions,
                         OnUpdateContentRestrictions)
@@ skipping 25 matching lines @@
   client_->OnSaveURL(web_contents());
   web_contents()->SaveFrame(url, referrer);
 }
 
 void PDFWebContentsHelper::OnUpdateContentRestrictions(
     int content_restrictions) {
   client_->UpdateContentRestrictions(web_contents(), content_restrictions);
 }
 
 }  // namespace pdf