chrome/browser/notifications/message_center_notifications_browsertest.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 <map>
   6 #include <string>
   7
   8 #include "base/command_line.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/strings/string_number_conversions.h"
  11 #include "base/strings/string_util.h"
  12 #include "base/strings/stringprintf.h"
  13 #include "base/strings/utf_string_conversions.h"
  14 #include "chrome/browser/browser_process.h"
  15 #include "chrome/browser/notifications/message_center_notification_manager.h"
  16 #include "chrome/browser/notifications/notification.h"
  17 #include "chrome/browser/notifications/notification_ui_manager.h"
  18 #include "chrome/browser/profiles/profile.h"
  19 #include "chrome/browser/ui/browser.h"
  20 #include "chrome/test/base/in_process_browser_test.h"
  21 #include "chrome/test/base/test_switches.h"
  22 #include "ui/message_center/message_center.h"
  23 #include "ui/message_center/message_center_types.h"
  24
  25 class TestAddObserver : public message_center::MessageCenterObserver {
  26  public:
  27   explicit TestAddObserver(message_center::MessageCenter* message_center)
  28       : message_center_(message_center) {
  29     message_center_->AddObserver(this);
  30   }
  31
  32   ~TestAddObserver() override { message_center_->RemoveObserver(this); }
  33
  34   void OnNotificationAdded(const std::string& id) override {
  35     std::string log = logs_[id];
  36     if (!log.empty())
  37       log += "_";
  38     logs_[id] = log + "add-" + id;
  39   }
  40
  41   void OnNotificationUpdated(const std::string& id) override {
  42     std::string log = logs_[id];
  43     if (!log.empty())
  44       log += "_";
  45     logs_[id] = log + "update-" + id;
  46   }
  47
  48   const std::string log(const std::string& id) { return logs_[id]; }
  49   void reset_logs() { logs_.clear(); }
  50
  51  private:
  52   std::map<std::string, std::string> logs_;
  53   message_center::MessageCenter* message_center_;
  54 };
  55
  56 class MessageCenterNotificationsTest : public InProcessBrowserTest {
  57  public:
  58   MessageCenterNotificationsTest() {}
  59
  60   MessageCenterNotificationManager* manager() {
  61     return static_cast<MessageCenterNotificationManager*>(
  62         g_browser_process->notification_ui_manager());
  63   }
  64
  65   message_center::MessageCenter* message_center() {
  66     return g_browser_process->message_center();
  67   }
  68
  69   Profile* profile() { return browser()->profile(); }
  70
  71   class TestDelegate : public NotificationDelegate {
  72    public:
  73     explicit TestDelegate(const std::string& id) : id_(id) {}
  74
  75     void Display() override { log_ += "Display_"; }
  76     void Close(bool by_user) override {
  77       log_ += "Close_";
  78       log_ += (by_user ? "by_user_" : "programmatically_");
  79     }
  80     void Click() override { log_ += "Click_"; }
  81     void ButtonClick(int button_index) override {
  82       log_ += "ButtonClick_";
  83       log_ += base::IntToString(button_index) + "_";
  84     }
  85     std::string id() const override { return id_; }
  86
  87     const std::string& log() { return log_; }
  88
  89    private:
  90     ~TestDelegate() override {}
  91     std::string id_;
  92     std::string log_;
  93
  94     DISALLOW_COPY_AND_ASSIGN(TestDelegate);
  95   };
  96
  97   Notification CreateTestNotification(const std::string& delegate_id,
  98                                       TestDelegate** delegate = NULL) {
  99     TestDelegate* new_delegate = new TestDelegate(delegate_id);
 100     if (delegate) {
 101       *delegate = new_delegate;
 102       new_delegate->AddRef();
 103     }
 104
 105     return Notification(GURL("chrome-test://testing/"),
 106                         base::ASCIIToUTF16("title"),
 107                         base::ASCIIToUTF16("message"),
 108                         gfx::Image(),
 109                         base::UTF8ToUTF16("chrome-test://testing/"),
 110                         "REPLACE-ME",
 111                         new_delegate);
 112   }
 113
 114   Notification CreateRichTestNotification(const std::string& id,
 115                                           TestDelegate** delegate = NULL) {
 116     TestDelegate* new_delegate = new TestDelegate(id);
 117     if (delegate) {
 118       *delegate = new_delegate;
 119       new_delegate->AddRef();
 120     }
 121
 122     message_center::RichNotificationData data;
 123
 124     return Notification(
 125         message_center::NOTIFICATION_TYPE_BASE_FORMAT,
 126         base::ASCIIToUTF16("title"), base::ASCIIToUTF16("message"),
 127         gfx::Image(),
 128         message_center::NotifierId(message_center::NotifierId::APPLICATION,
 129                                    "extension_id"),
 130         base::UTF8ToUTF16("chrome-test://testing/"),
 131         GURL("chrome-test://testing/"), "REPLACE-ME", data, new_delegate);
 132   }
 133 };
 134
 135 // TODO(rsesek): Implement Message Center on Mac and get these tests passing
 136 // for real. http://crbug.com/179904
 137 #if !defined(OS_MACOSX)
 138
 139 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest, RetrieveBaseParts) {
 140   EXPECT_TRUE(manager());
 141   EXPECT_TRUE(message_center());
 142 }
 143
 144 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest, BasicAddCancel) {
 145 #if defined(OS_WIN) && defined(USE_ASH)
 146   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 147   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 148           switches::kAshBrowserTests))
 149     return;
 150 #endif
 151
 152   // Someone may create system notifications like "you're in multi-profile
 153   // mode..." or something which may change the expectation.
 154   // TODO(mukai): move this to SetUpOnMainThread() after fixing the side-effect
 155   // of canceling animation which prevents some Displayed() event.
 156   manager()->CancelAll();
 157   manager()->Add(CreateTestNotification("hey"), profile());
 158   EXPECT_EQ(1u, message_center()->NotificationCount());
 159   manager()->CancelById("hey", NotificationUIManager::GetProfileID(profile()));
 160   EXPECT_EQ(0u, message_center()->NotificationCount());
 161 }
 162
 163 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest, BasicDelegate) {
 164 #if defined(OS_WIN) && defined(USE_ASH)
 165   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 166   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 167           switches::kAshBrowserTests))
 168     return;
 169 #endif
 170
 171   TestDelegate* delegate;
 172   manager()->Add(CreateTestNotification("hey", &delegate), profile());
 173   // Verify that delegate accumulated correct log of events.
 174   EXPECT_EQ("Display_", delegate->log());
 175   manager()->CancelById("hey", NotificationUIManager::GetProfileID(profile()));
 176   // Verify that delegate accumulated correct log of events.
 177   EXPECT_EQ("Display_Close_programmatically_", delegate->log());
 178   delegate->Release();
 179 }
 180
 181 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest, ButtonClickedDelegate) {
 182 #if defined(OS_WIN) && defined(USE_ASH)
 183   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 184   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 185           switches::kAshBrowserTests))
 186     return;
 187 #endif
 188
 189   TestDelegate* delegate;
 190   manager()->Add(CreateTestNotification("n", &delegate), profile());
 191   const std::string notification_id =
 192       manager()->GetMessageCenterNotificationIdForTest("n", profile());
 193   message_center()->ClickOnNotificationButton(notification_id, 1);
 194   // Verify that delegate accumulated correct log of events.
 195   EXPECT_EQ("Display_ButtonClick_1_", delegate->log());
 196   delegate->Release();
 197 }
 198
 199 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest,
 200                        UpdateExistingNotification) {
 201 #if defined(OS_WIN) && defined(USE_ASH)
 202   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 203   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 204           switches::kAshBrowserTests))
 205     return;
 206 #endif
 207
 208   TestDelegate* delegate;
 209   manager()->Add(CreateTestNotification("n", &delegate), profile());
 210   TestDelegate* delegate2;
 211   manager()->Add(CreateRichTestNotification("n", &delegate2), profile());
 212
 213   manager()->CancelById("n", NotificationUIManager::GetProfileID(profile()));
 214   EXPECT_EQ("Display_", delegate->log());
 215   EXPECT_EQ("Close_programmatically_", delegate2->log());
 216
 217   delegate->Release();
 218   delegate2->Release();
 219 }
 220
 221 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest, QueueWhenCenterVisible) {
 222 #if defined(OS_WIN) && defined(USE_ASH)
 223   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 224   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 225           switches::kAshBrowserTests))
 226     return;
 227 #endif
 228
 229   TestAddObserver observer(message_center());
 230
 231   TestDelegate* delegate;
 232   TestDelegate* delegate2;
 233
 234   manager()->Add(CreateTestNotification("n", &delegate), profile());
 235   const std::string id_n =
 236       manager()->GetMessageCenterNotificationIdForTest("n", profile());
 237   message_center()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
 238   manager()->Add(CreateTestNotification("n2", &delegate2), profile());
 239   const std::string id_n2 =
 240       manager()->GetMessageCenterNotificationIdForTest("n2", profile());
 241
 242   // 'update-n' should happen since SetVisibility updates is_read status of n.
 243   // TODO(mukai): fix event handling to happen update-n just once.
 244   EXPECT_EQ(base::StringPrintf("add-%s_update-%s_update-%s",
 245                                id_n.c_str(),
 246                                id_n.c_str(),
 247                                id_n.c_str()),
 248             observer.log(id_n));
 249
 250   message_center()->SetVisibility(message_center::VISIBILITY_TRANSIENT);
 251
 252   EXPECT_EQ(base::StringPrintf("add-%s", id_n2.c_str()), observer.log(id_n2));
 253
 254   delegate->Release();
 255   delegate2->Release();
 256 }
 257
 258 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest,
 259                        UpdateNonProgressNotificationWhenCenterVisible) {
 260 #if defined(OS_WIN) && defined(USE_ASH)
 261   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 262   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 263           switches::kAshBrowserTests))
 264     return;
 265 #endif
 266
 267   TestAddObserver observer(message_center());
 268
 269   TestDelegate* delegate;
 270
 271   // Add a non-progress notification and update it while the message center
 272   // is visible.
 273   Notification notification = CreateTestNotification("n", &delegate);
 274   manager()->Add(notification, profile());
 275   const std::string notification_id =
 276       manager()->GetMessageCenterNotificationIdForTest("n", profile());
 277   message_center()->ClickOnNotification(notification_id);
 278   message_center()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
 279   observer.reset_logs();
 280   notification.set_title(base::ASCIIToUTF16("title2"));
 281   manager()->Update(notification, profile());
 282
 283   // Expect that the notification update is not done.
 284   EXPECT_EQ("", observer.log(notification_id));
 285
 286   message_center()->SetVisibility(message_center::VISIBILITY_TRANSIENT);
 287   EXPECT_EQ(base::StringPrintf("update-%s", notification_id.c_str()),
 288             observer.log(notification_id));
 289
 290   delegate->Release();
 291 }
 292
 293 IN_PROC_BROWSER_TEST_F(
 294     MessageCenterNotificationsTest,
 295     UpdateNonProgressToProgressNotificationWhenCenterVisible) {
 296 #if defined(OS_WIN) && defined(USE_ASH)
 297   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 298   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 299           switches::kAshBrowserTests))
 300     return;
 301 #endif
 302
 303   TestAddObserver observer(message_center());
 304
 305   TestDelegate* delegate;
 306
 307   // Add a non-progress notification and change the type to progress while the
 308   // message center is visible.
 309   Notification notification = CreateTestNotification("n", &delegate);
 310   manager()->Add(notification, profile());
 311   const std::string notification_id =
 312       manager()->GetMessageCenterNotificationIdForTest("n", profile());
 313   message_center()->ClickOnNotification(notification_id);
 314   message_center()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
 315   observer.reset_logs();
 316   notification.set_type(message_center::NOTIFICATION_TYPE_PROGRESS);
 317   manager()->Update(notification, profile());
 318
 319   // Expect that the notification update is not done.
 320   EXPECT_EQ("", observer.log(notification_id));
 321
 322   message_center()->SetVisibility(message_center::VISIBILITY_TRANSIENT);
 323   EXPECT_EQ(base::StringPrintf("update-%s", notification_id.c_str()),
 324             observer.log(notification_id));
 325
 326   delegate->Release();
 327 }
 328
 329 IN_PROC_BROWSER_TEST_F(MessageCenterNotificationsTest,
 330                        UpdateProgressNotificationWhenCenterVisible) {
 331 #if defined(OS_WIN) && defined(USE_ASH)
 332   // Disable this test in Metro+Ash for now (http://crbug.com/262796).
 333   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
 334           switches::kAshBrowserTests))
 335     return;
 336 #endif
 337
 338   TestAddObserver observer(message_center());
 339
 340   TestDelegate* delegate;
 341
 342   // Add a progress notification and update it while the message center
 343   // is visible.
 344   Notification notification = CreateTestNotification("n", &delegate);
 345   notification.set_type(message_center::NOTIFICATION_TYPE_PROGRESS);
 346   manager()->Add(notification, profile());
 347   const std::string notification_id =
 348       manager()->GetMessageCenterNotificationIdForTest("n", profile());
 349   message_center()->ClickOnNotification(notification_id);
 350   message_center()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
 351   observer.reset_logs();
 352   notification.set_progress(50);
 353   manager()->Update(notification, profile());
 354
 355   // Expect that the progress notification update is performed.
 356   EXPECT_EQ(base::StringPrintf("update-%s", notification_id.c_str()),
 357             observer.log(notification_id));
 358
 359   delegate->Release();
 360 }
 361
 362 #endif  // !defined(OS_MACOSX)