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