| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 #import "chrome/browser/ui/cocoa/profile_menu_controller.h" | |
| 6 | |
| 7 #include "base/mac/scoped_nsobject.h" | |
| 8 #include "base/strings/utf_string_conversions.h" | |
| 9 #include "base/threading/thread_restrictions.h" | |
| 10 #include "chrome/browser/browser_process.h" | |
| 11 #include "chrome/browser/prefs/pref_service_syncable.h" | |
| 12 #include "chrome/browser/profiles/profile_manager.h" | |
| 13 #include "chrome/browser/ui/browser_list.h" | |
| 14 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h" | |
| 15 #include "chrome/browser/ui/cocoa/run_loop_testing.h" | |
| 16 #include "chrome/common/pref_names.h" | |
| 17 #include "chrome/test/base/testing_profile.h" | |
| 18 #include "chrome/test/base/test_browser_window.h" | |
| 19 #include "grit/generated_resources.h" | |
| 20 #include "testing/gtest_mac.h" | |
| 21 #include "ui/base/l10n/l10n_util_mac.h" | |
| 22 | |
| 23 class ProfileMenuControllerTest : public CocoaProfileTest { | |
| 24 public: | |
| 25 ProfileMenuControllerTest() { | |
| 26 item_.reset([[NSMenuItem alloc] initWithTitle:@"Users" | |
| 27 action:nil | |
| 28 keyEquivalent:@""]); | |
| 29 controller_.reset( | |
| 30 [[ProfileMenuController alloc] initWithMainMenuItem:item_]); | |
| 31 } | |
| 32 | |
| 33 virtual void SetUp() { | |
| 34 CocoaProfileTest::SetUp(); | |
| 35 ASSERT_TRUE(profile()); | |
| 36 | |
| 37 // Spin the runloop so |-initializeMenu| gets called. | |
| 38 chrome::testing::NSRunLoopRunAllPending(); | |
| 39 } | |
| 40 | |
| 41 void TestBottomItems() { | |
| 42 NSMenu* menu = [controller() menu]; | |
| 43 NSInteger count = [menu numberOfItems]; | |
| 44 | |
| 45 ASSERT_GE(count, 4); | |
| 46 | |
| 47 NSMenuItem* item = [menu itemAtIndex:count - 4]; | |
| 48 EXPECT_TRUE([item isSeparatorItem]); | |
| 49 | |
| 50 item = [menu itemAtIndex:count - 3]; | |
| 51 EXPECT_EQ(@selector(editProfile:), [item action]); | |
| 52 | |
| 53 item = [menu itemAtIndex:count - 2]; | |
| 54 EXPECT_TRUE([item isSeparatorItem]); | |
| 55 | |
| 56 item = [menu itemAtIndex:count - 1]; | |
| 57 EXPECT_EQ(@selector(newProfile:), [item action]); | |
| 58 } | |
| 59 | |
| 60 void VerifyProfileNamedIsActive(NSString* title, int line) { | |
| 61 for (NSMenuItem* item in [[controller() menu] itemArray]) { | |
| 62 if ([[item title] isEqualToString:title]) { | |
| 63 EXPECT_EQ(NSOnState, [item state]) << [[item title] UTF8String] | |
| 64 << " (from line " << line << ")"; | |
| 65 } else { | |
| 66 EXPECT_EQ(NSOffState, [item state]) << [[item title] UTF8String] | |
| 67 << " (from line " << line << ")"; | |
| 68 } | |
| 69 } | |
| 70 } | |
| 71 | |
| 72 ProfileMenuController* controller() { return controller_.get(); } | |
| 73 | |
| 74 NSMenuItem* menu_item() { return item_.get(); } | |
| 75 | |
| 76 private: | |
| 77 base::scoped_nsobject<NSMenuItem> item_; | |
| 78 base::scoped_nsobject<ProfileMenuController> controller_; | |
| 79 }; | |
| 80 | |
| 81 TEST_F(ProfileMenuControllerTest, InitializeMenu) { | |
| 82 NSMenu* menu = [controller() menu]; | |
| 83 // <sep>, Edit, <sep>, New. | |
| 84 ASSERT_EQ(4, [menu numberOfItems]); | |
| 85 | |
| 86 TestBottomItems(); | |
| 87 | |
| 88 EXPECT_TRUE([menu_item() isHidden]); | |
| 89 } | |
| 90 | |
| 91 TEST_F(ProfileMenuControllerTest, CreateItemWithTitle) { | |
| 92 NSMenuItem* item = | |
| 93 [controller() createItemWithTitle:@"Title" | |
| 94 action:@selector(someSelector:)]; | |
| 95 EXPECT_NSEQ(@"Title", [item title]); | |
| 96 EXPECT_EQ(controller(), [item target]); | |
| 97 EXPECT_EQ(@selector(someSelector:), [item action]); | |
| 98 EXPECT_NSEQ(@"", [item keyEquivalent]); | |
| 99 } | |
| 100 | |
| 101 TEST_F(ProfileMenuControllerTest, RebuildMenu) { | |
| 102 NSMenu* menu = [controller() menu]; | |
| 103 EXPECT_EQ(4, [menu numberOfItems]); | |
| 104 | |
| 105 EXPECT_TRUE([menu_item() isHidden]); | |
| 106 | |
| 107 // Create some more profiles on the manager. | |
| 108 TestingProfileManager* manager = testing_profile_manager(); | |
| 109 manager->CreateTestingProfile("Profile 2"); | |
| 110 manager->CreateTestingProfile("Profile 3"); | |
| 111 | |
| 112 // Verify that the menu got rebuilt. | |
| 113 ASSERT_EQ(7, [menu numberOfItems]); | |
| 114 | |
| 115 NSMenuItem* item = [menu itemAtIndex:0]; | |
| 116 EXPECT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 117 | |
| 118 item = [menu itemAtIndex:1]; | |
| 119 EXPECT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 120 | |
| 121 item = [menu itemAtIndex:2]; | |
| 122 EXPECT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 123 | |
| 124 TestBottomItems(); | |
| 125 | |
| 126 EXPECT_FALSE([menu_item() isHidden]); | |
| 127 } | |
| 128 | |
| 129 TEST_F(ProfileMenuControllerTest, InsertItems) { | |
| 130 base::scoped_nsobject<NSMenu> menu([[NSMenu alloc] initWithTitle:@""]); | |
| 131 ASSERT_EQ(0, [menu numberOfItems]); | |
| 132 | |
| 133 // With only one profile, insertItems should be a no-op. | |
| 134 BOOL result = [controller() insertItemsIntoMenu:menu | |
| 135 atOffset:0 | |
| 136 fromDock:NO]; | |
| 137 EXPECT_FALSE(result); | |
| 138 EXPECT_EQ(0, [menu numberOfItems]); | |
| 139 [menu removeAllItems]; | |
| 140 | |
| 141 // Same for use in building the dock menu. | |
| 142 result = [controller() insertItemsIntoMenu:menu | |
| 143 atOffset:0 | |
| 144 fromDock:YES]; | |
| 145 EXPECT_FALSE(result); | |
| 146 EXPECT_EQ(0, [menu numberOfItems]); | |
| 147 [menu removeAllItems]; | |
| 148 | |
| 149 // Create one more profile on the manager. | |
| 150 TestingProfileManager* manager = testing_profile_manager(); | |
| 151 manager->CreateTestingProfile("Profile 2"); | |
| 152 | |
| 153 // With more than one profile, insertItems should return YES. | |
| 154 result = [controller() insertItemsIntoMenu:menu | |
| 155 atOffset:0 | |
| 156 fromDock:NO]; | |
| 157 EXPECT_TRUE(result); | |
| 158 ASSERT_EQ(2, [menu numberOfItems]); | |
| 159 | |
| 160 NSMenuItem* item = [menu itemAtIndex:0]; | |
| 161 EXPECT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 162 | |
| 163 item = [menu itemAtIndex:1]; | |
| 164 EXPECT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 165 [menu removeAllItems]; | |
| 166 | |
| 167 // And for the dock, the selector should be different and there should be a | |
| 168 // header item. | |
| 169 result = [controller() insertItemsIntoMenu:menu | |
| 170 atOffset:0 | |
| 171 fromDock:YES]; | |
| 172 EXPECT_TRUE(result); | |
| 173 ASSERT_EQ(3, [menu numberOfItems]); | |
| 174 | |
| 175 // First item is a label item. | |
| 176 item = [menu itemAtIndex:0]; | |
| 177 EXPECT_FALSE([item isEnabled]); | |
| 178 | |
| 179 item = [menu itemAtIndex:1]; | |
| 180 EXPECT_EQ(@selector(switchToProfileFromDock:), [item action]); | |
| 181 | |
| 182 item = [menu itemAtIndex:2]; | |
| 183 EXPECT_EQ(@selector(switchToProfileFromDock:), [item action]); | |
| 184 } | |
| 185 | |
| 186 TEST_F(ProfileMenuControllerTest, InitialActiveBrowser) { | |
| 187 [controller() activeBrowserChangedTo:NULL]; | |
| 188 VerifyProfileNamedIsActive(l10n_util::GetNSString(IDS_DEFAULT_PROFILE_NAME), | |
| 189 __LINE__); | |
| 190 } | |
| 191 | |
| 192 // Note: BrowserList::SetLastActive() is typically called as part of | |
| 193 // BrowserWindow::Show() and when a Browser becomes active. We don't need a full | |
| 194 // BrowserWindow, so it is called manually. | |
| 195 TEST_F(ProfileMenuControllerTest, SetActiveAndRemove) { | |
| 196 NSMenu* menu = [controller() menu]; | |
| 197 TestingProfileManager* manager = testing_profile_manager(); | |
| 198 TestingProfile* profile2 = manager->CreateTestingProfile("Profile 2"); | |
| 199 TestingProfile* profile3 = manager->CreateTestingProfile("Profile 3"); | |
| 200 ASSERT_EQ(7, [menu numberOfItems]); | |
| 201 | |
| 202 // Create a browser and "show" it. | |
| 203 Browser::CreateParams profile2_params(profile2, chrome::GetActiveDesktop()); | |
| 204 scoped_ptr<Browser> p2_browser( | |
| 205 chrome::CreateBrowserWithTestWindowForParams(&profile2_params)); | |
| 206 BrowserList::SetLastActive(p2_browser.get()); | |
| 207 VerifyProfileNamedIsActive(@"Profile 2", __LINE__); | |
| 208 | |
| 209 // Close the browser and make sure it's still active. | |
| 210 p2_browser.reset(); | |
| 211 VerifyProfileNamedIsActive(@"Profile 2", __LINE__); | |
| 212 | |
| 213 // Open a new browser and make sure it takes effect. | |
| 214 Browser::CreateParams profile3_params(profile3, chrome::GetActiveDesktop()); | |
| 215 scoped_ptr<Browser> p3_browser( | |
| 216 chrome::CreateBrowserWithTestWindowForParams(&profile3_params)); | |
| 217 BrowserList::SetLastActive(p3_browser.get()); | |
| 218 VerifyProfileNamedIsActive(@"Profile 3", __LINE__); | |
| 219 | |
| 220 p3_browser.reset(); | |
| 221 VerifyProfileNamedIsActive(@"Profile 3", __LINE__); | |
| 222 } | |
| 223 | |
| 224 TEST_F(ProfileMenuControllerTest, DeleteActiveProfile) { | |
| 225 TestingProfileManager* manager = testing_profile_manager(); | |
| 226 | |
| 227 manager->CreateTestingProfile("Profile 2"); | |
| 228 TestingProfile* profile3 = manager->CreateTestingProfile("Profile 3"); | |
| 229 ASSERT_EQ(3U, manager->profile_manager()->GetNumberOfProfiles()); | |
| 230 | |
| 231 const base::FilePath profile3_path = profile3->GetPath(); | |
| 232 manager->DeleteTestingProfile("Profile 3"); | |
| 233 | |
| 234 // Simulate an unloaded profile by setting the "last used" local state pref | |
| 235 // the profile that was just deleted. | |
| 236 PrefService* local_state = g_browser_process->local_state(); | |
| 237 local_state->SetString(prefs::kProfileLastUsed, | |
| 238 profile3_path.BaseName().MaybeAsASCII()); | |
| 239 | |
| 240 // Simulate the active browser changing to NULL and ensure a profile doesn't | |
| 241 // get created by disallowing IO operations temporarily. | |
| 242 const bool io_was_allowed = base::ThreadRestrictions::SetIOAllowed(false); | |
| 243 [controller() activeBrowserChangedTo:NULL]; | |
| 244 base::ThreadRestrictions::SetIOAllowed(io_was_allowed); | |
| 245 } | |
| 246 | |
| 247 TEST_F(ProfileMenuControllerTest, ManagedProfile) { | |
| 248 TestingProfileManager* manager = testing_profile_manager(); | |
| 249 TestingProfile* managed_profile = | |
| 250 manager->CreateTestingProfile("test1", | |
| 251 scoped_ptr<PrefServiceSyncable>(), | |
| 252 base::ASCIIToUTF16("Supervised User"), | |
| 253 0, | |
| 254 "TEST_ID", | |
| 255 TestingProfile::TestingFactories()); | |
| 256 BrowserList::SetLastActive(browser()); | |
| 257 | |
| 258 NSMenu* menu = [controller() menu]; | |
| 259 ASSERT_EQ(6, [menu numberOfItems]); | |
| 260 NSMenuItem* item = [menu itemAtIndex:0]; | |
| 261 ASSERT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 262 EXPECT_TRUE([controller() validateMenuItem:item]); | |
| 263 | |
| 264 item = [menu itemAtIndex:1]; | |
| 265 ASSERT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 266 EXPECT_TRUE([controller() validateMenuItem:item]); | |
| 267 | |
| 268 item = [menu itemAtIndex:5]; | |
| 269 ASSERT_EQ(@selector(newProfile:), [item action]); | |
| 270 EXPECT_TRUE([controller() validateMenuItem:item]); | |
| 271 | |
| 272 // Open a new browser for the managed user and switch to it. | |
| 273 Browser::CreateParams managed_profile_params( | |
| 274 managed_profile, chrome::HOST_DESKTOP_TYPE_NATIVE); | |
| 275 scoped_ptr<Browser> managed_browser( | |
| 276 chrome::CreateBrowserWithTestWindowForParams(&managed_profile_params)); | |
| 277 BrowserList::SetLastActive(managed_browser.get()); | |
| 278 | |
| 279 item = [menu itemAtIndex:0]; | |
| 280 ASSERT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 281 EXPECT_FALSE([controller() validateMenuItem:item]); | |
| 282 | |
| 283 item = [menu itemAtIndex:1]; | |
| 284 ASSERT_EQ(@selector(switchToProfileFromMenu:), [item action]); | |
| 285 EXPECT_TRUE([controller() validateMenuItem:item]); | |
| 286 | |
| 287 item = [menu itemAtIndex:5]; | |
| 288 ASSERT_EQ(@selector(newProfile:), [item action]); | |
| 289 EXPECT_FALSE([controller() validateMenuItem:item]); | |
| 290 } | |
| OLD | NEW |