Domain Reliability: Switch to BackoffEntry in Scheduler

components/domain_reliability/util.h

 // Copyright 2014 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 COMPONENTS_DOMAIN_RELIABILITY_UTIL_H_
 #define COMPONENTS_DOMAIN_RELIABILITY_UTIL_H_
 
 #include <map>
 
 #include "base/callback_forward.h"
 #include "base/compiler_specific.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/time/time.h"
 #include "base/tracked_objects.h"
 #include "components/domain_reliability/domain_reliability_export.h"
+#include "net/base/backoff_entry.h"
 #include "net/http/http_response_info.h"
 
 namespace domain_reliability {
 
 // Attempts to convert a net error and an HTTP response code into the status
 // string that should be recorded in a beacon. Returns true if it could.
 //
 // N.B.: This functions as the whitelist of "safe" errors to report; network-
 //       local errors are purposefully not converted to avoid revealing
 //       information about the local network to the remote server.
@@ skipping 49 matching lines @@
   ActualTime();
 
   ~ActualTime() override;
 
   // MockableTime implementation:
   base::Time Now() override;
   base::TimeTicks NowTicks() override;
   scoped_ptr<MockableTime::Timer> CreateTimer() override;
 };
 
+// A subclass of BackoffEntry that uses a MockableTime to keep track of time.
+class MockableTimeBackoffEntry : public net::BackoffEntry {
+ public:
+  MockableTimeBackoffEntry(const net::BackoffEntry::Policy* const policy,
+                           MockableTime* time);
+
+  virtual ~MockableTimeBackoffEntry();
+
+ protected:
+  virtual base::TimeTicks ImplGetTimeNow() const override;
+
+ private:
+  MockableTime* time_;
+};
+
 }  // namespace domain_reliability
 
 #endif  // COMPONENTS_DOMAIN_RELIABILITY_UTIL_H_