Notificaiton Message for Unified mode

ash/system/chromeos/tray_display_unittest.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "ash/system/chromeos/tray_display.h"
 
 #include "ash/display/display_manager.h"
 #include "ash/root_window_controller.h"
 #include "ash/screen_util.h"
 #include "ash/shell.h"
@@ skipping 61 matching lines @@
 }
 
 class TrayDisplayTest : public ash::test::AshTestBase {
  public:
   TrayDisplayTest();
   ~TrayDisplayTest() override;
 
   void SetUp() override;
 
  protected:
-  SystemTray* tray() { return tray_; }
-  TrayDisplay* tray_display() { return tray_display_; }
+  SystemTray* GetTray();
+  TrayDisplay* GetTrayDisplay();
+  void CheckUpdate();
 
   void CloseNotification();
   bool IsDisplayVisibleInTray() const;
   base::string16 GetTrayDisplayText() const;
   void CheckAccessibleName() const;
   base::string16 GetTrayDisplayTooltipText() const;
   base::string16 GetDisplayNotificationText() const;
   base::string16 GetDisplayNotificationAdditionalText() const;
 
  private:
   const message_center::Notification* GetDisplayNotification() const;
 
   // Weak reference, owned by Shell.
   SystemTray* tray_;
 
   // Weak reference, owned by |tray_|.
   TrayDisplay* tray_display_;
 
   DISALLOW_COPY_AND_ASSIGN(TrayDisplayTest);
 };
 
 TrayDisplayTest::TrayDisplayTest() : tray_(NULL), tray_display_(NULL) {
 }
 
 TrayDisplayTest::~TrayDisplayTest() {
 }
 
 void TrayDisplayTest::SetUp() {
   ash::test::AshTestBase::SetUp();
-  tray_ = Shell::GetPrimaryRootWindowController()->GetSystemTray();
-  tray_display_ = new TrayDisplay(tray_);
-  tray_->AddTrayItem(tray_display_);
+  // Populate tray_ and tray_display_.
+  CheckUpdate();
+}
+
+SystemTray* TrayDisplayTest::GetTray() {
+  CheckUpdate();
+  return tray_;
+}
+
+TrayDisplay* TrayDisplayTest::GetTrayDisplay() {
+  CheckUpdate();
+  return tray_display_;
+}
+
+void TrayDisplayTest::CheckUpdate() {
+  SystemTray* current =
+      Shell::GetPrimaryRootWindowController()->GetSystemTray();
+  if (tray_ != current) {
+    tray_ = current;
+    tray_display_ = new TrayDisplay(tray_);
+    tray_->AddTrayItem(tray_display_);
+  }
 }
 
 void TrayDisplayTest::CloseNotification() {
   message_center::MessageCenter::Get()->RemoveNotification(
       TrayDisplay::kNotificationId, false);
   RunAllPendingInMessageLoop();
 }
 
 bool TrayDisplayTest::IsDisplayVisibleInTray() const {
   return tray_->HasSystemBubble() &&
@@ skipping 41 matching lines @@
            notifications.begin(); iter != notifications.end(); ++iter) {
     if ((*iter)->id() == TrayDisplay::kNotificationId)
       return *iter;
   }
 
   return NULL;
 }
 
 TEST_F(TrayDisplayTest, NoInternalDisplay) {
   UpdateDisplay("400x400");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_FALSE(IsDisplayVisibleInTray());
 
   UpdateDisplay("400x400,200x200");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   base::string16 expected = l10n_util::GetStringUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED_NO_INTERNAL);
   base::string16 first_name = GetFirstDisplayName();
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(expected, GetFirstDisplayName(), "400x400",
                            GetSecondDisplayName(), "200x200"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   // mirroring
   Shell::GetInstance()->display_manager()->SetSoftwareMirroring(true);
   UpdateDisplay("400x400,200x200");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   expected = l10n_util::GetStringUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_MIRRORING_NO_INTERNAL);
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetMirroredTooltipText(expected, GetFirstDisplayName(), "400x400"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 }
 
 TEST_F(TrayDisplayTest, InternalDisplay) {
   UpdateDisplay("400x400");
   DisplayManager* display_manager = Shell::GetInstance()->display_manager();
   gfx::Display::SetInternalDisplayId(display_manager->first_display_id());
 
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_FALSE(IsDisplayVisibleInTray());
 
   // Extended
   UpdateDisplay("400x400,200x200");
   base::string16 expected = l10n_util::GetStringFUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED, GetSecondDisplayName());
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(expected, GetFirstDisplayName(), "400x400",
                            GetSecondDisplayName(), "200x200"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   // Mirroring
   display_manager->SetSoftwareMirroring(true);
   UpdateDisplay("400x400,200x200");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
 
   expected = l10n_util::GetStringFUTF16(IDS_ASH_STATUS_TRAY_DISPLAY_MIRRORING,
                                         GetMirroringDisplayName());
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetMirroredTooltipText(expected, GetFirstDisplayName(), "400x400"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 }
 
 TEST_F(TrayDisplayTest, InternalDisplayResized) {
   UpdateDisplay("400x400@1.5");
   DisplayManager* display_manager = Shell::GetInstance()->display_manager();
   gfx::Display::SetInternalDisplayId(display_manager->first_display_id());
 
   // Shows the tray_display even though there's a single-display.
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   base::string16 internal_info = l10n_util::GetStringFUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_SINGLE_DISPLAY,
       GetFirstDisplayName(), base::UTF8ToUTF16("600x600"));
   EXPECT_EQ(internal_info, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(base::string16(), GetFirstDisplayName(), "600x600",
                            base::string16(), std::string()),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   // Extended
   UpdateDisplay("400x400@1.5,200x200");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   base::string16 expected = l10n_util::GetStringFUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED, GetSecondDisplayName());
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(expected, GetFirstDisplayName(), "600x600",
                            GetSecondDisplayName(), "200x200"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   // Mirroring
   display_manager->SetSoftwareMirroring(true);
   UpdateDisplay("400x400@1.5,200x200");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   expected = l10n_util::GetStringFUTF16(IDS_ASH_STATUS_TRAY_DISPLAY_MIRRORING,
                                         GetMirroringDisplayName());
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetMirroredTooltipText(expected, GetFirstDisplayName(), "600x600"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   // Closed lid mode.
   display_manager->SetSoftwareMirroring(false);
   UpdateDisplay("400x400@1.5,200x200");
   gfx::Display::SetInternalDisplayId(ScreenUtil::GetSecondaryDisplay().id());
   UpdateDisplay("400x400@1.5");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   expected = l10n_util::GetStringUTF16(IDS_ASH_STATUS_TRAY_DISPLAY_DOCKED);
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(
       GetTooltipText(
           expected, GetFirstDisplayName(), "600x600", base::string16(), ""),
       GetTrayDisplayTooltipText());
   CheckAccessibleName();
+
+  // Unified mode
+  display_manager->SetDefaultMultiDisplayMode(DisplayManager::UNIFIED);
+  display_manager->SetMultiDisplayMode(DisplayManager::UNIFIED);
+  UpdateDisplay("300x200,400x500");
+  // Update the cache variables as the primary root window changed.
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  EXPECT_TRUE(IsDisplayVisibleInTray());
+  expected = l10n_util::GetStringUTF16(IDS_ASH_STATUS_TRAY_DISPLAY_UNIFIED);
+  EXPECT_EQ(expected, GetTrayDisplayText());
 }
 
 TEST_F(TrayDisplayTest, ExternalDisplayResized) {
   UpdateDisplay("400x400");
   DisplayManager* display_manager = Shell::GetInstance()->display_manager();
   gfx::Display::SetInternalDisplayId(display_manager->first_display_id());
 
   // Shows the tray_display even though there's a single-display.
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_FALSE(IsDisplayVisibleInTray());
 
   // Extended
   UpdateDisplay("400x400,200x200@1.5");
   const gfx::Display& secondary_display = ScreenUtil::GetSecondaryDisplay();
 
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   base::string16 expected = l10n_util::GetStringFUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED,
       l10n_util::GetStringFUTF16(
           IDS_ASH_STATUS_TRAY_DISPLAY_ANNOTATED_NAME,
           GetSecondDisplayName(),
           base::UTF8ToUTF16(secondary_display.size().ToString())));
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(expected, GetFirstDisplayName(), "400x400",
                            GetSecondDisplayName(), "300x300"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   // Mirroring
   display_manager->SetSoftwareMirroring(true);
   UpdateDisplay("400x400,200x200@1.5");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
   expected = l10n_util::GetStringFUTF16(IDS_ASH_STATUS_TRAY_DISPLAY_MIRRORING,
                                         GetMirroringDisplayName());
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetMirroredTooltipText(expected, GetFirstDisplayName(), "400x400"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 }
 
 TEST_F(TrayDisplayTest, OverscanDisplay) {
   UpdateDisplay("400x400,300x300/o");
   DisplayManager* display_manager = Shell::GetInstance()->display_manager();
   gfx::Display::SetInternalDisplayId(display_manager->first_display_id());
 
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_TRUE(IsDisplayVisibleInTray());
 
   // /o creates the default overscan, and if overscan is set, the annotation
   // should be the size.
   base::string16 overscan = l10n_util::GetStringUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_ANNOTATION_OVERSCAN);
   base::string16 headline = l10n_util::GetStringFUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED,
       l10n_util::GetStringFUTF16(
           IDS_ASH_STATUS_TRAY_DISPLAY_ANNOTATED_NAME,
@@ skipping 15 matching lines @@
           GetSecondDisplayName(), overscan));
   second_data = l10n_util::GetStringFUTF8(
       IDS_ASH_STATUS_TRAY_DISPLAY_ANNOTATION,
       base::UTF8ToUTF16("300x300"), overscan);
   EXPECT_EQ(GetTooltipText(headline, GetFirstDisplayName(), "400x400",
                            GetSecondDisplayName(), second_data),
             GetTrayDisplayTooltipText());
 }
 
 TEST_F(TrayDisplayTest, UpdateDuringDisplayConfigurationChange) {
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
   EXPECT_FALSE(IsDisplayVisibleInTray());
 
   UpdateDisplay("400x400@1.5");
-  EXPECT_TRUE(tray()->HasSystemBubble());
+  EXPECT_TRUE(GetTray()->HasSystemBubble());
   EXPECT_TRUE(IsDisplayVisibleInTray());
   base::string16 internal_info = l10n_util::GetStringFUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_SINGLE_DISPLAY,
       GetFirstDisplayName(), base::UTF8ToUTF16("600x600"));
   EXPECT_EQ(internal_info, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(base::string16(), GetFirstDisplayName(), "600x600",
                            base::string16(), std::string()),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   UpdateDisplay("400x400,200x200");
-  EXPECT_TRUE(tray()->HasSystemBubble());
+  EXPECT_TRUE(GetTray()->HasSystemBubble());
   EXPECT_TRUE(IsDisplayVisibleInTray());
   base::string16 expected = l10n_util::GetStringUTF16(
       IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED_NO_INTERNAL);
   base::string16 first_name = GetFirstDisplayName();
   EXPECT_EQ(expected, GetTrayDisplayText());
   EXPECT_EQ(GetTooltipText(expected, GetFirstDisplayName(), "400x400",
                            GetSecondDisplayName(), "200x200"),
             GetTrayDisplayTooltipText());
   CheckAccessibleName();
 
   UpdateDisplay("400x400@1.5");
-  tray()->ShowDefaultView(BUBBLE_USE_EXISTING);
+  GetTray()->ShowDefaultView(BUBBLE_USE_EXISTING);
 
   // Back to the default state, the display tray item should disappear.
   UpdateDisplay("400x400");
-  EXPECT_TRUE(tray()->HasSystemBubble());
+  EXPECT_TRUE(GetTray()->HasSystemBubble());
   EXPECT_FALSE(IsDisplayVisibleInTray());
 }
 
 TEST_F(TrayDisplayTest, DisplayNotifications) {
   test::TestSystemTrayDelegate* tray_delegate =
       static_cast<test::TestSystemTrayDelegate*>(
           Shell::GetInstance()->system_tray_delegate());
   tray_delegate->set_should_show_display_notification(true);
 
   UpdateDisplay("400x400");
@@ skipping 107 matching lines @@
 
   UpdateDisplay("400x400,200x200");
   EXPECT_EQ(
       l10n_util::GetStringUTF16(
           IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED_NO_INTERNAL),
       GetDisplayNotificationText());
 
   // OnDisplayConfigurationChanged() may be called more than once for a single
   // update display in case of primary is swapped or recovered from dock mode.
   // Should not remove the notification in such case.
-  tray_display()->OnDisplayConfigurationChanged();
+  GetTrayDisplay()->OnDisplayConfigurationChanged();
   EXPECT_EQ(
       l10n_util::GetStringUTF16(
           IDS_ASH_STATUS_TRAY_DISPLAY_EXTENDED_NO_INTERNAL),
       GetDisplayNotificationText());
 
   // Back to the single display. It SHOULD remove the notification since the
   // information is stale.
   UpdateDisplay("400x400");
   EXPECT_TRUE(GetDisplayNotificationText().empty());
 }
 
 TEST_F(TrayDisplayTest, UpdateAfterSuppressDisplayNotification) {
   UpdateDisplay("400x400,200x200");
 
   test::TestSystemTrayDelegate* tray_delegate =
       static_cast<test::TestSystemTrayDelegate*>(
           Shell::GetInstance()->system_tray_delegate());
   tray_delegate->set_should_show_display_notification(true);
 
   // rotate the second.
   UpdateDisplay("400x400,200x200/r");
   EXPECT_EQ(l10n_util::GetStringFUTF16(
                 IDS_ASH_STATUS_TRAY_DISPLAY_ROTATED, GetSecondDisplayName(),
                 l10n_util::GetStringUTF16(
                     IDS_ASH_STATUS_TRAY_DISPLAY_ORIENTATION_90)),
             GetDisplayNotificationAdditionalText());
 }
 
 }  // namespace ash