cros: Tighten profile loading

chrome/browser/chromeos/system/tray_accessibility_browsertest.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 "ash/accessibility_types.h"
 #include "ash/login_status.h"
 #include "ash/magnifier/magnification_controller.h"
 #include "ash/shell.h"
 #include "ash/system/tray/system_tray.h"
 #include "ash/system/tray/system_tray_test_api.h"
 #include "ash/system/tray_accessibility.h"
 #include "ash/test/shell_test_api.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/chromeos/accessibility/accessibility_manager.h"
 #include "chrome/browser/chromeos/accessibility/magnification_manager.h"
 #include "chrome/browser/chromeos/login/helper.h"
 #include "chrome/browser/chromeos/login/startup_utils.h"
+#include "chrome/browser/chromeos/profiles/profile_helper.h"
 #include "chrome/browser/extensions/api/braille_display_private/mock_braille_controller.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/ash/session_controller_client.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chromeos/chromeos_switches.h"
 #include "components/policy/core/browser/browser_policy_connector.h"
@@ skipping 24 matching lines @@
 
 enum PrefSettingMechanism {
   PREF_SERVICE,
   POLICY,
 };
 
 void SetMagnifierEnabled(bool enabled) {
   MagnificationManager::Get()->SetMagnifierEnabled(enabled);
 }
 
