[Mac] Redesign of the avatar menu button

chrome/browser/ui/cocoa/browser/new_avatar_button_controller_unittest.mm

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#import "chrome/browser/ui/cocoa/browser/new_avatar_button_controller.h"
+
+#include "base/command_line.h"
+#include "base/mac/scoped_nsobject.h"
+#include "base/strings/sys_string_conversions.h"
+#include "chrome/browser/profiles/profiles_state.h"
+#include "chrome/browser/profiles/profile_manager.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_window.h"
+#import "chrome/browser/ui/cocoa/browser/profile_chooser_controller.h"
+#include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
+#include "chrome/browser/ui/cocoa/info_bubble_window.h"
+#include "chrome/common/chrome_switches.h"
+
+static const char kDefaultProfileName[] = "default";
+
+class NewAvatarButtonControllerTest : public CocoaProfileTest {
+ public:
+  virtual void SetUp() OVERRIDE {
+    CommandLine::ForCurrentProcess()->AppendSwitch(
+        switches::kNewProfileManagement);
+    DCHECK(CommandLine::ForCurrentProcess()->HasSwitch(
+        switches::kNewProfileManagement));
+    DCHECK(profiles::IsMultipleProfilesEnabled());
+
+    CocoaProfileTest::SetUp();
+    ASSERT_TRUE(browser());
+
+    controller_.reset(
+        [[NewAvatarButtonController alloc] initWithBrowser:browser()]);
+  }
+
+  virtual void TearDown() OVERRIDE {
+    browser()->window()->Close();
+    CocoaProfileTest::TearDown();
+  }
+
+  NSButton* button() { return [controller_ buttonView]; }
+
+  NSView* view() { return [controller_ view]; }
+
+  NewAvatarButtonController* controller() { return controller_.get(); }
+
+ private:
+  base::scoped_nsobject<NewAvatarButtonController> controller_;
+};
+
+TEST_F(NewAvatarButtonControllerTest, ButtonShown) {
+  EXPECT_FALSE([view() isHidden]);
+  EXPECT_EQ(kDefaultProfileName, base::SysNSStringToUTF8([button() title]));
+}
+
+TEST_F(NewAvatarButtonControllerTest, DoubleOpen) {
+  EXPECT_FALSE([controller() menuController]);
+
+  [button() performClick:button()];
+
+  ProfileChooserController* menu = [controller() menuController];
+  EXPECT_TRUE(menu);
+
+  [button() performClick:button()];
+  EXPECT_EQ(menu, [controller() menuController]);
+
+  // Do not animate out because that is hard to test around.
+  static_cast<InfoBubbleWindow*>(menu.window).allowedAnimations =
+      info_bubble::kAnimateNone;
+  [menu close];
+  EXPECT_FALSE([controller() menuController]);
+}