chromeos: Convert system tray session length limit to mojo

chrome/browser/ui/ash/session_controller_client_unittest.cc

 // Copyright 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/ui/ash/session_controller_client.h"
 
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/run_loop.h"
+#include "base/time/time.h"
 #include "chrome/browser/chromeos/login/users/fake_chrome_user_manager.h"
 #include "chrome/browser/chromeos/login/users/multi_profile_user_controller.h"
 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
 #include "chrome/browser/chromeos/policy/policy_cert_service.h"
 #include "chrome/browser/chromeos/policy/policy_cert_service_factory.h"
 #include "chrome/browser/chromeos/policy/policy_cert_verifier.h"
 #include "chrome/browser/chromeos/profiles/profile_helper.h"
 #include "chrome/browser/supervised_user/supervised_user_service.h"
 #include "chrome/browser/supervised_user/supervised_user_service_factory.h"
 #include "chrome/common/pref_names.h"
@@ skipping 95 matching lines @@
   void UpdateUserSession(ash::mojom::UserSessionPtr user_session) override {
     last_user_session_ = user_session->Clone();
     update_user_session_count_++;
   }
   void SetUserSessionOrder(
       const std::vector<uint32_t>& user_session_order) override {}
   void StartLock(StartLockCallback callback) override {}
   void NotifyChromeLockAnimationsComplete() override {}
   void RunUnlockAnimation(RunUnlockAnimationCallback callback) override {}
   void NotifyChromeTerminating() override {}
+  void SetSessionLengthLimit(base::TimeDelta length_limit,
+                             base::TimeTicks start_time) override {
+    last_session_length_limit_ = length_limit;
+    last_session_start_time_ = start_time;
+  }
+
+  base::TimeDelta last_session_length_limit_;
+  base::TimeTicks last_session_start_time_;
 
  private:
   mojo::Binding<ash::mojom::SessionController> binding_;
 
   ash::mojom::SessionInfoPtr last_session_info_;
   ash::mojom::UserSessionPtr last_user_session_;
   int update_user_session_count_ = 0;
 
   DISALLOW_COPY_AND_ASSIGN(TestSessionController);
 };
@@ skipping 373 matching lines @@
 
   user_prefs->SetBoolean(prefs::kEnableAutoScreenLock, true);
   SessionControllerClient::FlushForTesting();
   EXPECT_TRUE(
       session_controller.last_session_info()->should_lock_screen_automatically);
   user_prefs->SetBoolean(prefs::kEnableAutoScreenLock, false);
   SessionControllerClient::FlushForTesting();
   EXPECT_FALSE(
       session_controller.last_session_info()->should_lock_screen_automatically);
 }
+
+TEST_F(SessionControllerClientTest, SessionLengthLimit) {
+  // Create an object to test and connect it to our test interface.
+  SessionControllerClient client;
+  TestSessionController session_controller;
+  client.session_controller_ = session_controller.CreateInterfacePtrAndBind();
+  client.Init();
+  SessionControllerClient::FlushForTesting();
+
+  // By default there is no session length limit.
+  EXPECT_TRUE(session_controller.last_session_length_limit_.is_zero());
+  EXPECT_TRUE(session_controller.last_session_start_time_.is_null());
+
+  // Setting a session length limit in local state sends it to ash.
+  const base::TimeDelta length_limit = base::TimeDelta::FromHours(1);
+  const base::TimeTicks start_time = base::TimeTicks::Now();
+  PrefService* local_state = TestingBrowserProcess::GetGlobal()->local_state();
+  local_state->SetInteger(prefs::kSessionLengthLimit,
+                          length_limit.InMilliseconds());
+  local_state->SetInt64(prefs::kSessionStartTime, start_time.ToInternalValue());
+  SessionControllerClient::FlushForTesting();
+  EXPECT_EQ(length_limit, session_controller.last_session_length_limit_);
+  EXPECT_EQ(start_time, session_controller.last_session_start_time_);
+}