NoState Prefetch: nostate prefetch browser tests.

chrome/browser/prerender/prerender_manager.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_BROWSER_PRERENDER_PRERENDER_MANAGER_H_
 #define CHROME_BROWSER_PRERENDER_PRERENDER_MANAGER_H_
 
 #include <stdint.h>
 
 #include <memory>
@@ skipping 81 matching lines @@
   };
 
   // One or more of these flags must be passed to ClearData() to specify just
   // what data to clear.  See function declaration for more information.
   enum ClearFlags {
     CLEAR_PRERENDER_CONTENTS = 0x1 << 0,
     CLEAR_PRERENDER_HISTORY = 0x1 << 1,
     CLEAR_MAX = 0x1 << 2
   };
 
+  // Used to manipulate time for testing.
+  class TimeOverride {
+   public:
+    virtual base::Time GetCurrentTime() const = 0;
+    virtual base::TimeTicks GetCurrentTimeTicks() const = 0;
+  };
+
   // Owned by a Profile object for the lifetime of the profile.
   explicit PrerenderManager(Profile* profile);
   ~PrerenderManager() override;
 
   // From KeyedService:
   void Shutdown() override;
 
   // Entry points for adding prerenders.
 
   // Adds a prerender for |url| if valid. |process_id| and |route_id| identify
@@ skipping 193 matching lines @@
 
   const Config& config() const { return config_; }
   Config& mutable_config() { return config_; }
 
   // Records that some visible tab navigated (or was redirected) to the
   // provided URL.
   void RecordNavigation(const GURL& url);
 
   Profile* profile() const { return profile_; }
 
-  // Classes which will be tested in prerender unit browser tests should use
-  // these methods to get times for comparison, so that the test framework can
-  // mock advancing/retarding time.
-  virtual base::Time GetCurrentTime() const;
-  virtual base::TimeTicks GetCurrentTimeTicks() const;
+  base::Time GetCurrentTime() const;
+  base::TimeTicks GetCurrentTimeTicks() const;
+
+  // For testing.
+  // TODO(mattcary): unify time testing by using base::SimpleTestClock and
+  // SimpleTestTickClock.
+  void SetTimeOverride(std::unique_ptr<TimeOverride> override);
 
   // Notification that a prerender has completed and its bytes should be
   // recorded.
   void RecordNetworkBytes(Origin origin, bool used, int64_t prerender_bytes);
 
   // Add to the running tally of bytes transferred over the network for this
   // profile if prerendering is currently enabled.
   void AddProfileNetworkBytesIfEnabled(int64_t bytes);
 
   // Registers a new ProcessHost performing a prerender. Called by
@@ skipping 259 matching lines @@
   // PrerenderManager is attached to.
   int64_t profile_network_bytes_;
 
   // The value of profile_network_bytes_ that was last recorded.
   int64_t last_recorded_profile_network_bytes_;
 
   // Set of process hosts being prerendered.
   using PrerenderProcessSet = std::set<content::RenderProcessHost*>;
   PrerenderProcessSet prerender_process_hosts_;
 
+  std::unique_ptr<TimeOverride> time_override_;
+
   base::WeakPtrFactory<PrerenderManager> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(PrerenderManager);
 };
 
 }  // namespace prerender
 
 #endif  // CHROME_BROWSER_PRERENDER_PRERENDER_MANAGER_H_