Browser test for prerendering in general

chrome/browser/prerender/prerender_manager.h

 // Copyright (c) 2011 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_BROWSER_PRERENDER_PRERENDER_MANAGER_H_
 #define CHROME_BROWSER_PRERENDER_PRERENDER_MANAGER_H_
 #pragma once
 
 #include <list>
 #include <vector>
 
 #include "base/ref_counted.h"
 #include "base/scoped_ptr.h"
-#include "base/threading/non_thread_safe.h"
 #include "base/time.h"
+#include "chrome/browser/prerender/prerender_contents.h"
 #include "googleurl/src/gurl.h"
 
-class PrerenderContents;
 class Profile;
 class TabContents;
 
 // PrerenderManager is responsible for initiating and keeping prerendered
 // views of webpages.
-class PrerenderManager : public base::RefCounted<PrerenderManager>,
-                         private base::NonThreadSafe {
+class PrerenderManager : public base::RefCounted<PrerenderManager> {
  public:
   // Owned by a Profile object for the lifetime of the profile.
   explicit PrerenderManager(Profile* profile);
 
   // Preloads the URL supplied.  alias_urls indicates URLs that redirect
   // to the same URL to be preloaded.
   void AddPreload(const GURL& url, const std::vector<GURL>& alias_urls);
 
   // For a given TabContents that wants to navigate to the URL supplied,
   // determines whether a preloaded version of the URL can be used,
@@ skipping 12 matching lines @@
   PrerenderContents* GetEntry(const GURL& url);
 
   base::TimeDelta max_prerender_age() const { return max_prerender_age_; }
   void set_max_prerender_age(base::TimeDelta td) { max_prerender_age_ = td; }
   unsigned int max_elements() const { return max_elements_; }
   void set_max_elements(unsigned int num) { max_elements_ = num; }
 
  protected:
   virtual ~PrerenderManager();
 
+  void SetPrerenderContentsFactory(
+      PrerenderContents::Factory* prerender_contents_factory);
+
  private:
+  // Test that needs needs access to internal functions.
+  friend class PrerenderBrowserTest;
+
   friend class base::RefCounted<PrerenderManager>;
   struct PrerenderContentsData;
 
   bool IsPrerenderElementFresh(const base::Time start) const;
   void DeleteOldEntries();
   virtual base::Time GetCurrentTime() const;
   virtual PrerenderContents* CreatePrerenderContents(
       const GURL& url,
       const std::vector<GURL>& alias_urls);
 
+  // Finds the specified PrerenderContents and returns it, if it exists.
+  // Returns NULL otherwise.  Unlike GetEntry, the PrerenderManager maintains
+  // ownership of the PrerenderContents.
+  PrerenderContents* FindEntry(const GURL& url);
+
   Profile* profile_;
 
   base::TimeDelta max_prerender_age_;
   unsigned int max_elements_;
 
   // List of prerendered elements.
   std::list<PrerenderContentsData> prerender_list_;
 
   // Default maximum permitted elements to prerender.
   static const unsigned int kDefaultMaxPrerenderElements = 1;
 
   // Default maximum age a prerendered element may have, in seconds.
   static const int kDefaultMaxPrerenderAgeSeconds = 20;
 
+  scoped_ptr<PrerenderContents::Factory> prerender_contents_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(PrerenderManager);
 };
 
 #endif  // CHROME_BROWSER_PRERENDER_PRERENDER_MANAGER_H_