ash/system/tray/system_tray_unittest.cc

   1 // Copyright (c) 2012 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 "ash/system/tray/system_tray.h"
   6
   7 #include <vector>
   8
   9 #include "ash/root_window_controller.h"
  10 #include "ash/system/status_area_widget.h"
  11 #include "ash/system/tray/system_tray_item.h"
  12 #include "ash/test/ash_test_base.h"
  13 #include "base/utf_string_conversions.h"
  14 #include "ui/views/controls/label.h"
  15 #include "ui/views/layout/fill_layout.h"
  16 #include "ui/views/view.h"
  17 #include "ui/views/widget/widget.h"
  18
  19 namespace ash {
  20 namespace test {
  21
  22 namespace {
  23
  24 SystemTray* GetSystemTray() {
  25   return Shell::GetPrimaryRootWindowController()->status_area_widget()->
  26       system_tray();
  27 }
  28
  29 // Trivial item implementation that tracks its views for testing.
  30 class TestItem : public SystemTrayItem {
  31  public:
  32   TestItem() : SystemTrayItem(GetSystemTray()), tray_view_(NULL) {}
  33
  34   virtual views::View* CreateTrayView(user::LoginStatus status) {
  35     tray_view_ = new views::View;
  36     // Add a label so it has non-zero width.
  37     tray_view_->SetLayoutManager(new views::FillLayout);
  38     tray_view_->AddChildView(new views::Label(UTF8ToUTF16("Tray")));
  39     return tray_view_;
  40   }
  41
  42   virtual views::View* CreateDefaultView(user::LoginStatus status) {
  43     default_view_ = new views::View;
  44     default_view_->SetLayoutManager(new views::FillLayout);
  45     default_view_->AddChildView(new views::Label(UTF8ToUTF16("Default")));
  46     return default_view_;
  47   }
  48
  49   virtual views::View* CreateDetailedView(user::LoginStatus status) {
  50     detailed_view_ = new views::View;
  51     detailed_view_->SetLayoutManager(new views::FillLayout);
  52     detailed_view_->AddChildView(new views::Label(UTF8ToUTF16("Detailed")));
  53     return detailed_view_;
  54   }
  55
  56   virtual views::View* CreateNotificationView(user::LoginStatus status) {
  57     notification_view_ = new views::View;
  58     return notification_view_;
  59   }
  60
  61   virtual void DestroyTrayView() {
  62     tray_view_ = NULL;
  63   }
  64
  65   virtual void DestroyDefaultView() {
  66     default_view_ = NULL;
  67   }
  68
  69   virtual void DestroyDetailedView() {
  70     detailed_view_ = NULL;
  71   }
  72
  73   virtual void DestroyNotificationView() {
  74     notification_view_ = NULL;
  75   }
  76
  77   virtual void UpdateAfterLoginStatusChange(user::LoginStatus status) {
  78   }
  79
  80   views::View* tray_view() const { return tray_view_; }
  81   views::View* default_view() const { return default_view_; }
  82   views::View* detailed_view() const { return detailed_view_; }
  83   views::View* notification_view() const { return notification_view_; }
  84
  85  private:
  86   views::View* tray_view_;
  87   views::View* default_view_;
  88   views::View* detailed_view_;
  89   views::View* notification_view_;
  90 };
  91
  92 // Trivial item implementation that returns NULL from tray/default/detailed view
  93 // creation methods.
  94 class TestNoViewItem : public SystemTrayItem {
  95  public:
  96   TestNoViewItem() : SystemTrayItem(GetSystemTray()) {}
  97
  98   virtual views::View* CreateTrayView(user::LoginStatus status) OVERRIDE {
  99     return NULL;
 100   }
 101
 102   virtual views::View* CreateDefaultView(user::LoginStatus status) OVERRIDE {
 103     return NULL;
 104   }
 105
 106   virtual views::View* CreateDetailedView(user::LoginStatus status) OVERRIDE {
 107     return NULL;
 108   }
 109
 110   virtual views::View* CreateNotificationView(
 111       user::LoginStatus status) OVERRIDE {
 112     return NULL;
 113   }
 114
 115   virtual void DestroyTrayView() OVERRIDE {}
 116   virtual void DestroyDefaultView() OVERRIDE {}
 117   virtual void DestroyDetailedView() OVERRIDE {}
 118   virtual void DestroyNotificationView() OVERRIDE {}
 119   virtual void UpdateAfterLoginStatusChange(user::LoginStatus status) OVERRIDE {
 120   }
 121 };
 122
 123 }  // namespace
 124
 125 typedef AshTestBase SystemTrayTest;
 126
 127 TEST_F(SystemTrayTest, SystemTrayDefaultView) {
 128   SystemTray* tray = GetSystemTray();
 129   ASSERT_TRUE(tray->GetWidget());
 130
 131   tray->ShowDefaultView(BUBBLE_CREATE_NEW);
 132
 133   // Ensure that closing the bubble destroys it.
 134   ASSERT_TRUE(tray->CloseBubbleForTest());
 135   RunAllPendingInMessageLoop();
 136   ASSERT_FALSE(tray->CloseBubbleForTest());
 137 }
 138
 139 TEST_F(SystemTrayTest, SystemTrayTestItems) {
 140   SystemTray* tray = GetSystemTray();
 141   ASSERT_TRUE(tray->GetWidget());
 142
 143   TestItem* test_item = new TestItem;
 144   TestItem* detailed_item = new TestItem;
 145   tray->AddTrayItem(test_item);
 146   tray->AddTrayItem(detailed_item);
 147
 148   // Ensure the tray views are created.
 149   ASSERT_TRUE(test_item->tray_view() != NULL);
 150   ASSERT_TRUE(detailed_item->tray_view() != NULL);
 151
 152   // Ensure a default views are created.
 153   tray->ShowDefaultView(BUBBLE_CREATE_NEW);
 154   ASSERT_TRUE(test_item->default_view() != NULL);
 155   ASSERT_TRUE(detailed_item->default_view() != NULL);
 156
 157   // Show the detailed view, ensure it's created and the default view destroyed.
 158   tray->ShowDetailedView(detailed_item, 0, false, BUBBLE_CREATE_NEW);
 159   RunAllPendingInMessageLoop();
 160   ASSERT_TRUE(test_item->default_view() == NULL);
 161   ASSERT_TRUE(detailed_item->detailed_view() != NULL);
 162
 163   // Show the default view, ensure it's created and the detailed view destroyed.
 164   tray->ShowDefaultView(BUBBLE_CREATE_NEW);
 165   RunAllPendingInMessageLoop();
 166   ASSERT_TRUE(test_item->default_view() != NULL);
 167   ASSERT_TRUE(detailed_item->detailed_view() == NULL);
 168 }
 169
 170 TEST_F(SystemTrayTest, SystemTrayNoViewItems) {
 171   SystemTray* tray = GetSystemTray();
 172   ASSERT_TRUE(tray->GetWidget());
 173
 174   // Verify that no crashes occur on items lacking some views.
 175   TestNoViewItem* no_view_item = new TestNoViewItem;
 176   tray->AddTrayItem(no_view_item);
 177   tray->ShowDefaultView(BUBBLE_CREATE_NEW);
 178   tray->ShowDetailedView(no_view_item, 0, false, BUBBLE_USE_EXISTING);
 179   RunAllPendingInMessageLoop();
 180 }
 181
 182 TEST_F(SystemTrayTest, TrayWidgetAutoResizes) {
 183   SystemTray* tray = GetSystemTray();
 184   ASSERT_TRUE(tray->GetWidget());
 185
 186   // Add an initial tray item so that the tray gets laid out correctly.
 187   TestItem* initial_item = new TestItem;
 188   tray->AddTrayItem(initial_item);
 189
 190   gfx::Size initial_size = tray->GetWidget()->GetWindowBoundsInScreen().size();
 191
 192   TestItem* new_item = new TestItem;
 193   tray->AddTrayItem(new_item);
 194
 195   gfx::Size new_size = tray->GetWidget()->GetWindowBoundsInScreen().size();
 196
 197   // Adding the new item should change the size of the tray.
 198   EXPECT_NE(initial_size.ToString(), new_size.ToString());
 199
 200   // Hiding the tray view of the new item should also change the size of the
 201   // tray.
 202   new_item->tray_view()->SetVisible(false);
 203   EXPECT_EQ(initial_size.ToString(),
 204             tray->GetWidget()->GetWindowBoundsInScreen().size().ToString());
 205
 206   new_item->tray_view()->SetVisible(true);
 207   EXPECT_EQ(new_size.ToString(),
 208             tray->GetWidget()->GetWindowBoundsInScreen().size().ToString());
 209 }
 210
 211 TEST_F(SystemTrayTest, SystemTrayNotifications) {
 212   SystemTray* tray = GetSystemTray();
 213   ASSERT_TRUE(tray->GetWidget());
 214
 215   TestItem* test_item = new TestItem;
 216   TestItem* detailed_item = new TestItem;
 217   tray->AddTrayItem(test_item);
 218   tray->AddTrayItem(detailed_item);
 219
 220   // Ensure the tray views are created.
 221   ASSERT_TRUE(test_item->tray_view() != NULL);
 222   ASSERT_TRUE(detailed_item->tray_view() != NULL);
 223
 224   // Ensure a notification view is created.
 225   tray->ShowNotificationView(test_item);
 226   ASSERT_TRUE(test_item->notification_view() != NULL);
 227
 228   // Show the default view, ensure the notification view is destroyed.
 229   tray->ShowDefaultView(BUBBLE_CREATE_NEW);
 230   RunAllPendingInMessageLoop();
 231   ASSERT_TRUE(test_item->notification_view() == NULL);
 232
 233   // Show the detailed view, ensure the notificaiton view is created again.
 234   tray->ShowDetailedView(detailed_item, 0, false, BUBBLE_CREATE_NEW);
 235   RunAllPendingInMessageLoop();
 236   ASSERT_TRUE(detailed_item->detailed_view() != NULL);
 237   ASSERT_TRUE(test_item->notification_view() != NULL);
 238
 239   // Hide the detailed view, ensure the notificaiton view still exists.
 240   ASSERT_TRUE(tray->CloseBubbleForTest());
 241   RunAllPendingInMessageLoop();
 242   ASSERT_TRUE(detailed_item->detailed_view() == NULL);
 243   ASSERT_TRUE(test_item->notification_view() != NULL);
 244 }
 245
 246 TEST_F(SystemTrayTest, BubbleCreationTypesTest) {
 247   SystemTray* tray = GetSystemTray();
 248   ASSERT_TRUE(tray->GetWidget());
 249
 250   TestItem* test_item = new TestItem;
 251   tray->AddTrayItem(test_item);
 252
 253   // Ensure the tray views are created.
 254   ASSERT_TRUE(test_item->tray_view() != NULL);
 255
 256   // Show the default view, ensure the notification view is destroyed.
 257   tray->ShowDefaultView(BUBBLE_CREATE_NEW);
 258   RunAllPendingInMessageLoop();
 259
 260   views::Widget* widget = test_item->default_view()->GetWidget();
 261   gfx::Rect bubble_bounds = widget->GetWindowBoundsInScreen();
 262
 263   tray->ShowDetailedView(test_item, 0, true, BUBBLE_USE_EXISTING);
 264   RunAllPendingInMessageLoop();
 265
 266   EXPECT_FALSE(test_item->default_view());
 267
 268   EXPECT_EQ(bubble_bounds.ToString(), test_item->detailed_view()->GetWidget()->
 269       GetWindowBoundsInScreen().ToString());
 270   EXPECT_EQ(widget, test_item->detailed_view()->GetWidget());
 271
 272   tray->ShowDefaultView(BUBBLE_USE_EXISTING);
 273   RunAllPendingInMessageLoop();
 274
 275   EXPECT_EQ(bubble_bounds.ToString(), test_item->default_view()->GetWidget()->
 276       GetWindowBoundsInScreen().ToString());
 277   EXPECT_EQ(widget, test_item->default_view()->GetWidget());
 278 }
 279
 280 }  // namespace test
 281 }  // namespace ash