Add support code to test the cast system tray item.

ash/system/cast/tray_cast.cc

 // Copyright 2015 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/cast/tray_cast.h"
 
 #include "ash/session/session_state_delegate.h"
 #include "ash/shelf/shelf_types.h"
 #include "ash/shell.h"
 #include "ash/system/chromeos/screen_security/screen_tray_item.h"
 #include "ash/system/tray/fixed_sized_image_view.h"
 #include "ash/system/tray/fixed_sized_scroll_view.h"
 #include "ash/system/tray/hover_highlight_view.h"
 #include "ash/system/tray/system_tray.h"
 #include "ash/system/tray/system_tray_delegate.h"
 #include "ash/system/tray/system_tray_notifier.h"
 #include "ash/system/tray/throbber_view.h"
 #include "ash/system/tray/tray_constants.h"
-#include "ash/system/tray/tray_details_view.h"
-#include "ash/system/tray/tray_item_more.h"
 #include "ash/system/tray/tray_item_view.h"
-#include "ash/system/tray/tray_popup_label_button.h"
 #include "base/bind.h"
 #include "grit/ash_resources.h"
 #include "grit/ash_strings.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/gfx/image/image.h"
-#include "ui/views/controls/button/button.h"
 #include "ui/views/controls/image_view.h"
 #include "ui/views/controls/label.h"
 #include "ui/views/layout/box_layout.h"
 #include "ui/views/layout/fill_layout.h"
 
 namespace ash {
 
 namespace {
 const int kStopButtonRightPadding = 18;
 }  // namespace
 
 namespace tray {
 
-// This view is displayed in the system tray when the cast extension is active.
-// It asks the user if they want to cast the desktop. If they click on the
-// chevron, then a detail view will replace this view where the user will
-// actually pick the cast receiver.
-class CastSelectDefaultView : public TrayItemMore {
- public:
-  CastSelectDefaultView(SystemTrayItem* owner,
-                        CastConfigDelegate* cast_config_delegate,
-                        bool show_more);
-  ~CastSelectDefaultView() override;
-
-  // Updates the label based on the current set of receivers (if there are or
-  // are not any available receivers).
-  void UpdateLabel();
-
- private:
-  void UpdateLabelCallback(
-      const CastConfigDelegate::ReceiversAndActivites& receivers_activities);
-
-  CastConfigDelegate* cast_config_delegate_;
-  base::WeakPtrFactory<CastSelectDefaultView> weak_ptr_factory_;
-  DISALLOW_COPY_AND_ASSIGN(CastSelectDefaultView);
-};
-
 CastSelectDefaultView::CastSelectDefaultView(
     SystemTrayItem* owner,
     CastConfigDelegate* cast_config_delegate,
     bool show_more)
     : TrayItemMore(owner, show_more),
       cast_config_delegate_(cast_config_delegate),
       weak_ptr_factory_(this) {
   ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
   SetImage(rb.GetImageNamed(IDR_AURA_UBER_TRAY_CAST).ToImageSkia());
 
@@ skipping 22 matching lines @@
 void CastSelectDefaultView::UpdateLabel() {
   if (cast_config_delegate_ == nullptr ||
       cast_config_delegate_->HasCastExtension() == false)
     return;
 
   cast_config_delegate_->GetReceiversAndActivities(
       base::Bind(&CastSelectDefaultView::UpdateLabelCallback,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
-// This view is displayed when the screen is actively being casted; it allows
-// the user to easily stop casting. It fully replaces the
-// |CastSelectDefaultView| view inside of the |CastDuplexView|.
-class CastCastView : public views::View, public views::ButtonListener {
- public:
-  explicit CastCastView(CastConfigDelegate* cast_config_delegate);
-  ~CastCastView() override;
-
-  // Updates the label for the stop view to include information about the
-  // current device that is being casted.
-  void UpdateLabel();
-
- private:
-  void UpdateLabelCallback(
-      const CastConfigDelegate::ReceiversAndActivites& receivers_activities);
-
-  // Overridden from views::View.
-  int GetHeightForWidth(int width) const override;
-  void Layout() override;
-
-  // Overridden from views::ButtonListener.
-  void ButtonPressed(views::Button* sender, const ui::Event& event) override;
-
-  CastConfigDelegate* cast_config_delegate_;
-  views::ImageView* icon_;
-  views::Label* label_;
-  TrayPopupLabelButton* stop_button_;
-  base::WeakPtrFactory<CastCastView> weak_ptr_factory_;
-
-  DISALLOW_COPY_AND_ASSIGN(CastCastView);
-};
-
 CastCastView::CastCastView(CastConfigDelegate* cast_config_delegate)
     : cast_config_delegate_(cast_config_delegate), weak_ptr_factory_(this) {
   // We will initialize the primary tray view which shows a stop button here.
 
   set_background(views::Background::CreateSolidBackground(kBackgroundColor));
   ui::ResourceBundle& bundle = ui::ResourceBundle::GetSharedInstance();
   SetLayoutManager(new views::BoxLayout(views::BoxLayout::kHorizontal,
                                         kTrayPopupPaddingHorizontal, 0,
                                         kTrayPopupPaddingBetweenItems));
   icon_ = new FixedSizedImageView(0, kTrayPopupItemHeight);
@@ skipping 48 matching lines @@
 
   // Adjust the label's bounds in case it got cut off by |stop_button_|.
   if (label_->bounds().Intersects(stop_button_->bounds())) {
     gfx::Rect label_bounds = label_->bounds();
     label_bounds.set_width(stop_button_->x() - kTrayPopupPaddingBetweenItems -
                            label_->x());
     label_->SetBoundsRect(label_bounds);
   }
 }
 
+void CastCastView::StopCasting() {
+  cast_config_delegate_->StopCasting();
+  Shell::GetInstance()->metrics()->RecordUserMetricsAction(
+      ash::UMA_STATUS_AREA_CAST_STOP_CAST);
+}
+
 void CastCastView::UpdateLabel() {
   if (cast_config_delegate_ == nullptr ||
       cast_config_delegate_->HasCastExtension() == false)
     return;
 
   cast_config_delegate_->GetReceiversAndActivities(base::Bind(
       &CastCastView::UpdateLabelCallback, weak_ptr_factory_.GetWeakPtr()));
 }
 
 void CastCastView::UpdateLabelCallback(
@@ skipping 19 matching lines @@
       PreferredSizeChanged();
       Layout();
       break;
     }
   }
 }
 
 void CastCastView::ButtonPressed(views::Button* sender,
                                  const ui::Event& event) {
   DCHECK(sender == stop_button_);
-  cast_config_delegate_->StopCasting();
-  Shell::GetInstance()->metrics()->RecordUserMetricsAction(
-      ash::UMA_STATUS_AREA_CAST_STOP_CAST);
+  StopCasting();
 }
 
-// This view by itself does very little. It acts as a front-end for managing
-// which of the two child views (|CastSelectDefaultView| and |CastCastView|)
-// is active.
-class CastDuplexView : public views::View {
- public:
-  CastDuplexView(SystemTrayItem* owner,
-                 CastConfigDelegate* config_delegate,
-                 bool show_more);
-  ~CastDuplexView() override;
-
-  // Activate either the casting or select view.
-  void ActivateCastView();
-  void ActivateSelectView();
-
-  CastSelectDefaultView* select_view() { return select_view_; }
-  CastCastView* cast_view() { return cast_view_; }
-
- private:
-  // Overridden from views::View.
-  void ChildPreferredSizeChanged(views::View* child) override;
-  void Layout() override;
-
-  // Only one of |select_view_| or |cast_view_| will be displayed at any given
-  // time. This will return the view is being displayed.
-  views::View* ActiveChildView();
-
-  CastSelectDefaultView* select_view_;
-  CastCastView* cast_view_;
-
-  DISALLOW_COPY_AND_ASSIGN(CastDuplexView);
-};
-
 CastDuplexView::CastDuplexView(SystemTrayItem* owner,
                                CastConfigDelegate* config_delegate,
                                bool show_more) {
   select_view_ = new CastSelectDefaultView(owner, config_delegate, show_more);
   cast_view_ = new CastCastView(config_delegate);
   SetLayoutManager(new views::FillLayout());
 
   ActivateSelectView();
 }
 
@@ skipping 76 matching lines @@
       break;
     case ash::SHELF_ALIGNMENT_LEFT:
     case ash::SHELF_ALIGNMENT_RIGHT:
       layout = views::BoxLayout::kVertical;
       break;
   }
   SetLayoutManager(new views::BoxLayout(layout, 0, 0, 0));
   Layout();
 }
 
-// This view displays a list of cast receivers that can be clicked on and casted
-// to. It is activated by clicking on the chevron inside of
-// |CastSelectDefaultView|.
-class CastDetailedView : public TrayDetailsView, public ViewClickListener {
- public:
-  CastDetailedView(SystemTrayItem* owner,
-                   CastConfigDelegate* cast_config_delegate,
-                   user::LoginStatus login);
-  ~CastDetailedView() override;
-
- private:
-  void CreateItems();
-
-  void UpdateReceiverList();
-  void UpdateReceiverListCallback(
-      const CastConfigDelegate::ReceiversAndActivites&
-          new_receivers_and_activities);
-  void UpdateReceiverListFromCachedData();
-  views::View* AddToReceiverList(
-      const CastConfigDelegate::ReceiverAndActivity& receiverActivity);
-
-  void AppendSettingsEntries();
-  void AppendHeaderEntry();
-
-  // Overridden from ViewClickListener.
-  void OnViewClicked(views::View* sender) override;
-
-  CastConfigDelegate* cast_config_delegate_;
-  user::LoginStatus login_;
-  views::View* options_ = nullptr;
-  CastConfigDelegate::ReceiversAndActivites receivers_and_activities_;
-  // A mapping from the view pointer to the associated activity id
-  std::map<views::View*, std::string> receiver_activity_map_;
-  base::WeakPtrFactory<CastDetailedView> weak_ptr_factory_;
-
-  DISALLOW_COPY_AND_ASSIGN(CastDetailedView);
-};
-
 CastDetailedView::CastDetailedView(SystemTrayItem* owner,
                                    CastConfigDelegate* cast_config_delegate,
                                    user::LoginStatus login)
     : TrayDetailsView(owner),
       cast_config_delegate_(cast_config_delegate),
       login_(login),
       weak_ptr_factory_(this) {
   CreateItems();
   UpdateReceiverList();
 }
 
 CastDetailedView::~CastDetailedView() {
 }
 
+void CastDetailedView::SimulateViewClickedForTest(
+    const std::string& receiver_id) {
+  for (auto& it : receiver_activity_map_) {
+    if (it.second == receiver_id) {
+      OnViewClicked(it.first);
+      break;
+    }
+  }
+}
+
 void CastDetailedView::CreateItems() {
   CreateScrollableList();
   AppendSettingsEntries();
   AppendHeaderEntry();
 }
 
 void CastDetailedView::UpdateReceiverList() {
   cast_config_delegate_->GetReceiversAndActivities(
       base::Bind(&CastDetailedView::UpdateReceiverListCallback,
                  weak_ptr_factory_.GetWeakPtr()));
@@ skipping 185 matching lines @@
   is_casting_ = started;
   UpdatePrimaryView();
 }
 
 void TrayCast::UpdateAfterShelfAlignmentChange(ShelfAlignment alignment) {
   if (tray_)
     tray_->UpdateAlignment(alignment);
 }
 
 }  // namespace ash