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