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"
@@ skipping 96 matching lines @@
 }
 
 // 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;
 
+  void StopCasting();
+
   // 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;
@@ skipping 72 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);
@@ skipping 120 matching lines @@
 // 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;
 
+  // Makes the detail view think the view associated with the given receiver_id
+  // was clicked. This will start a cast.
+  void SimulateViewClickedForTest(const std::string& receiver_id);
+
  private:
   void CreateItems();
 
   void UpdateReceiverList();
   void UpdateReceiverListCallback(
       const CastConfigDelegate::ReceiversAndActivites&
           new_receivers_and_activities);
   void UpdateReceiverListFromCachedData();
   views::View* AddToReceiverList(
       const CastConfigDelegate::ReceiverAndActivity& receiverActivity);
@@ skipping 22 matching lines @@
       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 98 matching lines @@
                                 ->system_tray_delegate()
                                 ->GetCastConfigDelegate()),
       weak_ptr_factory_(this) {
   Shell::GetInstance()->AddShellObserver(this);
 }
 
 TrayCast::~TrayCast() {
   Shell::GetInstance()->RemoveShellObserver(this);
 }
 
+void TrayCast::StartCastForTest(const std::string& receiver_id) {
+  if (detailed_ != nullptr)
+    detailed_->SimulateViewClickedForTest(receiver_id);
+}
+
+void TrayCast::StopCastForTest() {
+  default_->cast_view()->StopCasting();
+}
+
+const views::View* TrayCast::GetDefaultView() const {
+  return default_;
+}
+
 views::View* TrayCast::CreateTrayView(user::LoginStatus status) {
   CHECK(tray_ == nullptr);
   tray_ = new tray::CastTrayView(this);
   tray_->SetVisible(is_casting_);
   return tray_;
 }
 
 views::View* TrayCast::CreateDefaultView(user::LoginStatus status) {
   CHECK(default_ == nullptr);
+
   default_ = new tray::CastDuplexView(this, cast_config_delegate_,
                                       status != user::LOGGED_IN_LOCKED);
+  default_->set_id(TRAY_VIEW);
+  default_->select_view()->set_id(SELECT_VIEW);
+  default_->cast_view()->set_id(CAST_VIEW);
+
   UpdatePrimaryView();
   return default_;
 }
 
 views::View* TrayCast::CreateDetailedView(user::LoginStatus status) {
   Shell::GetInstance()->metrics()->RecordUserMetricsAction(
       ash::UMA_STATUS_AREA_DETAILED_CAST_VIEW);
   CHECK(detailed_ == nullptr);
   detailed_ = new tray::CastDetailedView(this, cast_config_delegate_, status);
   return detailed_;
@@ skipping 56 matching lines @@
   is_casting_ = started;
   UpdatePrimaryView();
 }
 
 void TrayCast::UpdateAfterShelfAlignmentChange(ShelfAlignment alignment) {
   if (tray_)
     tray_->UpdateAlignment(alignment);
 }
 
 }  // namespace ash