Add support in TestMockTimeTaskRunner for vending out mock Time and mock Clocks.

chrome/browser/chromeos/session_length_limiter_unittest.cc

 // 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.
 
 #include "chrome/browser/chromeos/session_length_limiter.h"
 
 #include "base/compiler_specific.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/prefs/testing_pref_service.h"
@@ skipping 154 matching lines @@
   if (session_length_limiter_)
     session_length_limiter_->OnUserActivity(NULL);
   UpdateSessionStartTimeIfWaitingForUserActivity();
   user_activity_seen_ = true;
 }
 
 void SessionLengthLimiterTest::
     UpdateSessionStartTimeIfWaitingForUserActivity() {
   if (!user_activity_seen_ &&
       local_state_.GetBoolean(prefs::kSessionWaitForInitialUserActivity)) {
-    session_start_time_ = runner_->GetCurrentMockTime();
+    session_start_time_ = runner_->NowTicks();
   }
 }
 
 void SessionLengthLimiterTest::ExpectStopSession() {
   Mock::VerifyAndClearExpectations(delegate_);
   EXPECT_CALL(*delegate_, StopSession())
       .Times(1)
       .WillOnce(Invoke(this, &SessionLengthLimiterTest::SaveSessionStopTime));
 }
 
 void SessionLengthLimiterTest::SaveSessionStopTime() {
-  session_stop_time_ = runner_->GetCurrentMockTime();
+  session_stop_time_ = runner_->NowTicks();
 }
 
 void SessionLengthLimiterTest::CreateSessionLengthLimiter(
     bool browser_restarted) {
   user_activity_seen_ = false;
-  session_start_time_ = runner_->GetCurrentMockTime();
+  session_start_time_ = runner_->NowTicks();
 
   EXPECT_FALSE(delegate_);
   delegate_ = new NiceMock<MockSessionLengthLimiterDelegate>;
   ON_CALL(*delegate_, GetCurrentTime())
-      .WillByDefault(Invoke(
-          runner_.get(), &base::TestMockTimeTaskRunner::GetCurrentMockTime));
+      .WillByDefault(
+          Invoke(runner_.get(), &base::TestMockTimeTaskRunner::NowTicks));
   EXPECT_CALL(*delegate_, StopSession()).Times(0);
   session_length_limiter_.reset(
       new SessionLengthLimiter(delegate_, browser_restarted));
 }
 
 void SessionLengthLimiterTest::DestroySessionLengthLimiter() {
   session_length_limiter_.reset();
   delegate_ = NULL;
 }
 
@@ skipping 67 matching lines @@
   // Pref indicating user activity set. Session start time not set.
   SetSessionUserActivitySeenPref(true);
   ClearSessionStartTimePref();
   CreateSessionLengthLimiter(false);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity not set. Session start time in the future.
   ClearSessionUserActivitySeenPref();
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() + base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() + base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(false);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity set. Session start time in the future.
   SetSessionUserActivitySeenPref(true);
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() + base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() + base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(false);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity not set. Session start time valid.
   ClearSessionUserActivitySeenPref();
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() - base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() - base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(false);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity set. Session start time valid.
   SetSessionUserActivitySeenPref(true);
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() - base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() - base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(false);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 }
 
 // Verifies that when not instructed to wait for initial user activity, local
 // state is correctly updated during restart after a crash:
 // * If no valid session start time is found in local state, the session start
 //   time is set and the pref indicating user activity is cleared.
@@ skipping 11 matching lines @@
   // Pref indicating user activity set. Session start time not set.
   SetSessionUserActivitySeenPref(true);
   ClearSessionStartTimePref();
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_EQ(session_start_time_, GetSessionStartTimePref());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity not set. Session start time in the future.
   ClearSessionUserActivitySeenPref();
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() + base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() + base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_EQ(session_start_time_, GetSessionStartTimePref());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity set. Session start time in the future.
   SetSessionUserActivitySeenPref(true);
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() + base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() + base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_EQ(session_start_time_, GetSessionStartTimePref());
   DestroySessionLengthLimiter();
 
   const base::TimeTicks stored_session_start_time =
-      runner_->GetCurrentMockTime() - base::TimeDelta::FromHours(2);
+      runner_->NowTicks() - base::TimeDelta::FromHours(2);
 
   // Pref indicating user activity not set. Session start time valid.
   ClearSessionUserActivitySeenPref();
   SetSessionStartTimePref(stored_session_start_time);
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_EQ(stored_session_start_time, GetSessionStartTimePref());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity set. Session start time valid.
@@ skipping 26 matching lines @@
   // Pref indicating user activity set. Session start time not set.
   SetSessionUserActivitySeenPref(true);
   ClearSessionStartTimePref();
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity not set. Session start time in the future.
   ClearSessionUserActivitySeenPref();
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() + base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() + base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity set. Session start time in the future.
   SetSessionUserActivitySeenPref(true);
-  SetSessionStartTimePref(
-      runner_->GetCurrentMockTime() + base::TimeDelta::FromHours(2));
+  SetSessionStartTimePref(runner_->NowTicks() + base::TimeDelta::FromHours(2));
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_FALSE(IsSessionStartTimePrefSet());
   DestroySessionLengthLimiter();
 
   const base::TimeTicks stored_session_start_time =
-      runner_->GetCurrentMockTime() - base::TimeDelta::FromHours(2);
+      runner_->NowTicks() - base::TimeDelta::FromHours(2);
 
   // Pref indicating user activity not set. Session start time valid.
   ClearSessionUserActivitySeenPref();
   SetSessionStartTimePref(stored_session_start_time);
   CreateSessionLengthLimiter(true);
   EXPECT_FALSE(IsSessionUserActivitySeenPrefSet());
   EXPECT_EQ(stored_session_start_time, GetSessionStartTimePref());
   DestroySessionLengthLimiter();
 
   // Pref indicating user activity set. Session start time valid.
@@ skipping 300 matching lines @@
   runner_->FastForwardBy(base::TimeDelta::FromSeconds(50));
 
   // Remove the session length limit.
   ClearSessionLengthLimitPref();
 
   // Verify that no timer fires to terminate the session.
   runner_->FastForwardUntilNoTasksRemain();
 }
 
 }  // namespace chromeos