Implement Non-Closable Notification

ui/message_center/views/message_center_view_unittest.cc

 // Copyright 2013 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/message_center_view.h"
 
 #include <map>
 #include <utility>
 
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/utf_string_conversions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/message_center/fake_message_center.h"
 #include "ui/message_center/message_center_style.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_button_bar.h"
 #include "ui/message_center/views/message_center_controller.h"
 #include "ui/message_center/views/message_list_view.h"
 #include "ui/message_center/views/notification_view.h"
+#include "ui/views/controls/slide_out_view.h"
 
 namespace message_center {
 
 static const char* kNotificationId1 = "notification id 1";
 static const char* kNotificationId2 = "notification id 2";
 
 /* Types **********************************************************************/
 
 enum CallType {
   GET_PREFERRED_SIZE,
   GET_HEIGHT_FOR_WIDTH,
   LAYOUT
 };
 
+class DummyEvent : public ui::Event {
+ public:
+  DummyEvent() : Event(ui::ET_UNKNOWN, base::TimeDelta(), 0) {}
+  ~DummyEvent() override {}
+};
+
 /* Instrumented/Mock NotificationView subclass ********************************/
 
 class MockNotificationView : public NotificationView {
  public:
   class Test {
    public:
     virtual void RegisterCall(CallType type) = 0;
   };
 
   explicit MockNotificationView(MessageCenterController* controller,
@@ skipping 40 matching lines @@
 }
 
 class FakeMessageCenterImpl : public FakeMessageCenter {
  public:
   const NotificationList::Notifications& GetVisibleNotifications() override {
     return visible_notifications_;
   }
   void SetVisibleNotifications(NotificationList::Notifications notifications) {
     visible_notifications_ = notifications;
   }
+  void RemoveAllNotifications(bool by_user, RemoveType type) override {
+    if (type == RemoveType::NON_PINNED)
+      remove_all_closable_notification_called_ = true;
+  }
+  bool remove_all_closable_notification_called_ = false;
   NotificationList::Notifications visible_notifications_;
 };
 
 /* Test fixture ***************************************************************/
 
 class MessageCenterViewTest : public testing::Test,
                               public MockNotificationView::Test,
                               public MessageCenterController {
  public:
   MessageCenterViewTest();
   ~MessageCenterViewTest() override;
 
   void SetUp() override;
   void TearDown() override;
 
   MessageCenterView* GetMessageCenterView();
   MessageListView* GetMessageListView();
+  FakeMessageCenterImpl* GetMessageCenter() const;
   NotificationView* GetNotificationView(const std::string& id);
   views::BoundsAnimator* GetAnimator();
   int GetNotificationCount();
   int GetCallCount(CallType type);
   int GetCalculatedMessageListViewHeight();
   void AddNotification(scoped_ptr<Notification> notification);
   void UpdateNotification(const std::string& notification_id,
                           scoped_ptr<Notification> notification);
 
  // Overridden from MessageCenterController:
   void ClickOnNotification(const std::string& notification_id) override;
   void RemoveNotification(const std::string& notification_id,
                           bool by_user) override;
   scoped_ptr<ui::MenuModel> CreateMenuModel(
       const NotifierId& notifier_id,
       const base::string16& display_source) override;
   bool HasClickedListener(const std::string& notification_id) override;
   void ClickOnNotificationButton(const std::string& notification_id,
                                  int button_index) override;
   void ClickOnSettingsButton(const std::string& notification_id) override;
 
   // Overridden from MockNotificationView::Test
   void RegisterCall(CallType type) override;
 
   void FireOnMouseExitedEvent();
 
   void LogBounds(int depth, views::View* view);
 
+  MessageCenterButtonBar* GetButtonBar() const;
+
  private:
   views::View* MakeParent(views::View* child1, views::View* child2);
 
   base::MessageLoopForUI message_loop_;
 
   NotificationList::Notifications notifications_;
   scoped_ptr<MessageCenterView> message_center_view_;
-  FakeMessageCenterImpl message_center_;
+  scoped_ptr<FakeMessageCenterImpl> message_center_;
   std::map<CallType,int> callCounts_;
 
   DISALLOW_COPY_AND_ASSIGN(MessageCenterViewTest);
 };
 
 MessageCenterViewTest::MessageCenterViewTest() {
 }
 
 MessageCenterViewTest::~MessageCenterViewTest() {
 }
 
 void MessageCenterViewTest::SetUp() {
+  message_center_.reset(new FakeMessageCenterImpl());
+
   // Create a dummy notification.
   Notification* notification1 = new Notification(
       NOTIFICATION_TYPE_SIMPLE, std::string(kNotificationId1),
       base::UTF8ToUTF16("title"), base::UTF8ToUTF16("message1"), gfx::Image(),
       base::UTF8ToUTF16("display source"), GURL(),
       NotifierId(NotifierId::APPLICATION, "extension_id"),
       message_center::RichNotificationData(), NULL);
 
   Notification* notification2 = new Notification(
       NOTIFICATION_TYPE_SIMPLE, std::string(kNotificationId2),
       base::UTF8ToUTF16("title2"), base::UTF8ToUTF16("message2"), gfx::Image(),
       base::UTF8ToUTF16("display source"), GURL(),
       NotifierId(NotifierId::APPLICATION, "extension_id"),
       message_center::RichNotificationData(), NULL);
 
   // ...and a list for it.
   notifications_.insert(notification1);
   notifications_.insert(notification2);
-  message_center_.SetVisibleNotifications(notifications_);
+  message_center_->SetVisibleNotifications(notifications_);
 
   // Then create a new MessageCenterView with that single notification.
   base::string16 title;
   message_center_view_.reset(new MessageCenterView(
-      &message_center_, NULL, 100, false, /*top_down =*/false, title));
+      message_center_.get(), NULL, 100, false, /*top_down =*/false, title));
   GetMessageListView()->quit_message_loop_after_animation_for_test_ = true;
   GetMessageCenterView()->SetBounds(0, 0, 380, 600);
   message_center_view_->SetNotifications(notifications_);
 
   // Wait until the animation finishes if available.
   if (GetAnimator()->IsAnimating())
     base::MessageLoop::current()->Run();
 }
 
 void MessageCenterViewTest::TearDown() {
   message_center_view_.reset();
   STLDeleteElements(&notifications_);
 }
 
 MessageCenterView* MessageCenterViewTest::GetMessageCenterView() {
   return message_center_view_.get();
 }
 
 MessageListView* MessageCenterViewTest::GetMessageListView() {
   return message_center_view_->message_list_view_.get();
 }
 
