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]); |
+} |