Add PDFExtensionTest.DownloadPDFButton

components/pdf/browser/pdf_web_contents_helper.h

 // 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.
 
 #ifndef COMPONENTS_PDF_BROWSER_PDF_WEB_CONTENTS_HELPER_H_
 #define COMPONENTS_PDF_BROWSER_PDF_WEB_CONTENTS_HELPER_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 21 matching lines @@
       content::WebContents* contents,
       std::unique_ptr<PDFWebContentsHelperClient> client);
 
   OpenPDFInReaderPromptClient* open_in_reader_prompt() const {
     return open_in_reader_prompt_.get();
   }
 
   void ShowOpenInReaderPrompt(
       std::unique_ptr<OpenPDFInReaderPromptClient> prompt);
 
+  void SetClientForTest(std::unique_ptr<PDFWebContentsHelperClient> client);
+
  private:
   PDFWebContentsHelper(content::WebContents* web_contents,
                        std::unique_ptr<PDFWebContentsHelperClient> client);
   ~PDFWebContentsHelper() override;
 
   // content::WebContentsObserver overrides:
   bool OnMessageReceived(const IPC::Message& message,
                          content::RenderFrameHost* render_frame_host) override;
   void DidNavigateMainFrame(
       const content::LoadCommittedDetails& details,
@@ skipping 11 matching lines @@
   // The model for the confirmation prompt to open a PDF in Adobe Reader.
   std::unique_ptr<OpenPDFInReaderPromptClient> open_in_reader_prompt_;
   std::unique_ptr<PDFWebContentsHelperClient> client_;
 
   DISALLOW_COPY_AND_ASSIGN(PDFWebContentsHelper);
 };
 
 }  // namespace pdf
 
 #endif  // COMPONENTS_PDF_BROWSER_PDF_WEB_CONTENTS_HELPER_H_