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"
 #include "base/files/scoped_temp_dir.h"
 #include "base/run_loop.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/login/users/chrome_user_manager_impl.h"
 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
 #include "chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager.h"
 #include "chrome/browser/chromeos/profiles/profile_helper.h"
 #include "chrome/browser/chromeos/settings/scoped_cros_settings_test_helper.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/scoped_testing_local_state.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chromeos/chromeos_switches.h"
 #include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/settings/cros_settings_names.h"
 #include "components/prefs/pref_service.h"
+#include "components/user_manager/known_user.h"
 #include "components/user_manager/user.h"
 #include "components/user_manager/user_manager.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chromeos {
 
 class UnittestProfileManager : public ::ProfileManagerWithoutInit {
  public:
@@ skipping 194 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, ProfileInitialized) {
+  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]->profile_ever_initialized());
+  ResetUserManager();
+  users = &user_manager::UserManager::Get()->GetUsers();
+  ASSERT_EQ(1U, users->size());
+  EXPECT_FALSE((*users)[0]->profile_ever_initialized());
+}
+
+TEST_F(UserManagerTest, ProfileInitializedMigration) {
+  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]->profile_ever_initialized());
+
+  // Clear the stored user data - when UserManager loads again, it should
+  // migrate existing users by setting session_initialized to true for them.
+  user_manager::known_user::RemovePrefsForTesting((*users)[0]->GetAccountId());
+  ResetUserManager();
+  users = &user_manager::UserManager::Get()->GetUsers();
+  ASSERT_EQ(1U, users->size());
+  EXPECT_TRUE((*users)[0]->profile_ever_initialized());
+}
+
 }  // namespace chromeos