Updates the display message in the uber tray.

ash/system/chromeos/tray_display.h

 // Copyright (c) 2012 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.
 
 #ifndef ASH_SYSTEM_CHROMEOS_TRAY_DISPLAY_H_
 #define ASH_SYSTEM_CHROMEOS_TRAY_DISPLAY_H_
 
+#include <map>
+
+#include "ash/ash_export.h"
 #include "ash/display/display_controller.h"
+#include "ash/display/display_info.h"
+#include "ash/system/tray/actionable_view.h"
 #include "ash/system/tray/system_tray_item.h"
+#include "base/strings/string16.h"
+
+namespace gfx {
+class Point;
+}
 
 namespace views {
+class ImageView;
+class Label;
 class View;
 }
 
 namespace ash {
+namespace test {
+class AshTestBase;
+}
+
 namespace internal {
 
-enum TrayDisplayMode {
-  TRAY_DISPLAY_SINGLE,
-  TRAY_DISPLAY_EXTENDED,
-  TRAY_DISPLAY_MIRRORED,
-  TRAY_DISPLAY_DOCKED,
+class DisplayNotificationView;
+
+class DisplayView : public ash::internal::ActionableView {
+ public:
+  explicit DisplayView(user::LoginStatus login_status);
+  virtual ~DisplayView();
+
+  void Update();
+
+  views::Label* label() { return label_; }
+
+  // Overridden from views::View.
+  virtual bool GetTooltipText(const gfx::Point& p,
+                              string16* tooltip) const OVERRIDE;
+
+ private:
+  base::string16 GetInternalDisplayInfo() const;
+
+  // Overridden from ActionableView.
+  virtual bool PerformAction(const ui::Event& event) OVERRIDE;
+  virtual void OnBoundsChanged(const gfx::Rect& previous_bounds) OVERRIDE;
+
+  user::LoginStatus login_status_;
+  views::ImageView* image_;
+  views::Label* label_;
+
+  DISALLOW_COPY_AND_ASSIGN(DisplayView);
 };
 
-class DisplayView;
-class DisplayNotificationView;
-
-class TrayDisplay : public SystemTrayItem,
-                    public DisplayController::Observer {
+class ASH_EXPORT TrayDisplay : public SystemTrayItem,
+                               public DisplayController::Observer {
  public:
   explicit TrayDisplay(SystemTray* system_tray);
   virtual ~TrayDisplay();
 
  private:
+  friend class test::AshTestBase;
+  friend class TrayDisplayTest;
+
+  // Checks the current display settings and determine what message should be
+  // shown for notification.
+  base::string16 GetDisplayMessageForNotification();
+
   // Overridden from SystemTrayItem.
   virtual views::View* CreateDefaultView(user::LoginStatus status) OVERRIDE;
   virtual views::View* CreateNotificationView(
       user::LoginStatus status) OVERRIDE;
   virtual void DestroyDefaultView() OVERRIDE;
   virtual void DestroyNotificationView() OVERRIDE;
   virtual bool ShouldShowLauncher() const OVERRIDE;
 
   // Overridden from DisplayControllerObserver:
   virtual void OnDisplayConfigurationChanged() OVERRIDE;
 
+  // Call this to true when the test case wants to see the display notification.
+  // Default is false on test.
+  static void SetDisplayNotificationsEnabledForTest(bool enabled);
+
+  // Test utilities.
+  DisplayView* default_view() { return default_; }
+  const string16& current_message() const { return current_message_; }
+
   DisplayView* default_;
   DisplayNotificationView* notification_;
-  TrayDisplayMode current_mode_;
+  string16 current_message_;
+  std::map<int64, DisplayInfo> display_info_;
 
   DISALLOW_COPY_AND_ASSIGN(TrayDisplay);
 };
 
 }  // namespace internal
 }  // namespace ash
 
 #endif  // ASH_SYSTEM_CHROMEOS_TRAY_DISPLAY_H_