+// Simulates how UserSessionManager creates and starts a user session.
+void CreateAndStartUserSession(const AccountId& account_id) {
+  using session_manager::SessionManager;
+
+  const std::string user_id_hash =
+      ProfileHelper::GetUserIdHashByUserIdForTesting(account_id.GetUserEmail());
+
+  SessionManager::Get()->CreateSession(account_id, user_id_hash);
+  ProfileHelper::GetProfileByUserIdHashForTest(user_id_hash);
+  SessionManager::Get()->SessionStarted();
+}
+
 class TrayAccessibilityTest
     : public InProcessBrowserTest,
       public WithParamInterface<PrefSettingMechanism> {
  protected:
   TrayAccessibilityTest() {}
   virtual ~TrayAccessibilityTest() {}
 
   // The profile which should be used by these tests.
   Profile* GetProfile() { return ProfileManager::GetActiveUserProfile(); }
 
@@ skipping 268 matching lines @@
         *braille_controller_.GetDisplayState());
   }
 
   policy::MockConfigurationPolicyProvider provider_;
   MockBrailleController braille_controller_;
 };
 
 IN_PROC_BROWSER_TEST_P(TrayAccessibilityTest, LoginStatus) {
   EXPECT_EQ(ash::LoginStatus::NOT_LOGGED_IN, GetLoginStatus());
 
-  session_manager::SessionManager::Get()->CreateSession(
-      AccountId::FromUserEmail("owner@invalid.domain"), "owner@invalid.domain");
-  session_manager::SessionManager::Get()->SessionStarted();
+  CreateAndStartUserSession(AccountId::FromUserEmail("owner@invalid.domain"));
   // Flush to ensure the session state reaches ash and updates login status.
   SessionControllerClient::FlushForTesting();
 
   EXPECT_EQ(ash::LoginStatus::USER, GetLoginStatus());
 }
 
 IN_PROC_BROWSER_TEST_P(TrayAccessibilityTest, ShowTrayIcon) {
   SetLoginStatus(ash::LoginStatus::NOT_LOGGED_IN);
 
   // Confirms that the icon is invisible before login.
   EXPECT_FALSE(IsTrayIconVisible());
 
-  session_manager::SessionManager::Get()->CreateSession(
-      AccountId::FromUserEmail("owner@invalid.domain"), "owner@invalid.domain");
-  session_manager::SessionManager::Get()->SessionStarted();
+  CreateAndStartUserSession(AccountId::FromUserEmail("owner@invalid.domain"));
 
   // Confirms that the icon is invisible just after login.
   EXPECT_FALSE(IsTrayIconVisible());
 
   // Toggling spoken feedback changes the visibillity of the icon.
   AccessibilityManager::Get()->EnableSpokenFeedback(
       true, ash::A11Y_NOTIFICATION_NONE);
   EXPECT_TRUE(IsTrayIconVisible());
   AccessibilityManager::Get()->EnableSpokenFeedback(
       false, ash::A11Y_NOTIFICATION_NONE);
@@ skipping 118 matching lines @@
   // the icon on the tray.
   SetShowAccessibilityOptionsInSystemTrayMenu(true);
   AccessibilityManager::Get()->EnableHighContrast(true);
   EXPECT_TRUE(IsTrayIconVisible());
   AccessibilityManager::Get()->EnableHighContrast(false);
   EXPECT_FALSE(IsTrayIconVisible());
 }
 
 IN_PROC_BROWSER_TEST_P(TrayAccessibilityTest, ShowMenu) {
   // Login
-  session_manager::SessionManager::Get()->CreateSession(
-      AccountId::FromUserEmail("owner@invalid.domain"), "owner@invalid.domain");
-  session_manager::SessionManager::Get()->SessionStarted();
+  CreateAndStartUserSession(AccountId::FromUserEmail("owner@invalid.domain"));
   // Flush to ensure the session state reaches ash and updates login status.
   SessionControllerClient::FlushForTesting();
 
   SetShowAccessibilityOptionsInSystemTrayMenu(false);
 
   // Confirms that the menu is hidden.
   EXPECT_FALSE(CanCreateMenuItem());
 
   // Toggling spoken feedback changes the visibillity of the menu.
   AccessibilityManager::Get()->EnableSpokenFeedback(
@@ skipping 117 matching lines @@
   AccessibilityManager::Get()->SetFocusHighlightEnabled(false);
   EXPECT_TRUE(CanCreateMenuItem());
   AccessibilityManager::Get()->EnableStickyKeys(false);
   EXPECT_TRUE(CanCreateMenuItem());
   AccessibilityManager::Get()->EnableTapDragging(false);
   EXPECT_FALSE(CanCreateMenuItem());
 }
 
 IN_PROC_BROWSER_TEST_P(TrayAccessibilityTest, ShowMenuWithShowMenuOption) {
   // Login
-  session_manager::SessionManager::Get()->CreateSession(
-      AccountId::FromUserEmail("owner@invalid.domain"), "owner@invalid.domain");
-  session_manager::SessionManager::Get()->SessionStarted();
+  CreateAndStartUserSession(AccountId::FromUserEmail("owner@invalid.domain"));
   // Flush to ensure the session state reaches ash and updates login status.
   SessionControllerClient::FlushForTesting();
 
   SetShowAccessibilityOptionsInSystemTrayMenu(true);
 
   // Confirms that the menu is visible.
   EXPECT_TRUE(CanCreateMenuItem());
 
   // The menu remains visible regardless of toggling spoken feedback.
   AccessibilityManager::Get()->EnableSpokenFeedback(
@@ skipping 968 matching lines @@
   EXPECT_TRUE(IsLargeCursorMenuShownOnDetailMenu());
   EXPECT_TRUE(IsMonoAudioMenuShownOnDetailMenu());
   EXPECT_TRUE(IsCaretHighlightMenuShownOnDetailMenu());
   EXPECT_TRUE(IsHighlightMouseCursorMenuShownOnDetailMenu());
   EXPECT_TRUE(IsHighlightKeyboardFocusMenuShownOnDetailMenu());
   EXPECT_TRUE(IsStickyKeysMenuShownOnDetailMenu());
   EXPECT_TRUE(IsTapDraggingMenuShownOnDetailMenu());
   CloseDetailMenu();
 
   // Simulate login.
-  session_manager::SessionManager::Get()->CreateSession(
-      AccountId::FromUserEmail("owner@invalid.domain"), "owner@invalid.domain");
-  session_manager::SessionManager::Get()->SessionStarted();
+  CreateAndStartUserSession(AccountId::FromUserEmail("owner@invalid.domain"));
   // Flush to ensure the session state reaches ash and updates login status.
   SessionControllerClient::FlushForTesting();
   EXPECT_TRUE(CreateDetailedMenu());
   EXPECT_TRUE(IsSpokenFeedbackMenuShownOnDetailMenu());
   EXPECT_TRUE(IsHighContrastMenuShownOnDetailMenu());
   EXPECT_TRUE(IsScreenMagnifierMenuShownOnDetailMenu());
   EXPECT_TRUE(IsAutoclickMenuShownOnDetailMenu());
   EXPECT_TRUE(IsVirtualKeyboardMenuShownOnDetailMenu());
   EXPECT_TRUE(IsHelpAvailableOnDetailMenu());
   EXPECT_TRUE(IsSettingsAvailableOnDetailMenu());
@@ skipping 50 matching lines @@
   EXPECT_TRUE(IsTapDraggingMenuShownOnDetailMenu());
   CloseDetailMenu();
 }
 
 INSTANTIATE_TEST_CASE_P(TrayAccessibilityTestInstance,
                         TrayAccessibilityTest,
                         testing::Values(PREF_SERVICE,
                                         POLICY));
 
 }  // namespace chromeos