| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #ifndef CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ | 5 #ifndef CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ |
| 6 #define CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ | 6 #define CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "content/public/test/mock_render_thread.h" | 10 #include "content/public/test/mock_render_thread.h" |
| 11 | 11 |
| 12 struct ExtensionMsg_ExternalConnectionInfo; | 12 struct ExtensionMsg_ExternalConnectionInfo; |
| 13 | 13 |
| 14 // Extends content::MockRenderThread to know about extension messages. | 14 // Extends content::MockRenderThread to know about extension messages. |
| 15 class ChromeMockRenderThread : public content::MockRenderThread { | 15 class ChromeMockRenderThread : public content::MockRenderThread { |
| 16 public: | 16 public: |
| 17 ChromeMockRenderThread(); | 17 ChromeMockRenderThread(); |
| 18 ~ChromeMockRenderThread() override; | 18 ~ChromeMockRenderThread() override; |
| 19 | 19 |
| 20 // content::RenderThread overrides. | 20 // content::RenderThread overrides. |
| 21 scoped_refptr<base::SingleThreadTaskRunner> GetIOMessageLoopProxy() override; | 21 scoped_refptr<base::SingleThreadTaskRunner> GetIOMessageLoopProxy() override; |
| 22 | 22 |
| 23 ////////////////////////////////////////////////////////////////////////// | 23 ////////////////////////////////////////////////////////////////////////// |
| 24 // The following functions are called by the test itself. | 24 // The following functions are called by the test itself. |
| 25 | 25 |
| 26 // Set IO message loop proxy. | 26 // Set IO message loop proxy. |
| 27 void set_io_message_loop_proxy( | 27 void set_io_message_loop_proxy( |
| 28 const scoped_refptr<base::MessageLoopProxy>& proxy); | 28 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner); |
| 29 | 29 |
| 30 private: | 30 private: |
| 31 // Overrides base class implementation to add custom handling for | 31 // Overrides base class implementation to add custom handling for |
| 32 // print and extensions. | 32 // print and extensions. |
| 33 bool OnMessageReceived(const IPC::Message& msg) override; | 33 bool OnMessageReceived(const IPC::Message& msg) override; |
| 34 | 34 |
| 35 #if defined(ENABLE_EXTENSIONS) | 35 #if defined(ENABLE_EXTENSIONS) |
| 36 // The callee expects to be returned a valid channel_id. | 36 // The callee expects to be returned a valid channel_id. |
| 37 void OnOpenChannelToExtension(int routing_id, | 37 void OnOpenChannelToExtension(int routing_id, |
| 38 const ExtensionMsg_ExternalConnectionInfo& info, | 38 const ExtensionMsg_ExternalConnectionInfo& info, |
| 39 const std::string& channel_name, | 39 const std::string& channel_name, |
| 40 bool include_tls_channel_id, | 40 bool include_tls_channel_id, |
| 41 int* port_id); | 41 int* port_id); |
| 42 #endif | 42 #endif |
| 43 | 43 |
| 44 scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy_; | 44 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; |
| 45 | 45 |
| 46 DISALLOW_COPY_AND_ASSIGN(ChromeMockRenderThread); | 46 DISALLOW_COPY_AND_ASSIGN(ChromeMockRenderThread); |
| 47 }; | 47 }; |
| 48 | 48 |
| 49 #endif // CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ | 49 #endif // CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ |
| OLD | NEW |