src/chrome/browser/ui/views/profiles/new_avatar_menu_button_browsertest.cc

   1 // Copyright 2014 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 "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/chrome_notification_types.h"
   9 #include "chrome/browser/profiles/profile_manager.h"
  10 #include "chrome/browser/profiles/profiles_state.h"
  11 #include "chrome/browser/ui/browser_list.h"
  12 #include "chrome/browser/ui/user_manager.h"
  13 #include "chrome/browser/ui/views/frame/browser_view.h"
  14 #include "chrome/browser/ui/views/profiles/avatar_menu_button.h"
  15 #include "chrome/browser/ui/views/profiles/new_avatar_button.h"
  16 #include "chrome/browser/ui/views/profiles/profile_chooser_view.h"
  17 #include "chrome/browser/ui/views/profiles/user_manager_view.h"
  18 #include "chrome/common/chrome_paths.h"
  19 #include "chrome/common/chrome_switches.h"
  20 #include "chrome/test/base/in_process_browser_test.h"
  21 #include "chrome/test/base/test_switches.h"
  22 #include "chrome/test/base/testing_browser_process.h"
  23 #include "components/signin/core/common/profile_management_switches.h"
  24 #include "content/public/test/test_utils.h"
  25 #include "ui/views/controls/button/label_button.h"
  26
  27 class NewAvatarMenuButtonTest : public InProcessBrowserTest {
  28  public:
  29   NewAvatarMenuButtonTest();
  30   virtual ~NewAvatarMenuButtonTest();
  31
  32  protected:
  33   virtual void SetUp() OVERRIDE;
  34   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE;
  35   void CreateTestingProfile();
  36   void StartAvatarMenu();
  37
  38  private:
  39   DISALLOW_COPY_AND_ASSIGN(NewAvatarMenuButtonTest);
  40 };
  41
  42 NewAvatarMenuButtonTest::NewAvatarMenuButtonTest() {
  43 }
  44
  45 NewAvatarMenuButtonTest::~NewAvatarMenuButtonTest() {
  46 }
  47
  48 void NewAvatarMenuButtonTest::SetUp() {
  49   InProcessBrowserTest::SetUp();
  50   DCHECK(switches::IsNewAvatarMenu());
  51 }
  52
  53 void NewAvatarMenuButtonTest::SetUpCommandLine(CommandLine* command_line) {
  54   switches::EnableNewAvatarMenuForTesting(command_line);
  55 }
  56
  57 void NewAvatarMenuButtonTest::CreateTestingProfile() {
  58   ProfileManager* profile_manager = g_browser_process->profile_manager();
  59   EXPECT_EQ(1u, profile_manager->GetNumberOfProfiles());
  60
  61   // Sign in the default profile
  62   ProfileInfoCache& cache = profile_manager->GetProfileInfoCache();
  63   cache.SetUserNameOfProfileAtIndex(0, base::UTF8ToUTF16("user_name"));
  64
  65   base::FilePath path;
  66   PathService::Get(chrome::DIR_USER_DATA, &path);
  67   path = path.AppendASCII("test_profile");
  68   if (!base::PathExists(path))
  69     ASSERT_TRUE(base::CreateDirectory(path));
  70   Profile* profile =
  71       Profile::CreateProfile(path, NULL, Profile::CREATE_MODE_SYNCHRONOUS);
  72   profile_manager->RegisterTestingProfile(profile, true, false);
  73   EXPECT_EQ(2u, profile_manager->GetNumberOfProfiles());
  74 }
  75
  76 void NewAvatarMenuButtonTest::StartAvatarMenu() {
  77   BrowserView* browser_view = reinterpret_cast<BrowserView*>(
  78       browser()->window());
  79
  80   // Ensure that the avatar icon button is not also showing.
  81   NewAvatarButton* button = browser_view->frame()->GetNewAvatarMenuButton();
  82   ASSERT_TRUE(button);
  83   ASSERT_FALSE(browser_view->frame()->GetAvatarMenuButton());
  84
  85   ProfileChooserView::clear_close_on_deactivate_for_testing();
  86   ui::MouseEvent mouse_ev(ui::ET_MOUSE_RELEASED, gfx::Point(), gfx::Point(), 0,
  87                           0);
  88   button->NotifyClick(mouse_ev);
  89   base::MessageLoop::current()->RunUntilIdle();
  90   EXPECT_TRUE(ProfileChooserView::IsShowing());
  91 }
  92
  93 #if defined(OS_CHROMEOS) || defined (OS_LINUX) || defined (OS_WIN)
  94 // This test doesn't make sense for ChromeOS since it has a different
  95 // multi-profiles menu in the system tray instead.
  96 //
  97 // Test fails flakily on Linux and Windows http://crbug.com/352710
  98 #define MAYBE_SignOut DISABLED_SignOut
  99 #else
 100 #define MAYBE_SignOut SignOut
 101 #endif
 102
 103 IN_PROC_BROWSER_TEST_F(NewAvatarMenuButtonTest, MAYBE_SignOut) {
 104   // If multiprofile mode is not enabled, you can't switch between profiles.
 105   if (!profiles::IsMultipleProfilesEnabled())
 106     return;
 107
 108   CreateTestingProfile();
 109   ASSERT_NO_FATAL_FAILURE(StartAvatarMenu());
 110
 111   BrowserList* browser_list =
 112       BrowserList::GetInstance(chrome::GetActiveDesktop());
 113   EXPECT_EQ(1U, browser_list->size());
 114   content::WindowedNotificationObserver window_close_observer(
 115       chrome::NOTIFICATION_BROWSER_CLOSED,
 116       content::Source<Browser>(browser()));
 117
 118   AvatarMenu* menu =
 119       ProfileChooserView::profile_bubble_->avatar_menu_.get();
 120   const AvatarMenu::Item& menu_item_before =
 121       menu->GetItemAt(menu->GetActiveProfileIndex());
 122   EXPECT_FALSE(menu_item_before.signin_required);
 123
 124   ui::MouseEvent mouse_ev(
 125       ui::ET_MOUSE_RELEASED, gfx::Point(), gfx::Point(), 0, 0);
 126   ProfileChooserView::profile_bubble_->ButtonPressed(
 127       ProfileChooserView::profile_bubble_->lock_button_, mouse_ev);
 128
 129   EXPECT_TRUE(menu->GetItemAt(menu->GetActiveProfileIndex()).signin_required);
 130
 131   window_close_observer.Wait();  // Rely on test timeout for failure indication.
 132   EXPECT_TRUE(browser_list->empty());
 133
 134   // If the User Manager hasn't shown yet, wait for it to show up.
 135   if (!UserManager::IsShowing())
 136     base::MessageLoop::current()->RunUntilIdle();
 137
 138   // We need to hide the User Manager or else the process can't die.
 139   UserManager::Hide();
 140 }