holding_space_test_api.cc 7.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214
  1. // Copyright 2020 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "ash/public/cpp/holding_space/holding_space_test_api.h"
  5. #include "ash/drag_drop/drag_drop_controller.h"
  6. #include "ash/public/cpp/holding_space/holding_space_constants.h"
  7. #include "ash/public/cpp/holding_space/holding_space_item.h"
  8. #include "ash/public/cpp/test/shell_test_api.h"
  9. #include "ash/shelf/shelf.h"
  10. #include "ash/shelf/shelf_widget.h"
  11. #include "ash/shell.h"
  12. #include "ash/system/holding_space/holding_space_item_chip_view.h"
  13. #include "ash/system/holding_space/holding_space_item_screen_capture_view.h"
  14. #include "ash/system/holding_space/holding_space_item_view.h"
  15. #include "ash/system/holding_space/holding_space_tray.h"
  16. #include "ash/system/status_area_widget.h"
  17. #include "ui/aura/window.h"
  18. #include "ui/events/test/event_generator.h"
  19. #include "ui/views/view.h"
  20. #include "ui/views/widget/widget.h"
  21. namespace ash {
  22. namespace {
  23. // Helpers ---------------------------------------------------------------------
  24. // Finds all descendents of `parent` matching a specific class.
  25. template <typename T>
  26. void FindDescendentsOfClass(views::View* parent,
  27. std::vector<views::View*>* descendents) {
  28. if (parent) {
  29. for (auto* child : parent->children()) {
  30. if (child->GetClassName() == T::kViewClassName)
  31. descendents->push_back(child);
  32. FindDescendentsOfClass<T>(child, descendents);
  33. }
  34. }
  35. }
  36. // Performs a tap on the specified `view`.
  37. void TapOn(const views::View* view) {
  38. ui::test::EventGenerator event_generator(Shell::GetRootWindowForNewWindows());
  39. event_generator.MoveTouch(view->GetBoundsInScreen().CenterPoint());
  40. event_generator.PressTouch();
  41. event_generator.ReleaseTouch();
  42. }
  43. } // namespace
  44. // HoldingSpaceTestApi ---------------------------------------------------------
  45. HoldingSpaceTestApi::HoldingSpaceTestApi()
  46. : holding_space_tray_(Shelf::ForWindow(Shell::GetRootWindowForNewWindows())
  47. ->shelf_widget()
  48. ->status_area_widget()
  49. ->holding_space_tray()) {
  50. holding_space_tray_->set_use_zero_previews_update_delay_for_testing(true);
  51. // Holding space tests perform drag/drop so we need to disable blocking.
  52. auto* drag_drop_controller = ShellTestApi().drag_drop_controller();
  53. drag_drop_controller->set_should_block_during_drag_drop(false);
  54. }
  55. HoldingSpaceTestApi::~HoldingSpaceTestApi() {
  56. if (!Shell::HasInstance())
  57. return;
  58. holding_space_tray_->set_use_zero_previews_update_delay_for_testing(false);
  59. // Enable blocking during drag/drop that was disabled for holding space tests.
  60. auto* drag_drop_controller = ShellTestApi().drag_drop_controller();
  61. drag_drop_controller->set_should_block_during_drag_drop(true);
  62. }
  63. // static
  64. aura::Window* HoldingSpaceTestApi::GetRootWindowForNewWindows() {
  65. return Shell::GetRootWindowForNewWindows();
  66. }
  67. void HoldingSpaceTestApi::Show() {
  68. if (!IsShowing())
  69. TapOn(holding_space_tray_);
  70. }
  71. void HoldingSpaceTestApi::Close() {
  72. if (IsShowing())
  73. TapOn(holding_space_tray_);
  74. }
  75. bool HoldingSpaceTestApi::IsShowing() {
  76. return holding_space_tray_ && holding_space_tray_->GetBubbleView() &&
  77. holding_space_tray_->GetBubbleView()->GetVisible();
  78. }
  79. bool HoldingSpaceTestApi::IsShowingInShelf() {
  80. return holding_space_tray_ && holding_space_tray_->GetVisible();
  81. }
  82. const base::FilePath& HoldingSpaceTestApi::GetHoldingSpaceItemFilePath(
  83. const views::View* item_view) const {
  84. return HoldingSpaceItemView::Cast(item_view)->item()->file_path();
  85. }
  86. const std::string& HoldingSpaceTestApi::GetHoldingSpaceItemId(
  87. const views::View* item_view) const {
  88. return HoldingSpaceItemView::Cast(item_view)->item_id();
  89. }
  90. views::View* HoldingSpaceTestApi::GetHoldingSpaceItemView(
  91. const std::vector<views::View*>& item_views,
  92. const std::string& item_id) {
  93. auto it = std::find_if(
  94. item_views.begin(), item_views.end(), [&](const views::View* item_view) {
  95. return !strcmp(HoldingSpaceItemView::Cast(item_view)->item_id().c_str(),
  96. item_id.c_str());
  97. });
  98. return it != item_views.end() ? *it : nullptr;
  99. }
  100. std::vector<views::View*> HoldingSpaceTestApi::GetHoldingSpaceItemViews() {
  101. std::vector<views::View*> item_views;
  102. if (holding_space_tray_->bubble_for_testing()) {
  103. for (HoldingSpaceItemView*& item_view :
  104. holding_space_tray_->bubble_for_testing()
  105. ->GetHoldingSpaceItemViews()) {
  106. item_views.push_back(item_view);
  107. }
  108. }
  109. return item_views;
  110. }
  111. views::View* HoldingSpaceTestApi::GetDownloadsSectionHeader() {
  112. return holding_space_tray_->GetBubbleView()
  113. ? holding_space_tray_->GetBubbleView()->GetViewByID(
  114. kHoldingSpaceDownloadsSectionHeaderId)
  115. : nullptr;
  116. }
  117. std::vector<views::View*> HoldingSpaceTestApi::GetDownloadChips() {
  118. std::vector<views::View*> download_chips;
  119. if (holding_space_tray_->GetBubbleView()) {
  120. FindDescendentsOfClass<HoldingSpaceItemChipView>(
  121. holding_space_tray_->GetBubbleView()->GetViewByID(
  122. kHoldingSpaceRecentFilesBubbleId),
  123. &download_chips);
  124. }
  125. return download_chips;
  126. }
  127. std::vector<views::View*> HoldingSpaceTestApi::GetPinnedFileChips() {
  128. std::vector<views::View*> pinned_file_chips;
  129. if (holding_space_tray_->GetBubbleView()) {
  130. FindDescendentsOfClass<HoldingSpaceItemChipView>(
  131. holding_space_tray_->GetBubbleView()->GetViewByID(
  132. kHoldingSpacePinnedFilesBubbleId),
  133. &pinned_file_chips);
  134. }
  135. return pinned_file_chips;
  136. }
  137. std::vector<views::View*> HoldingSpaceTestApi::GetScreenCaptureViews() {
  138. std::vector<views::View*> screen_capture_views;
  139. if (holding_space_tray_->GetBubbleView()) {
  140. FindDescendentsOfClass<HoldingSpaceItemScreenCaptureView>(
  141. holding_space_tray_->GetBubbleView()->GetViewByID(
  142. kHoldingSpaceRecentFilesBubbleId),
  143. &screen_capture_views);
  144. }
  145. return screen_capture_views;
  146. }
  147. views::View* HoldingSpaceTestApi::GetTray() {
  148. return holding_space_tray_;
  149. }
  150. views::View* HoldingSpaceTestApi::GetTrayDropTargetOverlay() {
  151. return holding_space_tray_->GetViewByID(kHoldingSpaceTrayDropTargetOverlayId);
  152. }
  153. views::View* HoldingSpaceTestApi::GetDefaultTrayIcon() {
  154. return holding_space_tray_->GetViewByID(kHoldingSpaceTrayDefaultIconId);
  155. }
  156. views::View* HoldingSpaceTestApi::GetPreviewsTrayIcon() {
  157. return holding_space_tray_->GetViewByID(kHoldingSpaceTrayPreviewsIconId);
  158. }
  159. views::View* HoldingSpaceTestApi::GetPinnedFilesBubble() {
  160. if (!holding_space_tray_->GetBubbleView())
  161. return nullptr;
  162. return holding_space_tray_->GetBubbleView()->GetViewByID(
  163. kHoldingSpacePinnedFilesBubbleId);
  164. }
  165. bool HoldingSpaceTestApi::PinnedFilesBubbleShown() const {
  166. if (!holding_space_tray_->GetBubbleView())
  167. return false;
  168. views::View* bubble = holding_space_tray_->GetBubbleView()->GetViewByID(
  169. kHoldingSpacePinnedFilesBubbleId);
  170. return bubble && bubble->GetVisible();
  171. }
  172. bool HoldingSpaceTestApi::RecentFilesBubbleShown() const {
  173. if (!holding_space_tray_->GetBubbleView())
  174. return false;
  175. views::View* bubble = holding_space_tray_->GetBubbleView()->GetViewByID(
  176. kHoldingSpaceRecentFilesBubbleId);
  177. return bubble && bubble->GetVisible();
  178. }
  179. } // namespace ash