Refactor the avatar button/icon class

chrome/browser/ui/views/profiles/avatar_menu_button.cc

 // 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.
 
 #include "chrome/browser/ui/views/profiles/avatar_menu_button.h"
 
 #include "base/command_line.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
@@ skipping 17 matching lines @@
 #include "ui/views/widget/widget.h"
 
 static inline int Round(double x) {
   return static_cast<int>(x + 0.5);
 }
 
 // static
 const char AvatarMenuButton::kViewClassName[] = "AvatarMenuButton";
 
 AvatarMenuButton::AvatarMenuButton(Browser* browser, bool disabled)
-    : MenuButton(NULL, base::string16(), this, false),
-      browser_(browser),
+    : AvatarBaseController(browser),
+      MenuButton(NULL, base::string16(), this, false),
       disabled_(disabled),
       is_rectangle_(false),
       old_height_(0),
       button_on_right_(false) {
   // In RTL mode, the avatar icon should be looking the opposite direction.
   EnableCanvasFlippingForRTLUI(true);
 
   SetEventTargeter(
       scoped_ptr<views::ViewTargeter>(new views::ViewTargeter(this)));
+
+  Update();
 }
 
 AvatarMenuButton::~AvatarMenuButton() {
 }
 
 const char* AvatarMenuButton::GetClassName() const {
   return kViewClassName;
 }
 
 void AvatarMenuButton::OnPaint(gfx::Canvas* canvas) {
@@ skipping 78 matching lines @@
 #endif
     } else {
       AvatarMenu::GetImageForMenuButton(profile->GetPath(),
                                         avatar,
                                         is_rectangle);
     }
   }
   return true;
 }
 
+void AvatarMenuButton::Update() {
+  // Check the validity of browser() so that this function can be bypassed in

[noms (inactive), 2015/03/26 15:26:56]

nit: maybe just "The browser can be null in tests"? Also, just to double check,
is this still true, after the TestingManager fix?

[yao, 2015/03/26 18:45:01]

Yes, because nullptr is passsed when the button is created. See:
https://code.google.com/p/chromium/codesearch#chromium/src/chrome/browser/ui/...

+  // unit tests.
+  if (!browser())
+    return;
+
+  gfx::Image avatar;
+  gfx::Image taskbar_badge_avatar;
+  bool is_rectangle = false;
+
+  // Update the avatar button in the window frame and the taskbar overlay.
+  bool should_show_avatar_menu = AvatarMenu::ShouldShowAvatarMenu();
+
+  if (!AvatarMenuButton::GetAvatarImages(
+          browser()->profile(), should_show_avatar_menu, &avatar,
+          &taskbar_badge_avatar, &is_rectangle)) {
+    return;
+  }
+
+  // Disable the menu when we should not show the menu.
+  if (!AvatarMenu::ShouldShowAvatarMenu())
+    this->SetEnabled(false);
+  this->SetAvatarIcon(avatar, is_rectangle);
+}
+
 // views::ViewTargeterDelegate:
 bool AvatarMenuButton::DoesIntersectRect(const views::View* target,
                                          const gfx::Rect& rect) const {
   CHECK_EQ(target, this);
   return !disabled_ &&
          views::ViewTargeterDelegate::DoesIntersectRect(target, rect);
 }
 
 // views::MenuButtonListener implementation
 void AvatarMenuButton::OnMenuButtonClicked(views::View* source,
                                            const gfx::Point& point) {
   if (!disabled_)
-    chrome::ShowAvatarMenu(browser_);
+    chrome::ShowAvatarMenu(browser());
 }