Notifications: Retain button hover state during content updates.

ui/message_center/views/notification_view_unittest.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 "ui/message_center/views/notification_view.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/utf_string_conversions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/image/image.h"
 #include "ui/message_center/notification.h"
 #include "ui/message_center/notification_list.h"
 #include "ui/message_center/notification_types.h"
 #include "ui/message_center/views/message_center_controller.h"
+#include "ui/message_center/views/notification_button.h"
 
 namespace message_center {
 
 /* Test fixture ***************************************************************/
 
 class NotificationViewTest : public testing::Test,
                              public MessageCenterController {
  public:
   NotificationViewTest();
   virtual ~NotificationViewTest();
@@ skipping 22 matching lines @@
   }
 
   const SkBitmap CreateBitmap(int width, int height) {
     SkBitmap bitmap;
     bitmap.setConfig(SkBitmap::kARGB_8888_Config, width, height);
     bitmap.allocPixels();
     bitmap.eraseRGB(0, 255, 0);
     return bitmap;
   }
 
+  std::vector<ButtonInfo> CreateButtons(int number) {
+    ButtonInfo info(base::ASCIIToUTF16("Test button title."));
+    info.icon = CreateTestImage(4, 4);
+    return std::vector<ButtonInfo>(number, info);
+  }
+
  private:
   scoped_ptr<RichNotificationData> data_;
   scoped_ptr<Notification> notification_;
   scoped_ptr<NotificationView> notification_view_;
 
   DISALLOW_COPY_AND_ASSIGN(NotificationViewTest);
 };
 
 NotificationViewTest::NotificationViewTest() {
 }
@@ skipping 100 matching lines @@
   notification()->set_context_message(base::UTF8ToUTF16("foo"));
   notification_view()->CreateOrUpdateViews(*notification());
 
   EXPECT_TRUE(notification_view()->context_message_view_ != NULL);
 
   EXPECT_EQ(1, notification_view()->GetMessageLineLimit(0, 360));
   EXPECT_EQ(1, notification_view()->GetMessageLineLimit(1, 360));
   EXPECT_EQ(0, notification_view()->GetMessageLineLimit(2, 360));
 }
 
+TEST_F(NotificationViewTest, UpdateButtonsStateTest) {
+  notification()->set_buttons(CreateButtons(2));
+  notification_view()->CreateOrUpdateViews(*notification());
+  notification_view()->action_buttons_[0]->set_animate_on_state_change(false);
+  notification_view()->action_buttons_[1]->set_animate_on_state_change(false);
+
+  EXPECT_TRUE(NULL == notification_view()->action_buttons_[0]->background());
+
+  notification_view()->action_buttons_[0]->SetState(
+      views::Button::STATE_HOVERED);
+
+  EXPECT_TRUE(NULL != notification_view()->action_buttons_[0]->background());
+
+  notification_view()->CreateOrUpdateViews(*notification());
+  notification_view()->action_buttons_[0]->set_animate_on_state_change(false);
+  notification_view()->action_buttons_[1]->set_animate_on_state_change(false);
+
+  EXPECT_TRUE(NULL != notification_view()->action_buttons_[0]->background());
+
+  notification_view()->action_buttons_[0]->SetState(
+      views::Button::STATE_NORMAL);
+
+  notification_view()->CreateOrUpdateViews(*notification());
+
+  EXPECT_TRUE(NULL == notification_view()->action_buttons_[0]->background());
+
+}
+
 }  // namespace message_center