Adding unit tests for PrerenderDispatcher.

chrome/renderer/prerender/prerender_dispatcher.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_
 #define CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_
 #pragma once
 
 #include <map>
 
 #include "base/compiler_specific.h"
 #include "base/memory/scoped_ptr.h"
+#include "base/gtest_prod_util.h"
 #include "content/public/renderer/render_process_observer.h"
 
 class GURL;
 
 namespace prerender {
 
 class PrerenderingSupport;
 
 // PrerenderDispatcher keeps track of which URLs are being prerendered. There
 // is only one PrerenderDispatcher per render process, and it will only be
 // aware of prerenders that are triggered by this render process.  As well,
 // it holds on to other objects that must exist once per-renderer process,
 // such as the PrerenderingSupport.
 class PrerenderDispatcher : public content::RenderProcessObserver {
  public:
   PrerenderDispatcher();
   virtual ~PrerenderDispatcher();
 
   bool IsPrerenderURL(const GURL & url) const;
 
  private:
+  friend class PrerenderDispatcherTest;
+  FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest, PrerenderDispatcherEmpty);

[gavinp, 2012/05/24 17:39:40]

Do you need these now?  I think you don't, and probably not the #include either.

[dominich, 2012/05/24 18:11:56]

Done.

+  FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest,
+                           PrerenderDispatcherSingleAdd);
+  FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest,
+                           PrerenderDispatcherMultipleAdd);
+  FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest,
+                           PrerenderDispatcherSingleRemove);
+  FRIEND_TEST_ALL_PREFIXES(PrerenderDispatcherTest,
+                           PrerenderDispatcherMultipleRemove);
   void OnAddPrerenderURL(const GURL& url);
   void OnRemovePrerenderURL(const GURL& url);
 
   // RenderProcessObserver:
   virtual bool OnControlMessageReceived(const IPC::Message& message) OVERRIDE;
 
   typedef std::map<GURL, int> PrerenderMap;
   PrerenderMap prerender_urls_;
 
   // There is one PrerenderingSupport object per renderer, and it provides
   // the interface to prerendering to the WebKit platform.
   scoped_ptr<PrerenderingSupport> prerendering_support_;
 };
 
 }  // namespace prerender
 
 #endif  // CHROME_RENDERER_PRERENDER_PRERENDER_DISPATCHER_H_