src/chrome/browser/profiles/profile_list_desktop_browsertest.cc

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/command_line.h"
   6 #include "base/path_service.h"
   7 #include "chrome/browser/chrome_notification_types.h"
   8 #include "chrome/browser/profiles/avatar_menu.h"
   9 #include "chrome/browser/profiles/profile_manager.h"
  10 #include "chrome/browser/profiles/profile_window.h"
  11 #include "chrome/browser/profiles/profiles_state.h"
  12 #include "chrome/browser/ui/browser.h"
  13 #include "chrome/browser/ui/browser_dialogs.h"
  14 #include "chrome/browser/ui/browser_list.h"
  15 #include "chrome/common/chrome_paths.h"
  16 #include "chrome/test/base/in_process_browser_test.h"
  17 #include "chrome/test/base/test_switches.h"
  18 #include "chrome/test/base/testing_browser_process.h"
  19 #include "content/public/test/test_utils.h"
  20
  21 namespace {
  22
  23 // An observer that returns back to test code after a new profile is
  24 // initialized.
  25 void OnUnblockOnProfileCreation(Profile* profile,
  26                                 Profile::CreateStatus status) {
  27   if (status == Profile::CREATE_STATUS_INITIALIZED)
  28     base::MessageLoop::current()->Quit();
  29 }
  30
  31 }  // namespace
  32
  33 class ProfileListDesktopBrowserTest : public InProcessBrowserTest {
  34  public:
  35   ProfileListDesktopBrowserTest() {}
  36
  37   AvatarMenu* GetAvatarMenu(ProfileInfoCache* cache) {
  38     // Reset the menu.
  39     avatar_menu_.reset(new AvatarMenu(
  40         cache,
  41         NULL,
  42         browser()));
  43     return avatar_menu_.get();
  44   }
  45
  46  private:
  47   scoped_ptr<AvatarMenu> avatar_menu_;
  48
  49   DISALLOW_COPY_AND_ASSIGN(ProfileListDesktopBrowserTest);
  50 };
  51
  52 IN_PROC_BROWSER_TEST_F(ProfileListDesktopBrowserTest, SignOut) {
  53   if (!profiles::IsMultipleProfilesEnabled())
  54     return;
  55
  56   ProfileManager* profile_manager = g_browser_process->profile_manager();
  57   Profile* current_profile = browser()->profile();
  58   ProfileInfoCache& cache = profile_manager->GetProfileInfoCache();
  59   size_t index = cache.GetIndexOfProfileWithPath(current_profile->GetPath());
  60
  61   AvatarMenu* menu = GetAvatarMenu(&cache);
  62   menu->RebuildMenu();
  63
  64   BrowserList* browser_list =
  65       BrowserList::GetInstance(chrome::GetActiveDesktop());
  66   EXPECT_EQ(1U, browser_list->size());
  67   content::WindowedNotificationObserver window_close_observer(
  68       chrome::NOTIFICATION_BROWSER_CLOSED,
  69       content::Source<Browser>(browser()));
  70
  71   EXPECT_FALSE(cache.ProfileIsSigninRequiredAtIndex(index));
  72   profiles::LockProfile(current_profile);
  73   window_close_observer.Wait();  // rely on test time-out for failure indication
  74
  75   EXPECT_TRUE(cache.ProfileIsSigninRequiredAtIndex(index));
  76   EXPECT_EQ(0U, browser_list->size());
  77
  78   // Signing out brings up the User Manager which we should close before exit.
  79   chrome::HideUserManager();
  80 }
  81
  82 IN_PROC_BROWSER_TEST_F(ProfileListDesktopBrowserTest, SwitchToProfile) {
  83 #if defined(OS_WIN) && defined(USE_ASH)
  84   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
  85   if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
  86     return;
  87 #endif
  88
  89   if (!profiles::IsMultipleProfilesEnabled())
  90     return;
  91
  92   ProfileManager* profile_manager = g_browser_process->profile_manager();
  93   Profile* current_profile = browser()->profile();
  94   ProfileInfoCache& cache = profile_manager->GetProfileInfoCache();
  95   base::FilePath path_profile1 = current_profile->GetPath();
  96   base::FilePath user_dir = cache.GetUserDataDir();
  97
  98   // Create an additional profile.
  99   base::FilePath path_profile2 = user_dir.Append(
 100       FILE_PATH_LITERAL("New Profile 2"));
 101   profile_manager->CreateProfileAsync(path_profile2,
 102                                       base::Bind(&OnUnblockOnProfileCreation),
 103                                       base::string16(), base::string16(),
 104                                       std::string());
 105
 106   // Spin to allow profile creation to take place, loop is terminated
 107   // by OnUnblockOnProfileCreation when the profile is created.
 108   content::RunMessageLoop();
 109   ASSERT_EQ(cache.GetNumberOfProfiles(), 2U);
 110
 111   AvatarMenu* menu = GetAvatarMenu(&cache);
 112   menu->RebuildMenu();
 113   BrowserList* browser_list =
 114       BrowserList::GetInstance(chrome::GetActiveDesktop());
 115   EXPECT_EQ(1U, browser_list->size());
 116   EXPECT_EQ(path_profile1, browser_list->get(0)->profile()->GetPath());
 117
 118   // Open a browser window for the first profile.
 119   menu->SwitchToProfile(cache.GetIndexOfProfileWithPath(path_profile1),
 120                         false, ProfileMetrics::SWITCH_PROFILE_ICON);
 121   EXPECT_EQ(1U, browser_list->size());
 122   EXPECT_EQ(path_profile1, browser_list->get(0)->profile()->GetPath());
 123
 124   // Open a browser window for the second profile.
 125   menu->SwitchToProfile(cache.GetIndexOfProfileWithPath(path_profile2),
 126                         false, ProfileMetrics::SWITCH_PROFILE_ICON);
 127   EXPECT_EQ(2U, browser_list->size());
 128
 129   // Switch to the first profile without opening a new window.
 130   menu->SwitchToProfile(cache.GetIndexOfProfileWithPath(path_profile1),
 131                         false, ProfileMetrics::SWITCH_PROFILE_ICON);
 132   EXPECT_EQ(2U, browser_list->size());
 133   EXPECT_EQ(path_profile1, browser_list->get(0)->profile()->GetPath());
 134   EXPECT_EQ(path_profile2, browser_list->get(1)->profile()->GetPath());
 135 }