| Index: chrome/browser/ui/cocoa/browser/avatar_button_controller_unittest.mm
|
| diff --git a/chrome/browser/ui/cocoa/browser/avatar_button_controller_unittest.mm b/chrome/browser/ui/cocoa/browser/avatar_button_controller_unittest.mm
|
| index 6f59c6f726d884714bfefdd86cc945706f564737..42f7b659b9f586ae67193b7fed721f9b8d2cf208 100644
|
| --- a/chrome/browser/ui/cocoa/browser/avatar_button_controller_unittest.mm
|
| +++ b/chrome/browser/ui/cocoa/browser/avatar_button_controller_unittest.mm
|
| @@ -10,6 +10,7 @@
|
| #import "chrome/browser/ui/cocoa/browser/avatar_menu_bubble_controller.h"
|
| #include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
|
| #include "chrome/browser/ui/cocoa/info_bubble_window.h"
|
| +#include "chrome/common/pref_names.h"
|
|
|
| class AvatarButtonControllerTest : public CocoaProfileTest {
|
| public:
|
| @@ -29,6 +30,8 @@ class AvatarButtonControllerTest : public CocoaProfileTest {
|
|
|
| NSButton* button() { return [controller_ buttonView]; }
|
|
|
| + NSView* view() { return [controller_ view]; }
|
| +
|
| AvatarButtonController* controller() { return controller_.get(); }
|
|
|
| private:
|
| @@ -36,20 +39,20 @@ class AvatarButtonControllerTest : public CocoaProfileTest {
|
| };
|
|
|
| TEST_F(AvatarButtonControllerTest, AddRemoveProfiles) {
|
| - EXPECT_TRUE([button() isHidden]);
|
| + EXPECT_TRUE([view() isHidden]);
|
|
|
| testing_profile_manager()->CreateTestingProfile("one");
|
|
|
| - EXPECT_FALSE([button() isHidden]);
|
| + EXPECT_FALSE([view() isHidden]);
|
|
|
| testing_profile_manager()->CreateTestingProfile("two");
|
| - EXPECT_FALSE([button() isHidden]);
|
| + EXPECT_FALSE([view() isHidden]);
|
|
|
| testing_profile_manager()->DeleteTestingProfile("one");
|
| - EXPECT_FALSE([button() isHidden]);
|
| + EXPECT_FALSE([view() isHidden]);
|
|
|
| testing_profile_manager()->DeleteTestingProfile("two");
|
| - EXPECT_TRUE([button() isHidden]);
|
| + EXPECT_TRUE([view() isHidden]);
|
| }
|
|
|
| TEST_F(AvatarButtonControllerTest, DoubleOpen) {
|
| @@ -74,3 +77,20 @@ TEST_F(AvatarButtonControllerTest, DoubleOpen) {
|
|
|
| testing_profile_manager()->DeleteTestingProfile("p2");
|
| }
|
| +
|
| +TEST_F(AvatarButtonControllerTest, ManagedUserLabel) {
|
| + // Create a second profile to enable the avatar menu.
|
| + testing_profile_manager()->CreateTestingProfile("p2");
|
| +
|
| + EXPECT_FALSE([controller() labelView]);
|
| +
|
| + // Transform the first profile to a managed user profile.
|
| + profile()->GetPrefs()->SetBoolean(prefs::kProfileIsManaged, true);
|
| +
|
| + // Build a new controller to check if it is initialized correctly for a
|
| + // managed user profile.
|
| + scoped_nsobject<AvatarButtonController> controller(
|
| + [[AvatarButtonController alloc] initWithBrowser:browser()]);
|
| +
|
| + EXPECT_TRUE([controller labelView]);
|
| +}
|
|
|