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);
   ~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_; }
 
+  // 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();
+
  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);
@@ skipping 92 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;
 
+  void SimulateViewClickedForTest(const std::string& id);

[achuithb, 2015/07/14 18:06:39]

I think it's worthwhile to add a function comment here.

id -> activity_id

[jdufault, 2015/07/15 17:35:01]

Done.

+
  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& id) {
+  for (auto& it : receiver_activity_map_) {
+    if (it.second == 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);
 }
 
+bool TrayCast::IsTrayInitializedForTest() const {
+  return default_ != nullptr;
+}
+
+bool TrayCast::IsTrayVisibleForTest() const {
+  return default_ != nullptr && default_->IsDrawn();
+}
+
+bool TrayCast::IsTrayCastViewVisibleForTest() const {
+  return default_ != nullptr &&
+         default_->ActiveChildView() == default_->cast_view();
+}
+
+bool TrayCast::IsTraySelectViewVisibleForTest() const {
+  return default_ != nullptr &&
+         default_->ActiveChildView() == default_->select_view();
+}
+
+void TrayCast::StartCastForTest(const std::string& id) {
+  if (detailed_ != nullptr)
+    detailed_->SimulateViewClickedForTest(id);
+}
+
+void TrayCast::StopCastForTest() {
+  default_->cast_view()->StopCasting();
+}
+
 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_,
@@ skipping 67 matching lines @@
   is_casting_ = started;
   UpdatePrimaryView();
 }
 
 void TrayCast::UpdateAfterShelfAlignmentChange(ShelfAlignment alignment) {
   if (tray_)
     tray_->UpdateAlignment(alignment);
 }
 
 }  // namespace ash