chrome/browser/ui/cocoa/notifications/message_center_tray_bridge_unittest.mm

   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 #import "chrome/browser/ui/cocoa/notifications/message_center_tray_bridge.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/run_loop.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #import "ui/gfx/test/ui_cocoa_test_helper.h"
  13 #import "ui/message_center/cocoa/status_item_view.h"
  14 #include "ui/message_center/message_center.h"
  15 #include "ui/message_center/message_center_switches.h"
  16 #include "ui/message_center/notification.h"
  17 #include "ui/message_center/notifier_settings.h"
  18
  19 class MessageCenterTrayBridgeTest : public ui::CocoaTest {
  20  public:
  21   virtual void SetUp() OVERRIDE {
  22     ui::CocoaTest::SetUp();
  23
  24     message_center::MessageCenter::Initialize();
  25     center_ = message_center::MessageCenter::Get();
  26
  27     bridge_.reset(new MessageCenterTrayBridge(center_));
  28   }
  29
  30   virtual void TearDown() OVERRIDE {
  31     bridge_.reset();
  32     message_center::MessageCenter::Shutdown();
  33     ui::CocoaTest::TearDown();
  34   }
  35
  36   MCStatusItemView* status_item() { return bridge_->status_item_view_.get(); }
  37
  38  protected:
  39   scoped_ptr<message_center::Notification> GetNotification() {
  40     message_center::RichNotificationData data;
  41     data.priority = -1;
  42     return make_scoped_ptr(new message_center::Notification(
  43         message_center::NOTIFICATION_TYPE_SIMPLE,
  44         "1",
  45         base::ASCIIToUTF16("First notification"),
  46         base::ASCIIToUTF16("This is a simple test."),
  47         gfx::Image(),
  48         base::string16(),
  49         message_center::NotifierId(),
  50         data,
  51         NULL));
  52   }
  53
  54   base::MessageLoop message_loop_;
  55   message_center::MessageCenter* center_;  // Weak, global.
  56   scoped_ptr<MessageCenterTrayBridge> bridge_;
  57 };
  58
  59 class MessageCenterTrayBridgeTestPrefNever
  60     : public MessageCenterTrayBridgeTest {
  61  public:
  62   virtual void SetUp() OVERRIDE {
  63     CommandLine::ForCurrentProcess()->AppendSwitchASCII(
  64         message_center::switches::kNotificationCenterTrayBehavior, "never");
  65     MessageCenterTrayBridgeTest::SetUp();
  66   }
  67 };
  68
  69 class MessageCenterTrayBridgeTestPrefAlways
  70     : public MessageCenterTrayBridgeTest {
  71  public:
  72   virtual void SetUp() OVERRIDE {
  73     CommandLine::ForCurrentProcess()->AppendSwitchASCII(
  74         message_center::switches::kNotificationCenterTrayBehavior, "always");
  75     MessageCenterTrayBridgeTest::SetUp();
  76   }
  77 };
  78
  79 class MessageCenterTrayBridgeTestPrefUnread
  80     : public MessageCenterTrayBridgeTest {
  81  public:
  82   virtual void SetUp() OVERRIDE {
  83     CommandLine::ForCurrentProcess()->AppendSwitchASCII(
  84         message_center::switches::kNotificationCenterTrayBehavior, "unread");
  85     MessageCenterTrayBridgeTest::SetUp();
  86   }
  87 };
  88
  89 TEST_F(MessageCenterTrayBridgeTest, StatusItemOnlyAfterFirstNotification) {
  90   EXPECT_FALSE(status_item());
  91
  92   center_->AddNotification(GetNotification());
  93
  94   base::RunLoop().RunUntilIdle();
  95
  96   EXPECT_TRUE(status_item());
  97   EXPECT_EQ(1u, [status_item() unreadCount]);
  98
  99   center_->RemoveNotification("1", /*by_user=*/true);
 100
 101   base::RunLoop().RunUntilIdle();
 102
 103   EXPECT_TRUE(status_item());
 104 }
 105
 106 TEST_F(MessageCenterTrayBridgeTestPrefNever, StatusItemNeverWithPref) {
 107   EXPECT_FALSE(status_item());
 108
 109   center_->AddNotification(GetNotification());
 110
 111   base::RunLoop().RunUntilIdle();
 112
 113   EXPECT_FALSE(status_item());
 114
 115   center_->RemoveNotification("1", /*by_user=*/true);
 116
 117   base::RunLoop().RunUntilIdle();
 118
 119   EXPECT_FALSE(status_item());
 120 }
 121
 122 TEST_F(MessageCenterTrayBridgeTestPrefAlways, StatusItemAlwaysWithPref) {
 123   EXPECT_TRUE(status_item());
 124
 125   center_->AddNotification(GetNotification());
 126
 127   base::RunLoop().RunUntilIdle();
 128
 129   EXPECT_TRUE(status_item());
 130   EXPECT_EQ(1u, [status_item() unreadCount]);
 131
 132   center_->RemoveNotification("1", /*by_user=*/true);
 133
 134   base::RunLoop().RunUntilIdle();
 135
 136   EXPECT_TRUE(status_item());
 137 }
 138
 139 TEST_F(MessageCenterTrayBridgeTestPrefUnread, StatusItemUnreadWithPref) {
 140   EXPECT_FALSE(status_item());
 141
 142   center_->AddNotification(GetNotification());
 143
 144   base::RunLoop().RunUntilIdle();
 145
 146   EXPECT_TRUE(status_item());
 147   EXPECT_EQ(1u, [status_item() unreadCount]);
 148
 149   center_->RemoveNotification("1", /*by_user=*/true);
 150
 151   base::RunLoop().RunUntilIdle();
 152
 153   EXPECT_FALSE(status_item());
 154 }