Home | History | Annotate | Download | only in renderer 
      1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_
      6 #define CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_
      7 
      8 #include <string>
      9 
     10 #include "base/compiler_specific.h"
     11 #include "content/public/test/mock_render_thread.h"
     12 
     13 namespace base {
     14 class DictionaryValue;
     15 }
     16 
     17 class MockPrinter;
     18 struct ExtensionMsg_ExternalConnectionInfo;
     19 struct PrintHostMsg_DidGetPreviewPageCount_Params;
     20 struct PrintHostMsg_DidPreviewPage_Params;
     21 struct PrintHostMsg_DidPrintPage_Params;
     22 struct PrintHostMsg_ScriptedPrint_Params;
     23 struct PrintMsg_PrintPages_Params;
     24 struct PrintMsg_Print_Params;
     25 
     26 // Extends content::MockRenderThread to know about printing and
     27 // extension messages.
     28 class ChromeMockRenderThread : public content::MockRenderThread {
     29  public:
     30   ChromeMockRenderThread();
     31   virtual ~ChromeMockRenderThread();
     32 
     33   // content::RenderThread overrides.
     34   virtual scoped_refptr<base::MessageLoopProxy> GetIOMessageLoopProxy()
     35       OVERRIDE;
     36 
     37   //////////////////////////////////////////////////////////////////////////
     38   // The following functions are called by the test itself.
     39 
     40   // Set IO message loop proxy.
     41   void set_io_message_loop_proxy(
     42       const scoped_refptr<base::MessageLoopProxy>& proxy);
     43 
     44 #if defined(ENABLE_PRINTING)
     45   // Returns the pseudo-printer instance.
     46   MockPrinter* printer();
     47 
     48   // Call with |response| set to true if the user wants to print.
     49   // False if the user decides to cancel.
     50   void set_print_dialog_user_response(bool response);
     51 
     52   // Cancel print preview when print preview has |page| remaining pages.
     53   void set_print_preview_cancel_page_number(int page);
     54 
     55   // Get the number of pages to generate for print preview.
     56   int print_preview_pages_remaining() const;
     57 #endif
     58 
     59  private:
     60   // Overrides base class implementation to add custom handling for
     61   // print and extensions.
     62   virtual bool OnMessageReceived(const IPC::Message& msg) OVERRIDE;
     63 
     64   // The callee expects to be returned a valid channel_id.
     65   void OnOpenChannelToExtension(int routing_id,
     66                                 const ExtensionMsg_ExternalConnectionInfo& info,
     67                                 const std::string& channel_name,
     68                                 bool include_tls_channel_id,
     69                                 int* port_id);
     70 
     71 #if defined(ENABLE_PRINTING)
     72 #if defined(OS_CHROMEOS) || defined(OS_ANDROID)
     73   void OnAllocateTempFileForPrinting(int render_view_id,
     74                                      base::FileDescriptor* renderer_fd,
     75                                      int* browser_fd);
     76   void OnTempFileForPrintingWritten(int render_view_id, int browser_fd);
     77 #endif
     78 
     79   // PrintWebViewHelper expects default print settings.
     80   void OnGetDefaultPrintSettings(PrintMsg_Print_Params* setting);
     81 
     82   // PrintWebViewHelper expects final print settings from the user.
     83   void OnScriptedPrint(const PrintHostMsg_ScriptedPrint_Params& params,
     84                        PrintMsg_PrintPages_Params* settings);
     85 
     86   void OnDidGetPrintedPagesCount(int cookie, int number_pages);
     87   void OnDidPrintPage(const PrintHostMsg_DidPrintPage_Params& params);
     88   void OnDidGetPreviewPageCount(
     89       const PrintHostMsg_DidGetPreviewPageCount_Params& params);
     90   void OnDidPreviewPage(const PrintHostMsg_DidPreviewPage_Params& params);
     91   void OnCheckForCancel(int32 preview_ui_id,
     92                         int preview_request_id,
     93                         bool* cancel);
     94 
     95 
     96   // For print preview, PrintWebViewHelper will update settings.
     97   void OnUpdatePrintSettings(int document_cookie,
     98                              const base::DictionaryValue& job_settings,
     99                              PrintMsg_PrintPages_Params* params,
    100                              bool* canceled);
    101 
    102   // A mock printer device used for printing tests.
    103   scoped_ptr<MockPrinter> printer_;
    104 
    105   // True to simulate user clicking print. False to cancel.
    106   bool print_dialog_user_response_;
    107 
    108   // Simulates cancelling print preview if |print_preview_pages_remaining_|
    109   // equals this.
    110   int print_preview_cancel_page_number_;
    111 
    112   // Number of pages to generate for print preview.
    113   int print_preview_pages_remaining_;
    114 #endif
    115 
    116   scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy_;
    117 
    118   DISALLOW_COPY_AND_ASSIGN(ChromeMockRenderThread);
    119 };
    120 
    121 #endif  // CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_
    122