OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_PRERENDER_PRERENDER_DISPATCHER_H_ | 5 #ifndef CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_ |
6 #define CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_ | 6 #define CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_ |
7 #pragma once | 7 #pragma once |
8 | 8 |
9 #include <map> | 9 #include <map> |
10 | 10 |
11 #include "base/compiler_specific.h" | 11 #include "base/compiler_specific.h" |
| 12 #include "base/gtest_prod_util.h" |
12 #include "content/public/renderer/render_process_observer.h" | 13 #include "content/public/renderer/render_process_observer.h" |
13 | 14 |
14 class GURL; | 15 class GURL; |
15 | 16 |
16 namespace prerender { | 17 namespace prerender { |
17 | 18 |
18 // PrerenderDispatcher keeps track of which URLs are being prerendered. There | 19 // PrerenderDispatcher keeps track of which URLs are being prerendered. There |
19 // is only one PrerenderDispatcher per render process, and it will only be | 20 // is only one PrerenderDispatcher per render process, and it will only be |
20 // aware of prerenders that are triggered by this render process. | 21 // aware of prerenders that are triggered by this render process. |
21 class PrerenderDispatcher : public content::RenderProcessObserver { | 22 class PrerenderDispatcher : public content::RenderProcessObserver { |
22 public: | 23 public: |
23 PrerenderDispatcher(); | 24 PrerenderDispatcher(); |
24 virtual ~PrerenderDispatcher(); | 25 virtual ~PrerenderDispatcher(); |
25 | 26 |
26 bool IsPrerenderURL(const GURL & url) const; | 27 bool IsPrerenderURL(const GURL & url) const; |
27 | 28 |
28 private: | 29 private: |
| 30 FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest, PrerenderDispatcherEmpty); |
| 31 FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest, |
| 32 PrerenderDispatcherSingleAdd); |
| 33 FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest, |
| 34 PrerenderDispatcherMultipleAdd); |
| 35 FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest, |
| 36 PrerenderDispatcherSingleRemove); |
| 37 FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest, |
| 38 PrerenderDispatcherMultipleRemove); |
| 39 |
29 void OnAddPrerenderURL(const GURL& url); | 40 void OnAddPrerenderURL(const GURL& url); |
30 void OnRemovePrerenderURL(const GURL& url); | 41 void OnRemovePrerenderURL(const GURL& url); |
31 | 42 |
32 // RenderProcessObserver: | 43 // RenderProcessObserver: |
33 virtual bool OnControlMessageReceived(const IPC::Message& message) OVERRIDE; | 44 virtual bool OnControlMessageReceived(const IPC::Message& message) OVERRIDE; |
34 | 45 |
35 typedef std::map<GURL, int> PrerenderMap; | 46 typedef std::map<GURL, int> PrerenderMap; |
36 PrerenderMap prerender_urls_; | 47 PrerenderMap prerender_urls_; |
37 }; | 48 }; |
38 | 49 |
39 } // namespace prerender | 50 } // namespace prerender |
40 | 51 |
41 #endif // CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_ | 52 #endif // CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_ |
OLD | NEW |