ui/message_center/views/message_center_view_unittest.cc

   1 // Copyright 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
   7 #include "base/logging.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/strings/utf_string_conversions.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11 #include "ui/message_center/fake_message_center.h"
  12 #include "ui/message_center/notification.h"
  13 #include "ui/message_center/notification_list.h"
  14 #include "ui/message_center/notification_types.h"
  15 #include "ui/message_center/views/message_center_view.h"
  16 #include "ui/message_center/views/notification_view.h"
  17
  18 namespace message_center {
  19
  20 /* Types **********************************************************************/
  21
  22 enum CallType {
  23   GET_PREFERRED_SIZE,
  24   GET_HEIGHT_FOR_WIDTH,
  25   LAYOUT
  26 };
  27
  28 /* Instrumented/Mock NotificationView subclass ********************************/
  29
  30 class MockNotificationView : public NotificationView {
  31  public:
  32   class Test {
  33    public:
  34     virtual void RegisterCall(int receiver_id, CallType type) = 0;
  35   };
  36
  37   explicit MockNotificationView(const Notification& notification,
  38                                 MessageCenter* message_center,
  39                                 Test* test,
  40                                 int view_id);
  41   virtual ~MockNotificationView();
  42
  43   virtual gfx::Size GetPreferredSize() OVERRIDE;
  44   virtual int GetHeightForWidth(int w) OVERRIDE;
  45   virtual void Layout() OVERRIDE;
  46
  47   int get_id() { return id_; };
  48
  49  private:
  50   void RegisterCall(CallType type);
  51
  52   Test* test_;
  53   int id_;
  54
  55   DISALLOW_COPY_AND_ASSIGN(MockNotificationView);
  56 };
  57
  58 MockNotificationView::MockNotificationView(const Notification& notification,
  59                                            MessageCenter* message_center,
  60                                            Test* test,
  61                                            int view_id)
  62     : NotificationView(notification, message_center, NULL, true),
  63       test_(test),
  64       id_(view_id) {
  65 }
  66
  67 MockNotificationView::~MockNotificationView() {
  68 }
  69
  70 gfx::Size MockNotificationView::GetPreferredSize() {
  71   RegisterCall(GET_PREFERRED_SIZE);
  72   return child_count() ? NotificationView::GetPreferredSize() :
  73                          gfx::Size(id_, id_);
  74 }
  75
  76 int MockNotificationView::GetHeightForWidth(int width) {
  77   RegisterCall(GET_HEIGHT_FOR_WIDTH);
  78   return child_count() ? NotificationView::GetHeightForWidth(width) : (id_);
  79 }
  80
  81 void MockNotificationView::Layout() {
  82   RegisterCall(LAYOUT);
  83   if (child_count())
  84     NotificationView::Layout();
  85 }
  86
  87 void MockNotificationView::RegisterCall(CallType type) {
  88   if (test_)
  89     test_->RegisterCall(id_, type);
  90 }
  91
  92 /* Test fixture ***************************************************************/
  93
  94 class MessageCenterViewTest : public testing::Test,
  95                               public MockNotificationView::Test {
  96  public:
  97   MessageCenterViewTest();
  98   virtual ~MessageCenterViewTest();
  99
 100   virtual void SetUp() OVERRIDE;
 101   virtual void TearDown() OVERRIDE;
 102
 103   MessageCenterView* GetMessageCenterView();
 104   int GetNotificationCount();
 105   int GetCallCount(CallType type);
 106
 107   virtual void RegisterCall(int receiver_id, CallType type) OVERRIDE;
 108
 109   void LogBounds(int depth, views::View* view);
 110
 111  private:
 112   views::View* MakeParent(views::View* child1, views::View* child2);
 113
 114
 115   scoped_ptr<MessageCenterView> message_center_view_;
 116   FakeMessageCenter message_center_;
 117   std::map<CallType,int> callCounts_;
 118
 119   DISALLOW_COPY_AND_ASSIGN(MessageCenterViewTest);
 120 };
 121
 122 MessageCenterViewTest::MessageCenterViewTest() {
 123 }
 124
 125 MessageCenterViewTest::~MessageCenterViewTest() {
 126 }
 127
 128 void MessageCenterViewTest::SetUp() {
 129   // Create a dummy notification.
 130   Notification notification(NOTIFICATION_TYPE_SIMPLE,
 131                             std::string("notification id"),
 132                             UTF8ToUTF16("title"),
 133                             UTF8ToUTF16("message"),
 134                             gfx::Image(),
 135                             UTF8ToUTF16("display source"),
 136                             std::string("extension id"),
 137                             message_center::RichNotificationData(),
 138                             NULL);
 139
 140   // ...and a list for it.
 141   NotificationList::Notifications notifications;
 142   notifications.insert(&notification);
 143
 144   // Then create a new MessageCenterView with that single notification.
 145   message_center_view_.reset(new MessageCenterView(
 146       &message_center_, NULL, 100, false, /*top_down =*/false));
 147   message_center_view_->SetNotifications(notifications);
 148
 149   // Remove and delete the NotificationView now owned by the MessageCenterView's
 150   // MessageListView and replace it with an instrumented MockNotificationView
 151   // that will become owned by the MessageListView.
 152   MockNotificationView* mock;
 153   mock = new MockNotificationView(notification, &message_center_, this, 42);
 154   message_center_view_->message_views_.push_back(mock);
 155   message_center_view_->SetNotificationViewForTest(mock);
 156 }
 157
 158 void MessageCenterViewTest::TearDown() {
 159   message_center_view_.reset();
 160 }
 161
 162 MessageCenterView* MessageCenterViewTest::GetMessageCenterView() {
 163   return message_center_view_.get();
 164 }
 165
 166 int MessageCenterViewTest::GetNotificationCount() {
 167   return 1;
 168 }
 169
 170 int MessageCenterViewTest::GetCallCount(CallType type) {
 171   return callCounts_[type];
 172 }
 173
 174 void MessageCenterViewTest::RegisterCall(int receiver_id, CallType type) {
 175   callCounts_[type] += 1;
 176 }
 177
 178 void MessageCenterViewTest::LogBounds(int depth, views::View* view) {
 179   string16 inset;
 180   for (int i = 0; i < depth; ++i)
 181     inset.append(UTF8ToUTF16("  "));
 182   gfx::Rect bounds = view->bounds();
 183   DVLOG(0) << inset << bounds.width() << " x " << bounds.height()
 184            << " @ " << bounds.x() << ", " << bounds.y();
 185   for (int i = 0; i < view->child_count(); ++i)
 186     LogBounds(depth + 1, view->child_at(i));
 187 }
 188
 189 /* Unit tests *****************************************************************/
 190
 191 TEST_F(MessageCenterViewTest, CallTest) {
 192   // Exercise (with size values that just need to be large enough).
 193   GetMessageCenterView()->SetBounds(0, 0, 100, 100);
 194
 195   // Verify that this didn't generate more than 2 Layout() call per descendant
 196   // NotificationView or more than a total of 20 GetPreferredSize() and
 197   // GetHeightForWidth() calls per descendant NotificationView. 20 is a very
 198   // large number corresponding to the current reality. That number will be
 199   // ratcheted down over time as the code improves.
 200   EXPECT_LE(GetCallCount(LAYOUT), GetNotificationCount() * 2);
 201   EXPECT_LE(GetCallCount(GET_PREFERRED_SIZE) +
 202             GetCallCount(GET_HEIGHT_FOR_WIDTH),
 203             GetNotificationCount() * 20);
 204 }
 205
 206 }  // namespace message_center