chrome/browser/chromeos/profiles/profile_list_chromeos_unittest.cc

   1 // Copyright 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 <string>
   6
   7 #include "ash/ash_switches.h"
   8 #include "base/command_line.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/strings/string16.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #include "chrome/browser/chromeos/login/fake_user_manager.h"
  13 #include "chrome/browser/chromeos/profiles/profile_helper.h"
  14 #include "chrome/browser/prefs/pref_service_syncable.h"
  15 #include "chrome/browser/profiles/avatar_menu.h"
  16 #include "chrome/browser/profiles/avatar_menu_observer.h"
  17 #include "chrome/browser/profiles/profile_info_cache.h"
  18 #include "chrome/browser/ui/ash/chrome_shell_delegate.h"
  19 #include "chrome/common/chrome_constants.h"
  20 #include "chrome/common/chrome_switches.h"
  21 #include "chrome/test/base/testing_browser_process.h"
  22 #include "chrome/test/base/testing_profile_manager.h"
  23 #include "testing/gtest/include/gtest/gtest.h"
  24 #include "ui/base/l10n/l10n_util.h"
  25
  26 using base::ASCIIToUTF16;
  27
  28 namespace {
  29
  30 // As defined in /chromeos/dbus/cryptohome_client.cc.
  31 static const char kUserIdHashSuffix[] = "-hash";
  32
  33 class MockObserver : public AvatarMenuObserver {
  34  public:
  35   MockObserver() : count_(0) {}
  36   virtual ~MockObserver() {}
  37
  38   virtual void OnAvatarMenuChanged(
  39       AvatarMenu* avatar_menu) OVERRIDE {
  40     ++count_;
  41   }
  42
  43   int change_count() const { return count_; }
  44
  45  private:
  46   int count_;
  47
  48   DISALLOW_COPY_AND_ASSIGN(MockObserver);
  49 };
  50
  51 }  // namespace
  52
  53 namespace chromeos {
  54
  55 class ProfileListChromeOSTest : public testing::Test {
  56  public:
  57   ProfileListChromeOSTest()
  58       : manager_(TestingBrowserProcess::GetGlobal()) {
  59   }
  60
  61   virtual void SetUp() {
  62     ASSERT_TRUE(manager_.SetUp());
  63
  64     // AvatarMenu and multiple profiles works after user logged in.
  65     manager_.SetLoggedIn(true);
  66
  67     // We only instantiate UserMenuModel if multi-profile mode is enabled.
  68     CommandLine* cl = CommandLine::ForCurrentProcess();
  69     cl->AppendSwitch(switches::kMultiProfiles);
  70
  71     // Initialize the UserManager singleton to a fresh FakeUserManager instance.
  72     user_manager_enabler_.reset(
  73         new ScopedUserManagerEnabler(new FakeUserManager));
  74   }
  75
  76   FakeUserManager* GetFakeUserManager() {
  77     return static_cast<FakeUserManager*>(UserManager::Get());
  78   }
  79
  80   void AddProfile(base::string16 name, bool log_in) {
  81     std::string email_string = base::UTF16ToASCII(name) + "@example.com";
  82
  83     // Add a user to the fake user manager.
  84     GetFakeUserManager()->AddUser(email_string);
  85     if (log_in) {
  86       GetFakeUserManager()->UserLoggedIn(
  87           email_string,
  88           email_string + kUserIdHashSuffix,
  89           false);
  90     }
  91
  92     // Create a profile for the user.
  93     manager()->CreateTestingProfile(
  94         chrome::kProfileDirPrefix + email_string + kUserIdHashSuffix,
  95         scoped_ptr<PrefServiceSyncable>(),
  96         ASCIIToUTF16(email_string), 0, std::string(),
  97         TestingProfile::TestingFactories());
  98   }
  99
 100   AvatarMenu* GetAvatarMenu() {
 101     // Reset the MockObserver.
 102     mock_observer_.reset(new MockObserver());
 103     EXPECT_EQ(0, change_count());
 104
 105     // Reset the menu.
 106     avatar_menu_.reset(new AvatarMenu(
 107         manager()->profile_info_cache(),
 108         mock_observer_.get(),
 109         NULL));
 110     avatar_menu_->RebuildMenu();
 111     EXPECT_EQ(0, change_count());
 112     return avatar_menu_.get();
 113   }
 114
 115   void ActiveUserChanged(ProfileHelper* profile_helper,
 116                          const std::string& hash) {
 117     profile_helper->ActiveUserHashChanged(hash);
 118   }
 119
 120   TestingProfileManager* manager() { return &manager_; }
 121
 122   int change_count() const { return mock_observer_->change_count(); }
 123
 124  private:
 125   TestingProfileManager manager_;
 126   scoped_ptr<MockObserver> mock_observer_;
 127   scoped_ptr<ScopedUserManagerEnabler> user_manager_enabler_;
 128   scoped_ptr<AvatarMenu> avatar_menu_;
 129   ChromeShellDelegate chrome_shell_delegate_;
 130
 131   DISALLOW_COPY_AND_ASSIGN(ProfileListChromeOSTest);
 132 };
 133
 134 TEST_F(ProfileListChromeOSTest, InitialCreation) {
 135   base::string16 name1(ASCIIToUTF16("p1"));
 136
 137   AddProfile(name1, true);
 138
 139   AvatarMenu* menu = GetAvatarMenu();
 140
 141   ASSERT_EQ(1U, menu->GetNumberOfItems());
 142
 143   const AvatarMenu::Item& item1 = menu->GetItemAt(0);
 144   EXPECT_EQ(0U, item1.menu_index);
 145   EXPECT_EQ(name1, item1.name);
 146 }
 147
 148 TEST_F(ProfileListChromeOSTest, ShowLoggedInUsers) {
 149   base::string16 name1(ASCIIToUTF16("p1"));
 150   base::string16 name2(ASCIIToUTF16("p2"));
 151   base::string16 name3(ASCIIToUTF16("p3"));
 152   base::string16 name4(ASCIIToUTF16("p4"));
 153
 154   AddProfile(name1, true);
 155   AddProfile(name2, false);
 156   AddProfile(name3, true);
 157   AddProfile(name4, false);
 158
 159   AvatarMenu* menu = GetAvatarMenu();
 160
 161   ASSERT_EQ(2U, menu->GetNumberOfItems());
 162
 163   const AvatarMenu::Item& item1 = menu->GetItemAt(0);
 164   EXPECT_EQ(0U, item1.menu_index);
 165   EXPECT_EQ(name1, item1.name);
 166
 167   const AvatarMenu::Item& item3 = menu->GetItemAt(1);
 168   EXPECT_EQ(1U, item3.menu_index);
 169   EXPECT_EQ(name3, item3.name);
 170 }
 171
 172 TEST_F(ProfileListChromeOSTest, DontShowManagedUsers) {
 173   base::string16 name1(ASCIIToUTF16("p1"));
 174   base::string16 managed_name(ASCIIToUTF16("p2@example.com"));
 175
 176   AddProfile(name1, true);
 177
 178   // Add a managed user profile.
 179   ProfileInfoCache* cache = manager()->profile_info_cache();
 180   manager()->profile_info_cache()->AddProfileToCache(
 181       cache->GetUserDataDir().AppendASCII("p2"), managed_name,
 182       base::string16(), 0, "TEST_ID");
 183
 184   GetFakeUserManager()->AddUser(base::UTF16ToASCII(managed_name));
 185
 186   AvatarMenu* menu = GetAvatarMenu();
 187   ASSERT_EQ(1U, menu->GetNumberOfItems());
 188
 189   const AvatarMenu::Item& item1 = menu->GetItemAt(0);
 190   EXPECT_EQ(0U, item1.menu_index);
 191   EXPECT_EQ(name1, item1.name);
 192 }
 193
 194 TEST_F(ProfileListChromeOSTest, ShowAddProfileLink) {
 195   base::string16 name1(ASCIIToUTF16("p1.com"));
 196   base::string16 name2(ASCIIToUTF16("p2.com"));
 197
 198   AddProfile(name1, true);
 199   AddProfile(name2, false);
 200
 201   AvatarMenu* menu = GetAvatarMenu();
 202
 203   ASSERT_EQ(1U, menu->GetNumberOfItems());
 204   EXPECT_TRUE(menu->ShouldShowAddNewProfileLink());
 205 }
 206
 207 TEST_F(ProfileListChromeOSTest, DontShowAddProfileLink) {
 208   base::string16 name1(ASCIIToUTF16("p1.com"));
 209   base::string16 name2(ASCIIToUTF16("p2.com"));
 210
 211   AddProfile(name1, true);
 212   AddProfile(name2, true);
 213
 214   AvatarMenu* menu = GetAvatarMenu();
 215
 216   ASSERT_EQ(2U, menu->GetNumberOfItems());
 217   EXPECT_FALSE(menu->ShouldShowAddNewProfileLink());
 218 }
 219
 220 TEST_F(ProfileListChromeOSTest, ActiveItem) {
 221   base::string16 name1(ASCIIToUTF16("p1.com"));
 222   base::string16 name2(ASCIIToUTF16("p2.com"));
 223
 224   AddProfile(name1, true);
 225   AddProfile(name2, true);
 226
 227   // Initialize ProfileHelper, it will be accessed from GetActiveProfileIndex.
 228   std::string email_string = base::UTF16ToASCII(name1) + "@example.com";
 229   std::string hash = email_string + kUserIdHashSuffix;
 230   ActiveUserChanged(
 231       g_browser_process->platform_part()->profile_helper(), hash);
 232
 233   AvatarMenu* menu = GetAvatarMenu();
 234
 235   ASSERT_EQ(2U, menu->GetNumberOfItems());
 236   // TODO(jeremy): Expand test to verify active profile index other than 0
 237   // crbug.com/100871
 238   ASSERT_EQ(0U, menu->GetActiveProfileIndex());
 239 }
 240
 241 TEST_F(ProfileListChromeOSTest, ModifyingNameResortsCorrectly) {
 242   base::string16 name1(ASCIIToUTF16("Alpha"));
 243   base::string16 name2(ASCIIToUTF16("Beta"));
 244   base::string16 newname1(ASCIIToUTF16("Gamma"));
 245
 246   AddProfile(name1, true);
 247   AddProfile(name2, true);
 248
 249   AvatarMenu* menu = GetAvatarMenu();
 250
 251   ASSERT_EQ(2U, menu->GetNumberOfItems());
 252
 253   const AvatarMenu::Item& item1 = menu->GetItemAt(0);
 254   EXPECT_EQ(0U, item1.menu_index);
 255   EXPECT_EQ(name1, item1.name);
 256
 257   const AvatarMenu::Item& item2 = menu->GetItemAt(1);
 258   EXPECT_EQ(1U, item2.menu_index);
 259   EXPECT_EQ(name2, item2.name);
 260
 261   // Change name of the first profile, to trigger resorting of the profiles:
 262   // now the first menu item should be named "beta", and the second be "gamma".
 263   GetFakeUserManager()->SaveUserDisplayName(
 264       base::UTF16ToASCII(name1) + "@example.com", newname1);
 265   manager()->profile_info_cache()->SetNameOfProfileAtIndex(0, newname1);
 266
 267   const AvatarMenu::Item& item1next = menu->GetItemAt(0);
 268   EXPECT_GT(change_count(), 1);
 269   EXPECT_EQ(0U, item1next.menu_index);
 270   EXPECT_EQ(name2, item1next.name);
 271
 272   const AvatarMenu::Item& item2next = menu->GetItemAt(1);
 273   EXPECT_EQ(1U, item2next.menu_index);
 274   EXPECT_EQ(newname1, item2next.name);
 275 }
 276
 277 TEST_F(ProfileListChromeOSTest, ChangeOnNotify) {
 278   base::string16 name1(ASCIIToUTF16("p1.com"));
 279   base::string16 name2(ASCIIToUTF16("p2.com"));
 280
 281   AddProfile(name1, true);
 282   AddProfile(name2, true);
 283
 284   AvatarMenu* menu = GetAvatarMenu();
 285   EXPECT_EQ(2U, menu->GetNumberOfItems());
 286
 287   base::string16 name3(ASCIIToUTF16("p3.com"));
 288   AddProfile(name3, true);
 289
 290   // Four changes happened via the call to CreateTestingProfile: adding the
 291   // profile to the cache, setting the user name, rebuilding the list of
 292   // profiles after the name change, and changing the avatar.
 293   // TODO(michaelpg): Determine why actual change number does not match comment.
 294   EXPECT_GE(change_count(), 4);
 295   ASSERT_EQ(3U, menu->GetNumberOfItems());
 296
 297   const AvatarMenu::Item& item1 = menu->GetItemAt(0);
 298   EXPECT_EQ(0U, item1.menu_index);
 299   EXPECT_EQ(name1, item1.name);
 300
 301   const AvatarMenu::Item& item2 = menu->GetItemAt(1);
 302   EXPECT_EQ(1U, item2.menu_index);
 303   EXPECT_EQ(name2, item2.name);
 304
 305   const AvatarMenu::Item& item3 = menu->GetItemAt(2);
 306   EXPECT_EQ(2U, item3.menu_index);
 307   EXPECT_EQ(name3, item3.name);
 308 }
 309
 310 TEST_F(ProfileListChromeOSTest, DontShowAvatarMenu) {
 311   // If in the new M-32 UX mode the icon gets shown, the menu will not.
 312   base::string16 name1(ASCIIToUTF16("p1"));
 313   base::string16 name2(ASCIIToUTF16("p2"));
 314
 315   AddProfile(name1, true);
 316
 317   // Should only show avatar menu with multiple users.
 318   EXPECT_FALSE(AvatarMenu::ShouldShowAvatarMenu());
 319
 320   AddProfile(name2, false);
 321
 322   EXPECT_FALSE(AvatarMenu::ShouldShowAvatarMenu());
 323 }
 324
 325 }  // namespace chromeos