ui/message_center/message_center_tray_unittest.cc

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "ui/message_center/message_center_tray.h"
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "testing/gtest/include/gtest/gtest.h"
   9 #include "ui/base/models/menu_model.h"
  10 #include "ui/message_center/message_center.h"
  11 #include "ui/message_center/notification.h"
  12 #include "ui/message_center/notification_types.h"
  13
  14 using base::ASCIIToUTF16;
  15
  16 namespace message_center {
  17 namespace {
  18
  19 class MockDelegate : public MessageCenterTrayDelegate {
  20  public:
  21   MockDelegate()
  22       : show_popups_success_(true),
  23         show_message_center_success_(true),
  24         enable_context_menu_(true) {}
  25   ~MockDelegate() override {}
  26   void OnMessageCenterTrayChanged() override {}
  27   bool ShowPopups() override { return show_message_center_success_; }
  28   void HidePopups() override {}
  29   bool ShowMessageCenter() override { return show_popups_success_; }
  30   void HideMessageCenter() override {}
  31   bool ShowNotifierSettings() override { return true; }
  32   bool IsContextMenuEnabled() const override { return enable_context_menu_; }
  33
  34   MessageCenterTray* GetMessageCenterTray() override { return NULL; }
  35
  36   bool show_popups_success_;
  37   bool show_message_center_success_;
  38   bool enable_context_menu_;
  39
  40  private:
  41   DISALLOW_COPY_AND_ASSIGN(MockDelegate);
  42 };
  43
  44 }  // namespace
  45
  46 class MessageCenterTrayTest : public testing::Test {
  47  public:
  48   MessageCenterTrayTest() {}
  49   ~MessageCenterTrayTest() override {}
  50
  51   void SetUp() override {
  52     MessageCenter::Initialize();
  53     delegate_.reset(new MockDelegate);
  54     message_center_ = MessageCenter::Get();
  55     message_center_tray_.reset(
  56         new MessageCenterTray(delegate_.get(), message_center_));
  57   }
  58
  59   void TearDown() override {
  60     message_center_tray_.reset();
  61     delegate_.reset();
  62     message_center_ = NULL;
  63     MessageCenter::Shutdown();
  64   }
  65
  66  protected:
  67   NotifierId DummyNotifierId() {
  68     return NotifierId();
  69   }
  70
  71   void AddNotification(const std::string& id) {
  72     scoped_ptr<Notification> notification(
  73         new Notification(message_center::NOTIFICATION_TYPE_SIMPLE,
  74                          id,
  75                          ASCIIToUTF16("Test Web Notification"),
  76                          ASCIIToUTF16("Notification message body."),
  77                          gfx::Image(),
  78                          ASCIIToUTF16("www.test.org"),
  79                          DummyNotifierId(),
  80                          message_center::RichNotificationData(),
  81                          NULL /* delegate */));
  82     message_center_->AddNotification(notification.Pass());
  83   }
  84   scoped_ptr<MockDelegate> delegate_;
  85   scoped_ptr<MessageCenterTray> message_center_tray_;
  86   MessageCenter* message_center_;
  87
  88  private:
  89   DISALLOW_COPY_AND_ASSIGN(MessageCenterTrayTest);
  90 };
  91
  92 TEST_F(MessageCenterTrayTest, BasicMessageCenter) {
  93   ASSERT_FALSE(message_center_tray_->popups_visible());
  94   ASSERT_FALSE(message_center_tray_->message_center_visible());
  95
  96   bool shown = message_center_tray_->ShowMessageCenterBubble();
  97   EXPECT_TRUE(shown);
  98
  99   ASSERT_FALSE(message_center_tray_->popups_visible());
 100   ASSERT_TRUE(message_center_tray_->message_center_visible());
 101
 102   message_center_tray_->HideMessageCenterBubble();
 103
 104   ASSERT_FALSE(message_center_tray_->popups_visible());
 105   ASSERT_FALSE(message_center_tray_->message_center_visible());
 106
 107   message_center_tray_->ToggleMessageCenterBubble();
 108
 109   ASSERT_FALSE(message_center_tray_->popups_visible());
 110   ASSERT_TRUE(message_center_tray_->message_center_visible());
 111
 112   message_center_tray_->ToggleMessageCenterBubble();
 113
 114   ASSERT_FALSE(message_center_tray_->popups_visible());
 115   ASSERT_FALSE(message_center_tray_->message_center_visible());
 116 }
 117
 118 TEST_F(MessageCenterTrayTest, BasicPopup) {
 119   ASSERT_FALSE(message_center_tray_->popups_visible());
 120   ASSERT_FALSE(message_center_tray_->message_center_visible());
 121
 122   message_center_tray_->ShowPopupBubble();
 123
 124   ASSERT_FALSE(message_center_tray_->popups_visible());
 125   ASSERT_FALSE(message_center_tray_->message_center_visible());
 126
 127   AddNotification("BasicPopup");
 128
 129   ASSERT_TRUE(message_center_tray_->popups_visible());
 130   ASSERT_FALSE(message_center_tray_->message_center_visible());
 131
 132   message_center_tray_->HidePopupBubble();
 133
 134   ASSERT_FALSE(message_center_tray_->popups_visible());
 135   ASSERT_FALSE(message_center_tray_->message_center_visible());
 136 }
 137
 138 TEST_F(MessageCenterTrayTest, MessageCenterClosesPopups) {
 139   ASSERT_FALSE(message_center_tray_->popups_visible());
 140   ASSERT_FALSE(message_center_tray_->message_center_visible());
 141
 142   AddNotification("MessageCenterClosesPopups");
 143
 144   ASSERT_TRUE(message_center_tray_->popups_visible());
 145   ASSERT_FALSE(message_center_tray_->message_center_visible());
 146
 147   bool shown = message_center_tray_->ShowMessageCenterBubble();
 148   EXPECT_TRUE(shown);
 149
 150   ASSERT_FALSE(message_center_tray_->popups_visible());
 151   ASSERT_TRUE(message_center_tray_->message_center_visible());
 152
 153   // The notification is queued if it's added when message center is visible.
 154   AddNotification("MessageCenterClosesPopups2");
 155
 156   message_center_tray_->ShowPopupBubble();
 157
 158   ASSERT_FALSE(message_center_tray_->popups_visible());
 159   ASSERT_TRUE(message_center_tray_->message_center_visible());
 160
 161   message_center_tray_->HideMessageCenterBubble();
 162
 163   // The queued notification appears as a popup.
 164   ASSERT_TRUE(message_center_tray_->popups_visible());
 165   ASSERT_FALSE(message_center_tray_->message_center_visible());
 166
 167   message_center_tray_->ShowMessageCenterBubble();
 168   message_center_tray_->HideMessageCenterBubble();
 169   ASSERT_FALSE(message_center_tray_->popups_visible());
 170   ASSERT_FALSE(message_center_tray_->message_center_visible());
 171 }
 172
 173 TEST_F(MessageCenterTrayTest, MessageCenterReopenPopupsForSystemPriority) {
 174   ASSERT_FALSE(message_center_tray_->popups_visible());
 175   ASSERT_FALSE(message_center_tray_->message_center_visible());
 176
 177   scoped_ptr<Notification> notification(
 178       new Notification(message_center::NOTIFICATION_TYPE_SIMPLE,
 179                        "MessageCenterReopnPopupsForSystemPriority",
 180                        ASCIIToUTF16("Test Web Notification"),
 181                        ASCIIToUTF16("Notification message body."),
 182                        gfx::Image(),
 183                        ASCIIToUTF16("www.test.org"),
 184                        DummyNotifierId(),
 185                        message_center::RichNotificationData(),
 186                        NULL /* delegate */));
 187   notification->SetSystemPriority();
 188   message_center_->AddNotification(notification.Pass());
 189
 190   ASSERT_TRUE(message_center_tray_->popups_visible());
 191   ASSERT_FALSE(message_center_tray_->message_center_visible());
 192
 193   bool shown = message_center_tray_->ShowMessageCenterBubble();
 194   EXPECT_TRUE(shown);
 195
 196   ASSERT_FALSE(message_center_tray_->popups_visible());
 197   ASSERT_TRUE(message_center_tray_->message_center_visible());
 198
 199   message_center_tray_->HideMessageCenterBubble();
 200
 201   ASSERT_TRUE(message_center_tray_->popups_visible());
 202   ASSERT_FALSE(message_center_tray_->message_center_visible());
 203 }
 204
 205 TEST_F(MessageCenterTrayTest, ShowBubbleFails) {
 206   // Now the delegate will signal that it was unable to show a bubble.
 207   delegate_->show_popups_success_ = false;
 208   delegate_->show_message_center_success_ = false;
 209
 210   ASSERT_FALSE(message_center_tray_->popups_visible());
 211   ASSERT_FALSE(message_center_tray_->message_center_visible());
 212
 213   AddNotification("ShowBubbleFails");
 214
 215   message_center_tray_->ShowPopupBubble();
 216
 217   ASSERT_FALSE(message_center_tray_->popups_visible());
 218   ASSERT_FALSE(message_center_tray_->message_center_visible());
 219
 220   bool shown = message_center_tray_->ShowMessageCenterBubble();
 221   EXPECT_FALSE(shown);
 222
 223   ASSERT_FALSE(message_center_tray_->popups_visible());
 224   ASSERT_FALSE(message_center_tray_->message_center_visible());
 225
 226   message_center_tray_->HideMessageCenterBubble();
 227
 228   ASSERT_FALSE(message_center_tray_->popups_visible());
 229   ASSERT_FALSE(message_center_tray_->message_center_visible());
 230
 231   message_center_tray_->ToggleMessageCenterBubble();
 232
 233   ASSERT_FALSE(message_center_tray_->popups_visible());
 234   ASSERT_FALSE(message_center_tray_->message_center_visible());
 235
 236   message_center_tray_->HidePopupBubble();
 237
 238   ASSERT_FALSE(message_center_tray_->popups_visible());
 239   ASSERT_FALSE(message_center_tray_->message_center_visible());
 240 }
 241
 242 TEST_F(MessageCenterTrayTest, ContextMenuTest) {
 243   const std::string id1 = "id1";
 244   const std::string id2 = "id2";
 245   const std::string id3 = "id3";
 246   AddNotification(id1);
 247
 248   base::string16 display_source = ASCIIToUTF16("www.test.org");
 249   NotifierId notifier_id = DummyNotifierId();
 250
 251   NotifierId notifier_id2(NotifierId::APPLICATION, "sample-app");
 252   scoped_ptr<Notification> notification(
 253       new Notification(message_center::NOTIFICATION_TYPE_SIMPLE,
 254                        id2,
 255                        ASCIIToUTF16("Test Web Notification"),
 256                        ASCIIToUTF16("Notification message body."),
 257                        gfx::Image(),
 258                        base::string16() /* empty display source */,
 259                        notifier_id2,
 260                        message_center::RichNotificationData(),
 261                        NULL /* delegate */));
 262   message_center_->AddNotification(notification.Pass());
 263
 264   AddNotification(id3);
 265
 266   scoped_ptr<ui::MenuModel> model(
 267       message_center_tray_->CreateNotificationMenuModel(
 268           notifier_id, display_source));
 269   EXPECT_EQ(2, model->GetItemCount());
 270   const int second_command = model->GetCommandIdAt(1);
 271
 272   // The second item is to open the settings.
 273   EXPECT_TRUE(model->IsEnabledAt(0));
 274   EXPECT_TRUE(model->IsEnabledAt(1));
 275   model->ActivatedAt(1);
 276   EXPECT_TRUE(message_center_tray_->message_center_visible());
 277
 278   message_center_tray_->HideMessageCenterBubble();
 279
 280   // The first item is to disable notifications from the notifier id. It also
 281   // removes all notifications from the same notifier, i.e. id1 and id3.
 282   model->ActivatedAt(0);
 283   NotificationList::Notifications notifications =
 284       message_center_->GetVisibleNotifications();
 285   EXPECT_EQ(1u, notifications.size());
 286   EXPECT_EQ(id2, (*notifications.begin())->id());
 287
 288   // Disables the context menu.
 289   delegate_->enable_context_menu_ = false;
 290
 291   // id2 doesn't have the display source, so it don't have the menu item for
 292   // disabling notifications.
 293   model = message_center_tray_->CreateNotificationMenuModel(
 294       notifier_id2, base::string16());
 295   EXPECT_EQ(1, model->GetItemCount());
 296   EXPECT_EQ(second_command, model->GetCommandIdAt(0));
 297
 298   // The command itself is disabled because delegate disables context menu.
 299   EXPECT_FALSE(model->IsEnabledAt(0));
 300 }
 301
 302 }  // namespace message_center