Repaint ImageButton when image is changed.

ui/views/controls/button/image_button.cc

 // Copyright (c) 2012 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 "ui/views/controls/button/image_button.h"
 
 #include <utility>
 
 #include "base/strings/utf_string_conversions.h"
 #include "ui/accessibility/ax_view_state.h"
@@ skipping 29 matching lines @@
   EnableCanvasFlippingForRTLUI(true);
 }
 
 ImageButton::~ImageButton() {
 }
 
 const gfx::ImageSkia& ImageButton::GetImage(ButtonState state) const {
   return images_[state];
 }
 
-void ImageButton::SetImage(ButtonState state, const gfx::ImageSkia* image) {
-  images_[state] = image ? *image : gfx::ImageSkia();
+void ImageButton::SetImage(ButtonState for_state, const gfx::ImageSkia* image) {
+  images_[for_state] = image ? *image : gfx::ImageSkia();
   PreferredSizeChanged();
+  if (state() == for_state)
+    SchedulePaint();
 }
 
 void ImageButton::SetBackground(SkColor color,
                                 const gfx::ImageSkia* image,
                                 const gfx::ImageSkia* mask) {
   if (image == NULL || mask == NULL) {
     background_image_ = gfx::ImageSkia();
     return;
   }
 
@@ skipping 201 matching lines @@
   *tooltip = toggled_tooltip_text_;
   return true;
 }
 
 void ToggleImageButton::GetAccessibleState(ui::AXViewState* state) {
   ImageButton::GetAccessibleState(state);
   GetTooltipText(gfx::Point(), &state->name);
 }
 
 }  // namespace views