+FakeMessageCenterImpl* MessageCenterViewTest::GetMessageCenter() const {
+  return message_center_.get();
+}
+
 NotificationView* MessageCenterViewTest::GetNotificationView(
     const std::string& id) {
   return message_center_view_->notification_views_[id];
 }
 
 int MessageCenterViewTest::GetCalculatedMessageListViewHeight() {
   return GetMessageListView()->GetHeightForWidth(GetMessageListView()->width());
 }
 
 views::BoundsAnimator* MessageCenterViewTest::GetAnimator() {
@@ skipping 11 matching lines @@
 void MessageCenterViewTest::ClickOnNotification(
     const std::string& notification_id) {
   // For this test, this method should not be invoked.
   NOTREACHED();
 }
 
 void MessageCenterViewTest::AddNotification(
     scoped_ptr<Notification> notification) {
   std::string notification_id = notification->id();
   notifications_.insert(notification.release());
-  message_center_.SetVisibleNotifications(notifications_);
+  message_center_->SetVisibleNotifications(notifications_);
   message_center_view_->OnNotificationAdded(notification_id);
 }
 
 void MessageCenterViewTest::UpdateNotification(
     const std::string& notification_id,
     scoped_ptr<Notification> notification) {
   for (auto it = notifications_.begin(); it != notifications_.end(); it++) {
     if ((*it)->id() == notification_id) {
       delete *it;
       notifications_.erase(it);
       break;
     }
   }
   // |notifications| is a "set" container so we don't need to be aware the
   // order.
   notifications_.insert(notification.release());
-  message_center_.SetVisibleNotifications(notifications_);
+  message_center_->SetVisibleNotifications(notifications_);
   message_center_view_->OnNotificationUpdated(notification_id);
 }
 
 void MessageCenterViewTest::RemoveNotification(
     const std::string& notification_id,
     bool by_user) {
   for (auto it = notifications_.begin(); it != notifications_.end(); it++) {
     if ((*it)->id() == notification_id) {
       delete *it;
       notifications_.erase(it);
       break;
     }
   }
-  message_center_.SetVisibleNotifications(notifications_);
+  message_center_->SetVisibleNotifications(notifications_);
   message_center_view_->OnNotificationRemoved(notification_id, by_user);
 }
 
 scoped_ptr<ui::MenuModel> MessageCenterViewTest::CreateMenuModel(
     const NotifierId& notifier_id,
     const base::string16& display_source) {
   // For this test, this method should not be invoked.
   NOTREACHED();
   return nullptr;
 }
@@ skipping 34 matching lines @@
   base::string16 inset;
   for (int i = 0; i < depth; ++i)
     inset.append(base::UTF8ToUTF16("  "));
   gfx::Rect bounds = view->bounds();
   DVLOG(0) << inset << bounds.width() << " x " << bounds.height()
            << " @ " << bounds.x() << ", " << bounds.y();
   for (int i = 0; i < view->child_count(); ++i)
     LogBounds(depth + 1, view->child_at(i));
 }
 
