session_manager: Tracks user sessions

chrome/browser/chromeos/login/crash_restore_browsertest.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 <string>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/run_loop.h"
 #include "chrome/browser/chromeos/login/session/user_session_manager.h"
 #include "chrome/browser/chromeos/login/session/user_session_manager_test_api.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chromeos/chromeos_switches.h"
 #include "chromeos/cryptohome/cryptohome_parameters.h"
 #include "chromeos/dbus/cryptohome_client.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/fake_session_manager_client.h"
 #include "chromeos/dbus/session_manager_client.h"
+#include "components/session_manager/core/session_manager.h"
 #include "components/user_manager/user.h"
 #include "components/user_manager/user_manager.h"
 #include "content/public/test/test_utils.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "third_party/cros_system_api/dbus/service_constants.h"
 
 namespace chromeos {
 
 namespace {
 
@@ skipping 37 matching lines @@
 };
 
 IN_PROC_BROWSER_TEST_F(CrashRestoreSimpleTest, RestoreSessionForOneUser) {
   user_manager::UserManager* user_manager = user_manager::UserManager::Get();
   user_manager::User* user = user_manager->GetActiveUser();
   ASSERT_TRUE(user);
   EXPECT_EQ(account_id1_, user->GetAccountId());
   EXPECT_EQ(CryptohomeClient::GetStubSanitizedUsername(cryptohome_id1_),
             user->username_hash());
   EXPECT_EQ(1UL, user_manager->GetLoggedInUsers().size());
+
+  auto* session_manager = session_manager::SessionManager::Get();
+  EXPECT_EQ(session_manager::SessionState::ACTIVE,
+            session_manager->session_state());
+  EXPECT_EQ(1u, session_manager->sessions().size());
 }
 
 // Observer that keeps track of user sessions restore event.
 class UserSessionRestoreObserver : public UserSessionStateObserver {
  public:
   UserSessionRestoreObserver()
       : running_loop_(false),
         user_sessions_restored_(
             UserSessionManager::GetInstance()->UserSessionsRestored()) {
     if (!user_sessions_restored_)
@@ skipping 69 matching lines @@
   // User that becomes active moves to the beginning of the list.
   EXPECT_EQ(account_id3_, users[0]->GetAccountId());
   EXPECT_EQ(CryptohomeClient::GetStubSanitizedUsername(cryptohome_id3_),
             users[0]->username_hash());
   EXPECT_EQ(account_id2_, users[1]->GetAccountId());
   EXPECT_EQ(CryptohomeClient::GetStubSanitizedUsername(cryptohome_id2_),
             users[1]->username_hash());
   EXPECT_EQ(account_id1_, users[2]->GetAccountId());
   EXPECT_EQ(CryptohomeClient::GetStubSanitizedUsername(cryptohome_id1_),
             users[2]->username_hash());
+
+  auto* session_manager = session_manager::SessionManager::Get();
+  EXPECT_EQ(session_manager::SessionState::ACTIVE,
+            session_manager->session_state());
+  EXPECT_EQ(3u, session_manager->sessions().size());
+  EXPECT_EQ(session_manager->sessions()[0].user_account_id, account_id1_);
+  EXPECT_EQ(session_manager->sessions()[1].user_account_id, account_id2_);
+  EXPECT_EQ(session_manager->sessions()[2].user_account_id, account_id3_);
 }
 
 }  // namespace chromeos