Track whether a given user session has completed initialization, and use

chrome/browser/chromeos/login/users/user_manager_unittest.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <cstdlib>
 #include <cstring>
 #include <memory>
 
 #include "base/command_line.h"
 #include "base/files/file_util.h"
@@ skipping 222 matching lines @@
   EXPECT_EQ(1U, user_manager::UserManager::Get()->GetUnlockUsers().size());
 
   // The user is not allowed to lock the screen.
   profile->GetPrefs()->SetBoolean(prefs::kAllowScreenLock, false);
   EXPECT_FALSE(user_manager::UserManager::Get()->CanCurrentUserLock());
   EXPECT_EQ(0U, user_manager::UserManager::Get()->GetUnlockUsers().size());
 
   ResetUserManager();
 }
 
+TEST_F(UserManagerTest, SessionInitialized) {
+  user_manager::UserManager::Get()->UserLoggedIn(
+      owner_account_id_at_invalid_domain_,
+      owner_account_id_at_invalid_domain_.GetUserEmail(), false);
+  const user_manager::UserList* users =
+      &user_manager::UserManager::Get()->GetUsers();
+  ASSERT_EQ(1U, users->size());
+  EXPECT_FALSE((*users)[0]->session_initialized());
+  ResetUserManager();
+  users = &user_manager::UserManager::Get()->GetUsers();
+  ASSERT_EQ(1U, users->size());
+  EXPECT_FALSE((*users)[0]->session_initialized());
+}
+
+TEST_F(UserManagerTest, SessionInitializedMigration) {
+  user_manager::UserManager::Get()->UserLoggedIn(
+      owner_account_id_at_invalid_domain_,
+      owner_account_id_at_invalid_domain_.GetUserEmail(), false);
+  const user_manager::UserList* users =
+      &user_manager::UserManager::Get()->GetUsers();
+  ASSERT_EQ(1U, users->size());
+  EXPECT_FALSE((*users)[0]->session_initialized());
+
+  // Clear the stored user data - when UserManager loads again, it should
+  // migrate existing users by setting session_initialized to true for them.
+  local_state_->Get()->ClearPref("UserSessionInitialized");
+  ResetUserManager();
+  users = &user_manager::UserManager::Get()->GetUsers();
+  ASSERT_EQ(1U, users->size());
+  EXPECT_TRUE((*users)[0]->session_initialized());
+}
+
 }  // namespace chromeos