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 "base/macros.h" | 10 #include "base/macros.h" |
11 #include "content/public/test/mock_render_thread.h" | 11 #include "content/public/test/mock_render_thread.h" |
| 12 #include "extensions/features/features.h" |
12 | 13 |
13 struct ExtensionMsg_ExternalConnectionInfo; | 14 struct ExtensionMsg_ExternalConnectionInfo; |
14 | 15 |
15 // Extends content::MockRenderThread to know about extension messages. | 16 // Extends content::MockRenderThread to know about extension messages. |
16 class ChromeMockRenderThread : public content::MockRenderThread { | 17 class ChromeMockRenderThread : public content::MockRenderThread { |
17 public: | 18 public: |
18 ChromeMockRenderThread(); | 19 ChromeMockRenderThread(); |
19 ~ChromeMockRenderThread() override; | 20 ~ChromeMockRenderThread() override; |
20 | 21 |
21 // content::RenderThread overrides. | 22 // content::RenderThread overrides. |
22 scoped_refptr<base::SingleThreadTaskRunner> GetIOTaskRunner() override; | 23 scoped_refptr<base::SingleThreadTaskRunner> GetIOTaskRunner() override; |
23 | 24 |
24 ////////////////////////////////////////////////////////////////////////// | 25 ////////////////////////////////////////////////////////////////////////// |
25 // The following functions are called by the test itself. | 26 // The following functions are called by the test itself. |
26 | 27 |
27 void set_io_task_runner( | 28 void set_io_task_runner( |
28 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner); | 29 const scoped_refptr<base::SingleThreadTaskRunner>& task_runner); |
29 | 30 |
30 protected: | 31 protected: |
31 // Overrides base class implementation to add custom handling for | 32 // Overrides base class implementation to add custom handling for |
32 // print and extensions. | 33 // print and extensions. |
33 bool OnMessageReceived(const IPC::Message& msg) override; | 34 bool OnMessageReceived(const IPC::Message& msg) override; |
34 | 35 |
35 private: | 36 private: |
36 #if defined(ENABLE_EXTENSIONS) | 37 #if BUILDFLAG(ENABLE_EXTENSIONS) |
37 // The callee expects to be returned a valid channel_id. | 38 // The callee expects to be returned a valid channel_id. |
38 void OnOpenChannelToExtension(int routing_id, | 39 void OnOpenChannelToExtension(int routing_id, |
39 const ExtensionMsg_ExternalConnectionInfo& info, | 40 const ExtensionMsg_ExternalConnectionInfo& info, |
40 const std::string& channel_name, | 41 const std::string& channel_name, |
41 bool include_tls_channel_id, | 42 bool include_tls_channel_id, |
42 int request_id); | 43 int request_id); |
43 #endif | 44 #endif |
44 | 45 |
45 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; | 46 scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; |
46 | 47 |
47 DISALLOW_COPY_AND_ASSIGN(ChromeMockRenderThread); | 48 DISALLOW_COPY_AND_ASSIGN(ChromeMockRenderThread); |
48 }; | 49 }; |
49 | 50 |
50 #endif // CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ | 51 #endif // CHROME_RENDERER_CHROME_MOCK_RENDER_THREAD_H_ |
OLD | NEW |