+MessageCenterButtonBar* MessageCenterViewTest::GetButtonBar() const {
+  return message_center_view_->button_bar_;
+}
+
 /* Unit tests *****************************************************************/
 
 TEST_F(MessageCenterViewTest, CallTest) {
   // Verify that this didn't generate more than 2 Layout() call per descendant
   // NotificationView or more than a total of 20 GetPreferredSize() and
   // GetHeightForWidth() calls per descendant NotificationView. 20 is a very
   // large number corresponding to the current reality. That number will be
   // ratcheted down over time as the code improves.
   EXPECT_LE(GetCallCount(LAYOUT), GetNotificationCount() * 2);
   EXPECT_LE(GetCallCount(GET_PREFERRED_SIZE) +
@@ skipping 146 matching lines @@
   // Mouse cursor moves out of the message center. This resets the reposition
   // target in the message list.
   FireOnMouseExitedEvent();
 
   // The height should shrink from the height of the removed notification 2.
   EXPECT_EQ(previous_height - previous_notification2_height -
                 (kMarginBetweenItems - MessageView::GetShadowInsets().bottom()),
             GetMessageListView()->height());
 }
 
+TEST_F(MessageCenterViewTest, CloseButton) {
+  views::Button* close_button = GetButtonBar()->GetCloseAllButtonForTest();
+  EXPECT_NE(nullptr, close_button);
+
+  ((views::ButtonListener*)GetButtonBar())
+      ->ButtonPressed(close_button, DummyEvent());
+  base::MessageLoop::current()->Run();
+  EXPECT_TRUE(GetMessageCenter()->remove_all_closable_notification_called_);
+}
+
+TEST_F(MessageCenterViewTest, CloseButtonEnablity) {
+  views::Button* close_button = GetButtonBar()->GetCloseAllButtonForTest();
+  EXPECT_NE(nullptr, close_button);
+
+  // There should be 2 non-pinned notifications.
+  EXPECT_EQ(2u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_TRUE(close_button->enabled());
+
+  RemoveNotification(kNotificationId1, false);
+  base::MessageLoop::current()->Run();
+
+  // There should be 1 non-pinned notification.
+  EXPECT_EQ(1u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_TRUE(close_button->enabled());
+
+  RemoveNotification(kNotificationId2, false);
+  base::MessageLoop::current()->Run();
+
+  // There should be no notification.
+  EXPECT_EQ(0u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_FALSE(close_button->enabled());
+
+  Notification normal_notification(
+      NOTIFICATION_TYPE_SIMPLE, std::string(kNotificationId1),
+      base::UTF8ToUTF16("title2"),
+      base::UTF8ToUTF16("message\nwhich\nis\nvertically\nlong\n."),
+      gfx::Image(), base::UTF8ToUTF16("display source"), GURL(),
+      NotifierId(NotifierId::APPLICATION, "extension_id"),
+      message_center::RichNotificationData(), NULL);
+
+#if defined(OS_CHROMEOS)
+  Notification pinned_notification(
+      NOTIFICATION_TYPE_SIMPLE, std::string(kNotificationId2),
+      base::UTF8ToUTF16("title2"),
+      base::UTF8ToUTF16("message\nwhich\nis\nvertically\nlong\n."),
+      gfx::Image(), base::UTF8ToUTF16("display source"), GURL(),
+      NotifierId(NotifierId::APPLICATION, "extension_id"),
+      message_center::RichNotificationData(), NULL);
+  pinned_notification.set_pinned(true);
+
+  AddNotification(
+      scoped_ptr<Notification>(new Notification(normal_notification)));
+
+  // There should be 1 non-pinned notification.
+  EXPECT_EQ(1u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_TRUE(close_button->enabled());
+
+  AddNotification(
+      scoped_ptr<Notification>(new Notification(pinned_notification)));
+
+  // There should be 1 normal notification and 1 pinned notification.
+  EXPECT_EQ(2u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_TRUE(close_button->enabled());
+
+  RemoveNotification(kNotificationId1, false);
+
+  // There should be 1 pinned notification.
+  EXPECT_EQ(1u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_FALSE(close_button->enabled());
+
+  RemoveNotification(kNotificationId2, false);
+
+  // There should be no notification.
+  EXPECT_EQ(0u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_FALSE(close_button->enabled());
+
+  AddNotification(
+      scoped_ptr<Notification>(new Notification(pinned_notification)));
+
+  // There should be 1 pinned notification.
+  EXPECT_EQ(1u, GetMessageCenter()->GetVisibleNotifications().size());
+  EXPECT_FALSE(close_button->enabled());
+#endif  // defined(OS_CHROMEOS)
+}
+
 }  // namespace message_center