Add TestMockTimeTaskRunner in base/test.

chrome/browser/chromeos/system/automatic_reboot_manager_unittest.cc

 // Copyright (c) 2013 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.
 
 #include "chrome/browser/chromeos/system/automatic_reboot_manager.h"
 
 #include <string>
-#include <utility>
 
 #include "ash/shell.h"
 #include "ash/test/test_shell_delegate.h"
 #include "base/compiler_specific.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/memory/ref_counted.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/prefs/pref_registry_simple.h"
 #include "base/prefs/testing_pref_service.h"
 #include "base/run_loop.h"
-#include "base/single_thread_task_runner.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/test/simple_test_tick_clock.h"
+#include "base/test/test_mock_time_task_runner.h"
 #include "base/thread_task_runner_handle.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/time/tick_clock.h"
 #include "base/values.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/login/users/mock_user_manager.h"
 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
 #include "chrome/browser/chromeos/system/automatic_reboot_manager_observer.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_browser_process.h"
@@ skipping 13 matching lines @@
 using ::testing::_;
 using ::testing::Invoke;
 using ::testing::Mock;
 using ::testing::ReturnPointee;
 
 namespace chromeos {
 namespace system {
 
 namespace {
 
-// A SingleThreadTaskRunner that mocks the current time and allows it to be
-// fast-forwarded. The current time in ticks is returned by Now(). The
-// corresponding device uptime is written to |uptime_file_|, providing a mock
-// for /proc/uptime.
-class MockTimeSingleThreadTaskRunner : public base::SingleThreadTaskRunner {
+// Provides a mock device uptime that follows the given |tick_clock_| with a
+// configurable offset. The mock uptime can also be written to |uptime_file_|,
+// thusly allowing to mock /proc/uptime.
+class MockUptimeProvider {
  public:
-  MockTimeSingleThreadTaskRunner();
+  explicit MockUptimeProvider(scoped_ptr<base::TickClock> tick_clock);
 
-  // base::SingleThreadTaskRunner:
-  virtual bool RunsTasksOnCurrentThread() const override;
-  virtual bool PostDelayedTask(const tracked_objects::Location& from_here,
-                               const base::Closure& task,
-                               base::TimeDelta delay) override;
-  virtual bool PostNonNestableDelayedTask(
-      const tracked_objects::Location& from_here,
-      const base::Closure& task,
-      base::TimeDelta delay) override;
+  void WriteUptimeToFile();
 
-  void SetUptimeFile(const base::FilePath& uptime_file);
+  // Adjusts the offset so that the current mock uptime will be |uptime|.
   void SetUptime(const base::TimeDelta& uptime);
 
-  const base::TimeDelta& Uptime() const;
-  const base::TimeTicks& Now() const;
+  void set_uptime_file_path(const base::FilePath& uptime_file_path) {
+    uptime_file_path_ = uptime_file_path;
+  }
 
-  void FastForwardBy(const base::TimeDelta& delta);
-  void FastForwardUntilNoTasksRemain();
-  void RunUntilIdle();
+  base::TimeDelta uptime() const {
+    return tick_clock_->NowTicks() - base::TimeTicks() + uptime_offset_;
+  }
 
  private:
-  // Strict weak temporal ordering of tasks.
-  class TemporalOrder {
-   public:
-    bool operator()(
-        const std::pair<base::TimeTicks, base::Closure>& first_task,
-        const std::pair<base::TimeTicks, base::Closure>& second_task) const;
-  };
+  scoped_ptr<base::TickClock> tick_clock_;
 
-  virtual ~MockTimeSingleThreadTaskRunner();
+  base::FilePath uptime_file_path_;
+  base::TimeDelta uptime_offset_;
 
-  base::FilePath uptime_file_;
-  base::TimeDelta uptime_;
-  base::TimeTicks now_;
-  std::priority_queue<std::pair<base::TimeTicks, base::Closure>,
-                      std::vector<std::pair<base::TimeTicks, base::Closure> >,
-                      TemporalOrder> tasks_;
-
-  DISALLOW_COPY_AND_ASSIGN(MockTimeSingleThreadTaskRunner);
+  DISALLOW_COPY_AND_ASSIGN(MockUptimeProvider);
 };
 
-class MockTimeTickClock : public base::TickClock {
+class TestAutomaticRebootManagerTaskRunner
+    : public base::TestMockTimeTaskRunner {
  public:
-  explicit MockTimeTickClock(
-      scoped_refptr<MockTimeSingleThreadTaskRunner> task_runner);
-  virtual ~MockTimeTickClock();
+  TestAutomaticRebootManagerTaskRunner();
 
-  // base::TickClock:
-  virtual base::TimeTicks NowTicks() override;
+  MockUptimeProvider* uptime_provider() const {
+    return uptime_provider_.get();
+  }
 
  private:
-  scoped_refptr<MockTimeSingleThreadTaskRunner> task_runner_;
+  ~TestAutomaticRebootManagerTaskRunner() override;
 
-  DISALLOW_COPY_AND_ASSIGN(MockTimeTickClock);
+  // base::TestMockTimeTaskRunner:
+  void OnBeforeSelectingTask() override;
+  void OnAfterTimePassed() override;
+  void OnAfterTaskRun() override;
+
+  scoped_ptr<MockUptimeProvider> uptime_provider_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestAutomaticRebootManagerTaskRunner);
 };
 
 class MockAutomaticRebootManagerObserver
     : public AutomaticRebootManagerObserver {
  public:
   MockAutomaticRebootManagerObserver();
   ~MockAutomaticRebootManagerObserver() override;
 
   void Init(AutomaticRebootManager* automatic_reboot_manger);
 
@@ skipping 40 matching lines @@
   bool ReadUpdateRebootNeededUptimeFromFile(base::TimeDelta* uptime);
   void VerifyRebootRequested(AutomaticRebootManagerObserver::Reason reason);
   void VerifyNoRebootRequested() const;
   void VerifyLoginScreenIdleTimerIsStopped() const;
   void VerifyNoGracePeriod() const;
   void VerifyGracePeriod(const base::TimeDelta& start_uptime) const;
 
   // Sets the status of |update_engine_client_| to NEED_REBOOT for tests.
   void SetUpdateStatusNeedReboot();
 
+  MockUptimeProvider* uptime_provider() const {
+    return task_runner_->uptime_provider();
+  }
+
   bool is_user_logged_in_;
   bool is_logged_in_as_kiosk_app_;
 
   // The uptime is read in the blocking thread pool and then processed on the
   // UI thread. This causes the UI thread to start processing the uptime when it
   // has increased by a small offset already. The offset is calculated and
   // stored in |uptime_processing_delay_| so that tests can accurately determine
   // the uptime seen by the UI thread.
   base::TimeDelta uptime_processing_delay_;
   base::TimeDelta update_reboot_needed_uptime_;
   base::TimeDelta uptime_limit_;
 
-  scoped_refptr<MockTimeSingleThreadTaskRunner> task_runner_;
+  scoped_refptr<TestAutomaticRebootManagerTaskRunner> task_runner_;
 
   MockAutomaticRebootManagerObserver automatic_reboot_manager_observer_;
   scoped_ptr<AutomaticRebootManager> automatic_reboot_manager_;
 
  private:
   void VerifyTimerIsStopped(const Timer* timer) const;
   void VerifyTimerIsRunning(const Timer* timer,
                             const base::TimeDelta& delay) const;
   void VerifyLoginScreenIdleTimerIsRunning() const;
 
@@ skipping 34 matching lines @@
                       const base::TimeDelta& uptime) {
   if (path.empty() || uptime == base::TimeDelta())
     return;
 
   const std::string uptime_seconds = base::DoubleToString(uptime.InSecondsF());
   ASSERT_EQ(static_cast<int>(uptime_seconds.size()),
             base::WriteFile(path, uptime_seconds.c_str(),
                             uptime_seconds.size()));
 }
 
-MockTimeSingleThreadTaskRunner::MockTimeSingleThreadTaskRunner() {
+MockUptimeProvider::MockUptimeProvider(scoped_ptr<base::TickClock> tick_clock)
+    : tick_clock_(tick_clock.Pass()) {
 }
 
-bool MockTimeSingleThreadTaskRunner::RunsTasksOnCurrentThread() const {
-  return true;
+void MockUptimeProvider::WriteUptimeToFile() {
+  SaveUptimeToFile(uptime_file_path_, uptime());
 }
 
-bool MockTimeSingleThreadTaskRunner::PostDelayedTask(
-    const tracked_objects::Location& from_here,
-    const base::Closure& task,
-    base::TimeDelta delay) {
-  tasks_.push(std::pair<base::TimeTicks, base::Closure>(now_ + delay, task));
-  return true;
+void MockUptimeProvider::SetUptime(const base::TimeDelta& uptime) {
+  uptime_offset_ = uptime - (tick_clock_->NowTicks() - base::TimeTicks());
+  WriteUptimeToFile();
 }
 
-bool MockTimeSingleThreadTaskRunner::PostNonNestableDelayedTask(
-    const tracked_objects::Location& from_here,
-    const base::Closure& task,
-    base::TimeDelta delay) {
-  NOTREACHED();
-  return false;
+TestAutomaticRebootManagerTaskRunner::TestAutomaticRebootManagerTaskRunner()
+    : uptime_provider_(new MockUptimeProvider(GetMockTickClock())) {
 }
 
-void MockTimeSingleThreadTaskRunner::SetUptimeFile(
-    const base::FilePath& uptime_file) {
-  uptime_file_ = uptime_file;
-  SaveUptimeToFile(uptime_file_, uptime_);
+TestAutomaticRebootManagerTaskRunner::~TestAutomaticRebootManagerTaskRunner() {
 }
 
-void MockTimeSingleThreadTaskRunner::SetUptime(const base::TimeDelta& uptime) {
-  uptime_ = uptime;
-  SaveUptimeToFile(uptime_file_, uptime_);
-}
-
-const base::TimeDelta& MockTimeSingleThreadTaskRunner::Uptime() const {
-  return uptime_;
-}
-
-const base::TimeTicks& MockTimeSingleThreadTaskRunner::Now() const {
-  return now_;
-}
-
-void MockTimeSingleThreadTaskRunner::FastForwardBy(
-    const base::TimeDelta& delta) {
-  const base::TimeTicks latest = now_ + delta;
+void TestAutomaticRebootManagerTaskRunner::OnBeforeSelectingTask() {
   base::SequencedWorkerPool* blocking_pool =
       content::BrowserThread::GetBlockingPool();
   blocking_pool->FlushForTesting();
-  while (!tasks_.empty() && tasks_.top().first <= latest) {
-    uptime_ += tasks_.top().first - now_;
-    SaveUptimeToFile(uptime_file_, uptime_);
-    now_ = tasks_.top().first;
-    base::Closure task = tasks_.top().second;
-    tasks_.pop();
-    task.Run();
-    blocking_pool->FlushForTesting();
-  }
-  uptime_ += latest - now_;
-  SaveUptimeToFile(uptime_file_, uptime_);
-  now_ = latest;
 }
 
-void MockTimeSingleThreadTaskRunner::FastForwardUntilNoTasksRemain() {
+void TestAutomaticRebootManagerTaskRunner::OnAfterTimePassed() {
+  uptime_provider_->WriteUptimeToFile();
+}
+
+void TestAutomaticRebootManagerTaskRunner::OnAfterTaskRun() {
   base::SequencedWorkerPool* blocking_pool =
       content::BrowserThread::GetBlockingPool();
   blocking_pool->FlushForTesting();
-  while (!tasks_.empty()) {
-    uptime_ += tasks_.top().first - now_;
-    SaveUptimeToFile(uptime_file_, uptime_);
-    now_ = tasks_.top().first;
-    base::Closure task = tasks_.top().second;
-    tasks_.pop();
-    task.Run();
-    blocking_pool->FlushForTesting();
-  }
-}
-
-void MockTimeSingleThreadTaskRunner::RunUntilIdle() {
-  base::SequencedWorkerPool* blocking_pool =
-      content::BrowserThread::GetBlockingPool();
-  blocking_pool->FlushForTesting();
-  while (!tasks_.empty() && tasks_.top().first <= now_) {
-    base::Closure task = tasks_.top().second;
-    tasks_.pop();
-    task.Run();
-    blocking_pool->FlushForTesting();
-  }
-}
-
-bool MockTimeSingleThreadTaskRunner::TemporalOrder::operator()(
-    const std::pair<base::TimeTicks, base::Closure>& first_task,
-    const std::pair<base::TimeTicks, base::Closure>& second_task) const {
-  return first_task.first > second_task.first;
-}
-
-MockTimeSingleThreadTaskRunner::~MockTimeSingleThreadTaskRunner() {
-}
-
-MockTimeTickClock::MockTimeTickClock(
-    scoped_refptr<MockTimeSingleThreadTaskRunner> task_runner)
-    : task_runner_(task_runner) {
-}
-
-MockTimeTickClock::~MockTimeTickClock() {
-}
-
-base::TimeTicks MockTimeTickClock::NowTicks() {
-  return task_runner_->Now();
 }
 
 MockAutomaticRebootManagerObserver::MockAutomaticRebootManagerObserver()
     : automatic_reboot_manger_(nullptr) {
   ON_CALL(*this, WillDestroyAutomaticRebootManager())
       .WillByDefault(
           Invoke(this,
                  &MockAutomaticRebootManagerObserver::StopObserving));
 }
 
@@ skipping 11 matching lines @@
 
 void MockAutomaticRebootManagerObserver::StopObserving() {
   ASSERT_TRUE(automatic_reboot_manger_);
   automatic_reboot_manger_->RemoveObserver(this);
   automatic_reboot_manger_ = nullptr;
 }
 
 AutomaticRebootManagerBasicTest::AutomaticRebootManagerBasicTest()
     : is_user_logged_in_(false),
       is_logged_in_as_kiosk_app_(false),
-      task_runner_(new MockTimeSingleThreadTaskRunner),
+      task_runner_(new TestAutomaticRebootManagerTaskRunner),
       reboot_after_update_(false),
       ui_thread_task_runner_handle_(task_runner_),
       mock_user_manager_(new MockUserManager),
       user_manager_enabler_(mock_user_manager_),
       power_manager_client_(NULL),
       update_engine_client_(NULL) {
 }
 
 AutomaticRebootManagerBasicTest::~AutomaticRebootManagerBasicTest() {
 }
 
 void AutomaticRebootManagerBasicTest::SetUp() {
   ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
   const base::FilePath& temp_dir = temp_dir_.path();
   const base::FilePath uptime_file = temp_dir.Append("uptime");
-  task_runner_->SetUptimeFile(uptime_file);
+  uptime_provider()->set_uptime_file_path(uptime_file);
   ASSERT_FALSE(base::WriteFile(uptime_file, NULL, 0));
   update_reboot_needed_uptime_file_ =
       temp_dir.Append("update_reboot_needed_uptime");
   ASSERT_FALSE(base::WriteFile(update_reboot_needed_uptime_file_, NULL, 0));
   ASSERT_TRUE(PathService::Override(chromeos::FILE_UPTIME, uptime_file));
   ASSERT_TRUE(PathService::Override(chromeos::FILE_UPDATE_REBOOT_NEEDED_UPTIME,
                                     update_reboot_needed_uptime_file_));
 
   TestingBrowserProcess::GetGlobal()->SetLocalState(&local_state_);
   AutomaticRebootManager::RegisterPrefs(local_state_.registry());
@@ skipping 118 matching lines @@
   Mock::VerifyAndClearExpectations(&automatic_reboot_manager_observer_);
   EXPECT_CALL(automatic_reboot_manager_observer_,
               WillDestroyAutomaticRebootManager()).Times(0);
   EXPECT_CALL(automatic_reboot_manager_observer_,
               OnRebootRequested(_)).Times(0);
 }
 
 void AutomaticRebootManagerBasicTest::CreateAutomaticRebootManager(
     bool expect_reboot) {
   automatic_reboot_manager_.reset(new AutomaticRebootManager(
-      scoped_ptr<base::TickClock>(new MockTimeTickClock(task_runner_))));
+      scoped_ptr<base::TickClock>(task_runner_->GetMockTickClock())));
   automatic_reboot_manager_observer_.Init(automatic_reboot_manager_.get());
   task_runner_->RunUntilIdle();
   EXPECT_EQ(expect_reboot ? 1 : 0,
             power_manager_client_->num_request_restart_calls());
 
   uptime_processing_delay_ =
       base::TimeTicks() - automatic_reboot_manager_->boot_time_ -
-      task_runner_->Uptime();
+      uptime_provider()->uptime();
   EXPECT_GE(uptime_processing_delay_, base::TimeDelta());
   EXPECT_LE(uptime_processing_delay_, base::TimeDelta::FromSeconds(1));
 
   if (is_user_logged_in_ || expect_reboot)
     VerifyLoginScreenIdleTimerIsStopped();
   else
     VerifyLoginScreenIdleTimerIsRunning();
 }
 
 bool AutomaticRebootManagerBasicTest::ReadUpdateRebootNeededUptimeFromFile(
@@ skipping 29 matching lines @@
 
 void AutomaticRebootManagerBasicTest::VerifyNoGracePeriod() const {
   EXPECT_FALSE(automatic_reboot_manager_->reboot_requested_);
   VerifyTimerIsStopped(automatic_reboot_manager_->grace_start_timer_.get());
   VerifyTimerIsStopped(automatic_reboot_manager_->grace_end_timer_.get());
 }
 
 void AutomaticRebootManagerBasicTest::VerifyGracePeriod(
     const base::TimeDelta& start_uptime) const {
   const base::TimeDelta start =
-      start_uptime - task_runner_->Uptime() - uptime_processing_delay_;
+      start_uptime - uptime_provider()->uptime() - uptime_processing_delay_;
   const base::TimeDelta end = start + base::TimeDelta::FromHours(24);
   if (start <= base::TimeDelta()) {
     EXPECT_TRUE(automatic_reboot_manager_->reboot_requested_);
     VerifyTimerIsStopped(automatic_reboot_manager_->grace_start_timer_.get());
     VerifyTimerIsRunning(automatic_reboot_manager_->grace_end_timer_.get(),
                          end);
   } else {
     EXPECT_FALSE(automatic_reboot_manager_->reboot_requested_);
     VerifyTimerIsRunning(automatic_reboot_manager_->grace_start_timer_.get(),
                          start);
@@ skipping 47 matching lines @@
   }
 }
 
 AutomaticRebootManagerTest::~AutomaticRebootManagerTest() {
 }
 
 // Chrome is showing the login screen. The current uptime is 12 hours.
 // Verifies that the idle timer is running. Further verifies that when a kiosk
 // app session begins, the idle timer is stopped.
 TEST_F(AutomaticRebootManagerBasicTest, LoginStopsIdleTimer) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested, the device does not reboot immediately
   // and the login screen idle timer is started.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Notify that a kiosk app session has been started.
   is_user_logged_in_ = true;
   is_logged_in_as_kiosk_app_ = true;
   automatic_reboot_manager_->Observe(
       chrome::NOTIFICATION_LOGIN_USER_CHANGED,
       content::Source<AutomaticRebootManagerBasicTest>(this),
       content::NotificationService::NoDetails());
 
   // Verify that the login screen idle timer is stopped.
   VerifyLoginScreenIdleTimerIsStopped();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is showing the login screen. The current uptime is 12 hours.
 // Verifies that the idle timer is running. Further verifies that when a
 // non-kiosk-app session begins, the idle timer is stopped.
 TEST_F(AutomaticRebootManagerBasicTest, NonKioskLoginStopsIdleTimer) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested, the device does not reboot immediately
   // and the login screen idle timer is started.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Notify that a non-kiosk-app session has been started.
   is_user_logged_in_ = true;
   automatic_reboot_manager_->Observe(
       chrome::NOTIFICATION_LOGIN_USER_CHANGED,
       content::Source<AutomaticRebootManagerBasicTest>(this),
       content::NotificationService::NoDetails());
 
   // Verify that the login screen idle timer is stopped.
   VerifyLoginScreenIdleTimerIsStopped();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is showing the login screen. The uptime limit is 6 hours. The current
 // uptime is 12 hours.
 // Verifies that user activity prevents the device from rebooting. Further
 // verifies that when user activity ceases, the devices reboots.
 TEST_F(AutomaticRebootManagerBasicTest, UserActivityResetsIdleTimer) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested, the device does not reboot immediately
   // and the login screen idle timer is started.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
@@ skipping 17 matching lines @@
   // Verify that the device reboots immediately.
   FastForwardBy(base::TimeDelta::FromSeconds(60), true);
 }
 
 // Chrome is running a kiosk app session. The current uptime is 10 days.
 // Verifies that when the device is suspended and then resumes, it does not
 // immediately reboot.
 TEST_F(AutomaticRebootManagerBasicTest, ResumeNoPolicy) {
   is_user_logged_in_ = true;
   is_logged_in_as_kiosk_app_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that the device has resumed from 1 hour of sleep. Verify that no
   // reboot is requested and the device does not reboot immediately.
   NotifyResumed(false);
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running a non-kiosk-app session. The current uptime is 10 days.
 // Verifies that when the device is suspended and then resumes, it does not
 // immediately reboot.
 TEST_F(AutomaticRebootManagerBasicTest, NonKioskResumeAppNoPolicy) {
   is_user_logged_in_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that the device has resumed from 1 hour of sleep. Verify that no
   // reboot is requested and the device does not reboot immediately.
   NotifyResumed(false);
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running a kiosk app session. The uptime limit is 24 hours. The
 // current uptime is 12 hours.
 // Verifies that when the device is suspended and then resumes, it does not
 // immediately reboot.
 TEST_F(AutomaticRebootManagerBasicTest, ResumeBeforeGracePeriod) {
   is_user_logged_in_ = true;
   is_logged_in_as_kiosk_app_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
 
   // Verify that a grace period has been scheduled to start in the future.
   VerifyGracePeriod(uptime_limit_);
 
   // Notify that the device has resumed from 1 hour of sleep. Verify that no
   // reboot is requested and the device does not reboot immediately.
   NotifyResumed(false);
 
   // Verify a reboot is requested and the device reboots eventually.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   FastForwardUntilNoTasksRemain(true);
 }
 
 // Chrome is running a non-kiosk-app session. The uptime limit is 24 hours. The
 // current uptime is 12 hours.
 // Verifies that when the device is suspended and then resumes, it does not
 // immediately reboot.
 TEST_F(AutomaticRebootManagerBasicTest, NonKioskResumeBeforeGracePeriod) {
   is_user_logged_in_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
@@ skipping 10 matching lines @@
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running a kiosk app session. The uptime limit is 6 hours. The
 // current uptime is 12 hours.
 // Verifies that when the device is suspended and then resumes, it immediately
 // reboots.
 TEST_F(AutomaticRebootManagerBasicTest, ResumeInGracePeriod) {
   is_user_logged_in_ = true;
   is_logged_in_as_kiosk_app_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Notify that the device has resumed from 1 hour of sleep. Verify that the
   // device reboots immediately.
   NotifyResumed(true);
 }
 
 // Chrome is running a non-kiosk-app session. The uptime limit is 6 hours. The
 // current uptime is 12 hours.
 // Verifies that when the device is suspended and then resumes, it does not
 // immediately reboot.
 TEST_F(AutomaticRebootManagerBasicTest, NonKioskResumeInGracePeriod) {
   is_user_logged_in_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
@@ skipping 10 matching lines @@
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running a kiosk app session. The uptime limit is 6 hours. The
 // current uptime is 29 hours 30 minutes.
 // Verifies that when the device is suspended and then resumes, it immediately
 // reboots.
 TEST_F(AutomaticRebootManagerBasicTest, ResumeAfterGracePeriod) {
   is_user_logged_in_ = true;
   is_logged_in_as_kiosk_app_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromHours(29) +
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(29) +
                           base::TimeDelta::FromMinutes(30));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Notify that the device has resumed from 1 hour of sleep. Verify that the
   // device reboots immediately.
   NotifyResumed(true);
 }
 
 // Chrome is running a non-kiosk-app session. The uptime limit is 6 hours. The
 // current uptime is 29 hours 30 minutes.
 // Verifies that when the device is suspended and then resumes, it does not
 // immediately reboot.
 TEST_F(AutomaticRebootManagerBasicTest, NonKioskResumeAfterGracePeriod) {
   is_user_logged_in_ = true;
-  task_runner_->SetUptime(base::TimeDelta::FromHours(29) +
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(29) +
                           base::TimeDelta::FromMinutes(30));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Notify that the device has resumed from 1 hour of sleep. Verify that the
   // device does not reboot immediately.
   NotifyResumed(false);
 
   // Verify that the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The current uptime is 10 days.
 // Verifies that when the browser terminates, the device does not immediately
 // reboot.
 TEST_P(AutomaticRebootManagerTest, TerminateNoPolicy) {
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that the browser is terminating. Verify that no reboot is requested
   // and the device does not reboot immediately.
   NotifyTerminating(false);
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The uptime limit is set to 24 hours. The current uptime is
 // 12 hours.
 // Verifies that when the browser terminates, it does not immediately reboot.
 TEST_P(AutomaticRebootManagerTest, TerminateBeforeGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
@@ skipping 10 matching lines @@
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 6 hours. The current uptime is
 // 12 hours.
 // Verifies that when the browser terminates, the device immediately reboots if
 // a kiosk app session is in progress.
 TEST_P(AutomaticRebootManagerTest, TerminateInGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Notify that the browser is terminating. Verify that the device immediately
   // reboots if a kiosk app session is in progress.
   NotifyTerminating(is_logged_in_as_kiosk_app_);
 
   // Verify that if a non-kiosk-app session is in progress, the device never
   // reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The current uptime is 12 hours.
 // Verifies that when the uptime limit is set to 24 hours, no reboot occurs and
 // a grace period is scheduled to begin after 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, BeforeUptimeLimitGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
 
   // Verify that a grace period has been scheduled to start in the future.
   VerifyGracePeriod(uptime_limit_);
 
   // Verify that a reboot is requested eventually and unless a non-kiosk-app
   // session is in progress, the device eventually reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The current uptime is 12 hours.
 // Verifies that when the uptime limit is set to 6 hours, a reboot is requested
 // and a grace period is started that will end after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, InUptimeLimitGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The current uptime is 10 days.
 // Verifies that when the uptime limit is set to 6 hours, the device reboots
 // immediately if no non-kiosk-app-session is in progress because the grace
 // period ended after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, AfterUptimeLimitGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Set the uptime limit. Verify that a reboot is requested and unless a
   // non-kiosk-app session is in progress, the the device immediately reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), !is_user_logged_in_ ||
                                                 is_logged_in_as_kiosk_app_);
 
   // Verify that if a non-kiosk-app session is in progress, the device never
   // reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 12 hours. The current uptime is
 // 6 hours.
 // Verifies that when the uptime limit is removed, the grace period is removed.
 TEST_P(AutomaticRebootManagerTest, UptimeLimitOffBeforeGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(6));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(6));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(12), false);
@@ skipping 13 matching lines @@
   VerifyNoGracePeriod();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The uptime limit is set to 12 hours. The current uptime is
 // 24 hours.
 // Verifies that when the uptime limit is removed, the grace period is removed.
 TEST_P(AutomaticRebootManagerTest, UptimeLimitOffInGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(24));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(24));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
@@ skipping 15 matching lines @@
 
   // Verify that the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The uptime limit is set to 12 hours. The current uptime is
 // 6 hours.
 // Verifies that when the uptime limit is extended to 24 hours, the grace period
 // is rescheduled to start further in the future.
 TEST_P(AutomaticRebootManagerTest, ExtendUptimeLimitBeforeGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(6));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(6));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(12), false);
@@ skipping 18 matching lines @@
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 12 hours. The current uptime is
 // 18 hours.
 // Verifies that when the uptime limit is extended to 24 hours, the grace period
 // is rescheduled to start in the future.
 TEST_P(AutomaticRebootManagerTest, ExtendUptimeLimitInGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(18));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(18));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
@@ skipping 18 matching lines @@
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 18 hours. The current uptime is
 // 12 hours.
 // Verifies that when the uptime limit is shortened to 6 hours, the grace period
 // is rescheduled to have already started.
 TEST_P(AutomaticRebootManagerTest, ShortenUptimeLimitBeforeToInGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(18), false);
@@ skipping 17 matching lines @@
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 24 hours. The current uptime is
 // 36 hours.
 // Verifies that when the uptime limit is shortened to 18 hours, the grace
 // period is rescheduled to have started earlier.
 TEST_P(AutomaticRebootManagerTest, ShortenUptimeLimitInToInGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(36));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(36));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
@@ skipping 19 matching lines @@
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 24 hours. The current uptime is
 // 36 hours.
 // Verifies that when the uptime limit is shortened to 6 hours, the device
 // reboots immediately if no non-kiosk-app session is in progress because the
 // grace period ended after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, ShortenUptimeLimitInToAfterGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(36));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(36));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
@@ skipping 18 matching lines @@
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The current uptime is 12 hours.
 // Verifies that when an update is applied, the current uptime is persisted as
 // the time at which a reboot became necessary. Further verifies that when the
 // policy to automatically reboot after an update is not enabled, no reboot
 // occurs and no grace period is scheduled.
 TEST_P(AutomaticRebootManagerTest, UpdateNoPolicy) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that no reboot is requested and the device does not reboot immediately.
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The current uptime is 12 hours.
 // Verifies that when an update is applied, the current uptime is persisted as
 // the time at which a reboot became necessary. Further verifies that when the
 // policy to automatically reboot after an update is enabled, a reboot is
 // requested and a grace period is started that will end 24 hours from now.
 TEST_P(AutomaticRebootManagerTest, Update) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested but the device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(update_reboot_needed_uptime_ + uptime_processing_delay_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The current uptime is 12 hours.
 // Verifies that when Chrome is notified twice that an update has been applied,
 // the second notification is ignored and the uptime at which it occurred does
 // not get persisted as the time at which an update became necessary.
 TEST_P(AutomaticRebootManagerTest, UpdateAfterUpdate) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested but the device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(update_reboot_needed_uptime_ + uptime_processing_delay_);
 
   // Fast forward the uptime by 20 seconds. Verify that the device does not
   // reboot immediately.
   FastForwardBy(base::TimeDelta::FromSeconds(20), false);
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that the device does not reboot immediately.
@@ skipping 12 matching lines @@
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The current uptime is 10 minutes.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, no reboot occurs and a grace period is scheduled to begin after the
 // minimum of 1 hour of uptime. Further verifies that when an update is applied,
 // the current uptime is persisted as the time at which a reboot became
 // necessary.
 TEST_P(AutomaticRebootManagerTest, UpdateBeforeMinimumUptime) {
-  task_runner_->SetUptime(base::TimeDelta::FromMinutes(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromMinutes(10));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that no reboot is requested and the device does not reboot immediately.
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that a grace period has been scheduled to begin in the future.
   VerifyGracePeriod(base::TimeDelta::FromHours(1));
 
   // Verify that a reboot is requested eventually and unless a non-kiosk-app
   // session is in progress, the device eventually reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. An update was applied and a reboot became necessary to
 // complete the update process after 6 hours of uptime. The current uptime is
 // 12 hours.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, a reboot is requested and a grace period is started that will end
 // after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, PolicyAfterUpdateInGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(6));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(6));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that no reboot is requested and the device does not reboot immediately.
   NotifyUpdateRebootNeeded();
@@ skipping 19 matching lines @@
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. An update was applied and a reboot became necessary to
 // complete the update process after 6 hours of uptime. The current uptime is
 // 10 days.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, the device reboots immediately if no non-kiosk-app session is in
 // progress because the grace period ended after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, PolicyAfterUpdateAfterGracePeriod) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(6));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(6));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that no reboot is requested and the device does not reboot immediately.
   NotifyUpdateRebootNeeded();
@@ skipping 18 matching lines @@
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. An update was applied and a reboot became necessary to
 // complete the update process after 6 hours of uptime. The policy to
 // automatically reboot after an update is enabled. The current uptime is
 // 6 hours 20 seconds.
 // Verifies that when the policy to automatically reboot after an update is
 // disabled, the reboot request and grace period are removed.
 TEST_P(AutomaticRebootManagerTest, PolicyOffAfterUpdate) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(6));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(6));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested but the device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   NotifyUpdateRebootNeeded();
 
   // Verify that a grace period has started.
-  VerifyGracePeriod(task_runner_->Uptime() + uptime_processing_delay_);
+  VerifyGracePeriod(uptime_provider()->uptime() + uptime_processing_delay_);
 
   // Fast forward the uptime by 20 seconds. Verify that the device does not
   // reboot immediately.
   FastForwardBy(base::TimeDelta::FromSeconds(20), false);
 
   // Disable automatic rebooting after an update has been applied. Verify that
   // no reboot is requested and the device does not reboot immediately.
   SetRebootAfterUpdate(false, false);
 
   // Verify that the grace period has been removed.
@@ skipping 45 matching lines @@
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The policy to automatically reboot after an update is
 // enabled. The current uptime is 12 hours.
 // Verifies that when an uptime limit of 6 hours is set, the availability of an
 // update does not cause the grace period to be rescheduled. Further verifies
 // that the current uptime is persisted as the time at which a reboot became
 // necessary.
 TEST_P(AutomaticRebootManagerTest, UptimeLimitBeforeUpdate) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
   // not reboot immediately.
@@ skipping 10 matching lines @@
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested again but the device does not reboot
   // immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that the grace period has not been rescheduled.
   VerifyGracePeriod(uptime_limit_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The policy to automatically reboot after an update is
 // enabled. The current uptime is 12 hours.
 // Verifies that when an uptime limit of 24 hours is set, the availability of an
 // update causes the grace period to be rescheduled so that it ends 24 hours
 // from now. Further verifies that the current uptime is persisted as the time
 // at which a reboot became necessary.
 TEST_P(AutomaticRebootManagerTest, UpdateBeforeUptimeLimit) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
 
   // Verify that a grace period has been scheduled to start in the future.
   VerifyGracePeriod(uptime_limit_);
 
   // Fast forward the uptime by 20 seconds. Verify that no reboot is requested
   // and the device does not reboot immediately.
   FastForwardBy(base::TimeDelta::FromSeconds(20), false);
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested but the device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that the grace period has been rescheduled to start at the time that
   // the update became available.
   VerifyGracePeriod(update_reboot_needed_uptime_ + uptime_processing_delay_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is running. The uptime limit is set to 24 hours. An update was applied
 // and a reboot became necessary to complete the update process after 12 hours.
 // The policy to automatically reboot after an update is enabled. The current
 // uptime is 12 hours 20 seconds.
 // Verifies that when the policy to reboot after an update is disabled, the
 // grace period is rescheduled to start after 12 hours of uptime. Further
 // verifies that when the uptime limit is removed, the grace period is removed.
 TEST_P(AutomaticRebootManagerTest, PolicyOffThenUptimeLimitOff) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that no reboot is requested and the device
   // does not reboot immediately.
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
 
   // Verify that a grace period has been scheduled to start in the future.
   VerifyGracePeriod(uptime_limit_);
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested but the device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that a grace period has been rescheduled to end 24 hours from now.
   VerifyGracePeriod(update_reboot_needed_uptime_ + uptime_processing_delay_);
 
   // Fast forward the uptime by 20 seconds. Verify that the device does not
   // reboot immediately.
   FastForwardBy(base::TimeDelta::FromSeconds(20), false);
 
   // Disable automatic reboot after an update has been applied. Verify that the
   // device does not reboot immediately.
@@ skipping 16 matching lines @@
 
 // Chrome is running. The uptime limit is set to 6 hours. An update was applied
 // and a reboot became necessary to complete the update process after 12 hours.
 // The policy to automatically reboot after an update is enabled. The current
 // uptime is 12 hours 20 seconds.
 // Verifies that when the uptime limit is removed, the grace period is
 // rescheduled to have started after 12 hours of uptime. Further verifies that
 // when the policy to reboot after an update is disabled, the reboot request and
 // grace period are removed.
 TEST_P(AutomaticRebootManagerTest, UptimeLimitOffThenPolicyOff) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Notify that an update has been applied and a reboot is necessary. Verify
   // that a reboot is requested but the device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   NotifyUpdateRebootNeeded();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that the grace period has started.
   VerifyGracePeriod(update_reboot_needed_uptime_ + uptime_processing_delay_);
 
   // Set the uptime limit. Verify that a reboot is requested again but the
   // device does not reboot immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
 
   // Verify that the grace period has been rescheduled to have started after
@@ skipping 22 matching lines @@
 
   // Verify that the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is running. The uptime limit is 6 hours. The current uptime is
 // 29 hours 59 minutes 59 seconds.
 // Verifies that if no non-kiosk-app session is in progress, the device reboots
 // immediately when the grace period ends after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, GracePeriodEnd) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(29) +
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(29) +
                           base::TimeDelta::FromMinutes(59) +
                           base::TimeDelta::FromSeconds(59));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Set the uptime limit. Verify that a reboot is requested but the device does
@@ skipping 12 matching lines @@
   // Verify that if a non-kiosk-app session is in progress, the device never
   // reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. The current uptime is 10 days.
 // Verifies that when no automatic reboot policy is enabled, no reboot occurs
 // and no grace period is scheduled.
 TEST_P(AutomaticRebootManagerTest, StartNoPolicy) {
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is starting. The uptime limit is set to 24 hours. The current uptime
 // is 12 hours.
 // Verifies that no reboot occurs and a grace period is scheduled to begin after
 // 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartBeforeUptimeLimitGracePeriod) {
   SetUptimeLimit(base::TimeDelta::FromHours(24), false);
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that a grace period has been scheduled to start in the future.
   VerifyGracePeriod(uptime_limit_);
 
   // Verify that a reboot is requested eventually and unless a non-kiosk-app
   // session is in progress, the device eventually reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. The uptime limit is set to 6 hours. The current uptime is
 // 10 days.
 // Verifies that if no non-kiosk-app session is in progress, the device reboots
 // immediately because the grace period ended after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartAfterUptimeLimitGracePeriod) {
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that a reboot is requested and unless a non-kiosk-app session is in
   // progress, the the device immediately reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   CreateAutomaticRebootManager(!is_user_logged_in_ ||
                                is_logged_in_as_kiosk_app_);
   VerifyRebootRequested(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
 
   // Verify that if a non-kiosk-app session is in progress, the device never
   // reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. The uptime limit is set to 6 hours. The current uptime is
 // 12 hours.
 // Verifies that a reboot is requested and a grace period is started that will
 // end after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartInUptimeLimitGracePeriod) {
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
 
   // Verify that a reboot is requested but the device does not reboot
   // immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   CreateAutomaticRebootManager(false);
   VerifyRebootRequested(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. An update was applied and a reboot became necessary to
 // complete the update process after 6 hours of uptime. The current uptime is
 // 10 days.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, the device reboots immediately if no non-kiosk-app session is in
 // progress because the grace period ended after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartAfterUpdateGracePeriod) {
   SetUpdateStatusNeedReboot();
   SetUpdateRebootNeededUptime(base::TimeDelta::FromHours(6));
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
   SetRebootAfterUpdate(true, false);
 
   // Verify that a reboot is requested and unless a non-kiosk-app session is in
   // progress, the the device immediately reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   CreateAutomaticRebootManager(!is_user_logged_in_ ||
                                is_logged_in_as_kiosk_app_);
   VerifyRebootRequested(
       AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
 
   // Verify that if a non-kiosk-app session is in progress, the device never
   // reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. An update was applied and a reboot became necessary to
 // complete the update process after 6 hours of uptime. The current uptime is
 // 12 hours.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, a reboot is requested and a grace period is started that will end
 // after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartInUpdateGracePeriod) {
   SetUpdateStatusNeedReboot();
   SetUpdateRebootNeededUptime(base::TimeDelta::FromHours(6));
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that a reboot is requested but the device does not reboot
   // immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   CreateAutomaticRebootManager(false);
   VerifyRebootRequested(
       AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(update_reboot_needed_uptime_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. An update was applied and a reboot became necessary to
 // complete the update process after 10 minutes of uptime. The current uptime is
 // 20 minutes.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, no reboot occurs and a grace period is scheduled to begin after the
 // minimum of 1 hour of uptime.
 TEST_P(AutomaticRebootManagerTest, StartBeforeUpdateGracePeriod) {
   SetUpdateStatusNeedReboot();
   SetUpdateRebootNeededUptime(base::TimeDelta::FromMinutes(10));
-  task_runner_->SetUptime(base::TimeDelta::FromMinutes(20));
+  uptime_provider()->SetUptime(base::TimeDelta::FromMinutes(20));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that a grace period has been scheduled to start in the future.
   VerifyGracePeriod(base::TimeDelta::FromHours(1));
 
   // Verify that a reboot is requested eventually and unless a non-kiosk-app
   // session is in progress, the device eventually reboots.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. An update was applied and a reboot became necessary to
 // complete the update process after 6 hours of uptime. The current uptime is
 // 10 days.
 // Verifies that when the policy to automatically reboot after an update is not
 // enabled, no reboot occurs and no grace period is scheduled.
 TEST_P(AutomaticRebootManagerTest, StartUpdateNoPolicy) {
   SetUpdateStatusNeedReboot();
   SetUpdateRebootNeededUptime(base::TimeDelta::FromHours(6));
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is starting. An update was applied and a reboot became necessary to
 // complete the update process but the time at which this happened was lost. The
 // current uptime is 10 days.
 // Verifies that the current uptime is persisted as the time at which a reboot
 // became necessary. Further verifies that when the policy to automatically
 // reboot after an update is enabled, a reboot is requested and a grace period
 // is started that will end 24 hours from now.
 TEST_P(AutomaticRebootManagerTest, StartUpdateTimeLost) {
   SetUpdateStatusNeedReboot();
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
   SetRebootAfterUpdate(true, false);
 
   // Verify that a reboot is requested but the device does not reboot
   // immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   CreateAutomaticRebootManager(false);
   VerifyRebootRequested(
       AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(update_reboot_needed_uptime_ + uptime_processing_delay_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. An update was applied and a reboot became necessary to
 // complete the update process but the time at which this happened was lost. The
 // current uptime is 10 days.
 // Verifies that the current uptime is persisted as the time at which a reboot
 // became necessary. Further verifies that when the policy to automatically
 // reboot after an update is not enabled, no reboot occurs and no grace period
 // is scheduled.
 TEST_P(AutomaticRebootManagerTest, StartUpdateNoPolicyTimeLost) {
   SetUpdateStatusNeedReboot();
-  task_runner_->SetUptime(base::TimeDelta::FromDays(10));
+  uptime_provider()->SetUptime(base::TimeDelta::FromDays(10));
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that the current uptime has been persisted as the time at which a
   // reboot became necessary.
   EXPECT_TRUE(ReadUpdateRebootNeededUptimeFromFile(
       &update_reboot_needed_uptime_));
-  EXPECT_EQ(task_runner_->Uptime(), update_reboot_needed_uptime_);
+  EXPECT_EQ(uptime_provider()->uptime(), update_reboot_needed_uptime_);
 
   // Verify that no grace period has started.
   VerifyNoGracePeriod();
 
   // Verify that a reboot is never requested and the device never reboots.
   FastForwardUntilNoTasksRemain(false);
 }
 
 // Chrome is starting. No update has been applied. The current uptime is
 // 12 hours.
 // Verifies that no time is persisted as the time at which a reboot became
 // necessary. Further verifies that no reboot occurs and no grace period is
 // scheduled.
 TEST_P(AutomaticRebootManagerTest, StartNoUpdate) {
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that no reboot is requested and the device does not reboot
   // immediately.
   ExpectNoRebootRequest();
   CreateAutomaticRebootManager(false);
   VerifyNoRebootRequested();
 
   // Verify that no time is persisted as the time at which a reboot became
   // necessary.
@@ skipping 10 matching lines @@
 // Chrome is starting. The uptime limit is set to 6 hours. Also, an update was
 // applied and a reboot became necessary to complete the update process after
 // 8 hours of uptime. The current uptime is 12 hours.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, a reboot is requested and a grace period is started that will end
 // after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartUptimeLimitBeforeUpdate) {
   SetUptimeLimit(base::TimeDelta::FromHours(6), false);
   SetUpdateStatusNeedReboot();
   SetUpdateRebootNeededUptime(base::TimeDelta::FromHours(8));
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that a reboot is requested but the device does not reboot
   // immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
   CreateAutomaticRebootManager(false);
   VerifyRebootRequested(AutomaticRebootManagerObserver::REBOOT_REASON_PERIODIC);
 
   // Verify that a grace period has started.
   VerifyGracePeriod(uptime_limit_);
 
   // Verify that unless a non-kiosk-app session is in progress, the device
   // eventually reboots.
   FastForwardUntilNoTasksRemain(!is_user_logged_in_ ||
                                 is_logged_in_as_kiosk_app_);
 }
 
 // Chrome is starting. The uptime limit is set to 8 hours. Also, an update was
 // applied and a reboot became necessary to complete the update process after
 // 6 hours of uptime. The current uptime is 12 hours.
 // Verifies that when the policy to automatically reboot after an update is
 // enabled, a reboot is requested and a grace period is started that will end
 // after 6 + 24 hours of uptime.
 TEST_P(AutomaticRebootManagerTest, StartUpdateBeforeUptimeLimit) {
   SetUptimeLimit(base::TimeDelta::FromHours(8), false);
   SetUpdateStatusNeedReboot();
   SetUpdateRebootNeededUptime(base::TimeDelta::FromHours(6));
-  task_runner_->SetUptime(base::TimeDelta::FromHours(12));
+  uptime_provider()->SetUptime(base::TimeDelta::FromHours(12));
   SetRebootAfterUpdate(true, false);
 
   // Verify that a reboot is requested but the device does not reboot
   // immediately.
   ExpectRebootRequest(AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
   CreateAutomaticRebootManager(false);
   VerifyRebootRequested(
       AutomaticRebootManagerObserver::REBOOT_REASON_OS_UPDATE);
 
   // Verify that a grace period has started.
@@ skipping 32 matching lines @@
 INSTANTIATE_TEST_CASE_P(
     AutomaticRebootManagerTestInstance,
     AutomaticRebootManagerTest,
     ::testing::Values(
         AUTOMATIC_REBOOT_MANAGER_TEST_SCENARIO_LOGIN_SCREEN,
         AUTOMATIC_REBOOT_MANAGER_TEST_SCENARIO_KIOSK_APP_SESSION,
         AUTOMATIC_REBOOT_MANAGER_TEST_SCENARIO_NON_KIOSK_APP_SESSION));
 
 }  // namespace system
 }  // namespace chromeos