| // Copyright (c) 2013 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/touch/touch_observer_hud.h" |
| |
| #include "ash/public/cpp/ash_switches.h" |
| #include "ash/root_window_controller.h" |
| #include "ash/shell.h" |
| #include "ash/test/ash_test_base.h" |
| #include "ash/touch/touch_devices_controller.h" |
| #include "ash/touch/touch_observer_hud.h" |
| #include "base/command_line.h" |
| #include "base/format_macros.h" |
| #include "base/strings/stringprintf.h" |
| #include "ui/aura/window.h" |
| #include "ui/display/manager/display_manager.h" |
| #include "ui/display/test/display_manager_test_api.h" |
| #include "ui/views/widget/widget.h" |
| |
| namespace ash { |
| |
| class TouchObserverHUDTest : public AshTestBase { |
| public: |
| TouchObserverHUDTest() = default; |
| ~TouchObserverHUDTest() override = default; |
| |
| void SetUp() override { |
| // Add ash-touch-hud flag to enable debug touch HUD. This flag should be set |
| // before Ash environment is set up. |
| base::CommandLine::ForCurrentProcess()->AppendSwitch( |
| ash::switches::kAshTouchHud); |
| |
| AshTestBase::SetUp(); |
| |
| // Initialize display infos. They should be initialized after Ash |
| // environment is set up, i.e., after AshTestBase::SetUp(). |
| internal_display_id_ = |
| display::test::DisplayManagerTestApi(Shell::Get()->display_manager()) |
| .SetFirstDisplayAsInternalDisplay(); |
| external_display_id_ = 10; |
| mirrored_display_id_ = 11; |
| |
| internal_display_info_ = |
| CreateDisplayInfo(internal_display_id_, gfx::Rect(0, 0, 500, 500)); |
| external_display_info_ = |
| CreateDisplayInfo(external_display_id_, gfx::Rect(1, 1, 100, 100)); |
| mirrored_display_info_ = |
| CreateDisplayInfo(mirrored_display_id_, gfx::Rect(0, 0, 100, 100)); |
| } |
| |
| display::Display GetPrimaryDisplay() { |
| return display::Screen::GetScreen()->GetPrimaryDisplay(); |
| } |
| |
| void SetupSingleDisplay() { |
| display_info_list_.clear(); |
| display_info_list_.push_back(internal_display_info_); |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void SetupDualDisplays() { |
| display_info_list_.clear(); |
| display_info_list_.push_back(internal_display_info_); |
| display_info_list_.push_back(external_display_info_); |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void SetInternalAsPrimary() { |
| GetWindowTreeHostManager()->SetPrimaryDisplayId(internal_display_id_); |
| } |
| |
| void SetExternalAsPrimary() { |
| GetWindowTreeHostManager()->SetPrimaryDisplayId(external_display_id_); |
| } |
| |
| void MirrorDisplays() { |
| DCHECK_EQ(2U, display_info_list_.size()); |
| DCHECK_EQ(internal_display_id_, display_info_list_[0].id()); |
| DCHECK_EQ(external_display_id_, display_info_list_[1].id()); |
| display_info_list_[1] = mirrored_display_info_; |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void UnmirrorDisplays() { |
| DCHECK_EQ(2U, display_info_list_.size()); |
| DCHECK_EQ(internal_display_id_, display_info_list_[0].id()); |
| DCHECK_EQ(mirrored_display_id_, display_info_list_[1].id()); |
| display_info_list_[1] = external_display_info_; |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void RemoveInternalDisplay() { |
| DCHECK_LT(0U, display_info_list_.size()); |
| DCHECK_EQ(internal_display_id_, display_info_list_[0].id()); |
| display_info_list_.erase(display_info_list_.begin()); |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void RemoveExternalDisplay() { |
| DCHECK_EQ(2U, display_info_list_.size()); |
| display_info_list_.pop_back(); |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void AddInternalDisplay() { |
| DCHECK_EQ(0U, display_info_list_.size()); |
| display_info_list_.push_back(internal_display_info_); |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void AddExternalDisplay() { |
| DCHECK_EQ(1U, display_info_list_.size()); |
| display_info_list_.push_back(external_display_info_); |
| display_manager()->OnNativeDisplaysChanged(display_info_list_); |
| } |
| |
| void CheckInternalDisplay() { |
| ASSERT_TRUE(GetInternalTouchHud()); |
| EXPECT_EQ(internal_display_id(), GetInternalTouchHud()->display_id_); |
| EXPECT_EQ(GetInternalRootWindow(), |
| GetRootWindowForTouchHud(GetInternalTouchHud())); |
| EXPECT_EQ(GetInternalRootWindow(), |
| GetWidgetForTouchHud(GetInternalTouchHud()) |
| ->GetNativeView() |
| ->GetRootWindow()); |
| EXPECT_EQ(GetInternalDisplay().size(), |
| GetWidgetForTouchHud(GetInternalTouchHud()) |
| ->GetWindowBoundsInScreen() |
| .size()); |
| } |
| |
| void CheckExternalDisplay() { |
| ASSERT_TRUE(GetExternalTouchHud()); |
| EXPECT_EQ(external_display_id(), GetExternalTouchHud()->display_id_); |
| EXPECT_EQ(GetExternalRootWindow(), |
| GetRootWindowForTouchHud(GetExternalTouchHud())); |
| EXPECT_EQ(GetExternalRootWindow(), |
| GetWidgetForTouchHud(GetExternalTouchHud()) |
| ->GetNativeView() |
| ->GetRootWindow()); |
| EXPECT_EQ(GetExternalDisplay().size(), |
| GetWidgetForTouchHud(GetExternalTouchHud()) |
| ->GetWindowBoundsInScreen() |
| .size()); |
| } |
| |
| int64_t internal_display_id() const { return internal_display_id_; } |
| |
| int64_t external_display_id() const { return external_display_id_; } |
| |
| protected: |
| |
| WindowTreeHostManager* GetWindowTreeHostManager() { |
| return Shell::Get()->window_tree_host_manager(); |
| } |
| |
| const display::Display& GetInternalDisplay() { |
| return display_manager()->GetDisplayForId(internal_display_id_); |
| } |
| |
| const display::Display& GetExternalDisplay() { |
| return display_manager()->GetDisplayForId(external_display_id_); |
| } |
| |
| aura::Window* GetInternalRootWindow() { |
| return Shell::GetRootWindowForDisplayId(internal_display_id_); |
| } |
| |
| aura::Window* GetExternalRootWindow() { |
| return Shell::GetRootWindowForDisplayId(external_display_id_); |
| } |
| |
| aura::Window* GetPrimaryRootWindow() { |
| const display::Display& display = GetPrimaryDisplay(); |
| return Shell::GetRootWindowForDisplayId(display.id()); |
| } |
| |
| aura::Window* GetSecondaryRootWindow() { |
| const display::Display& display = display_manager()->GetSecondaryDisplay(); |
| return Shell::GetRootWindowForDisplayId(display.id()); |
| } |
| |
| RootWindowController* GetInternalRootController() { |
| aura::Window* root = GetInternalRootWindow(); |
| return RootWindowController::ForWindow(root); |
| } |
| |
| RootWindowController* GetExternalRootController() { |
| aura::Window* root = GetExternalRootWindow(); |
| return RootWindowController::ForWindow(root); |
| } |
| |
| RootWindowController* GetPrimaryRootController() { |
| aura::Window* root = GetPrimaryRootWindow(); |
| return RootWindowController::ForWindow(root); |
| } |
| |
| RootWindowController* GetSecondaryRootController() { |
| aura::Window* root = GetSecondaryRootWindow(); |
| return RootWindowController::ForWindow(root); |
| } |
| |
| TouchObserverHUD* GetInternalTouchHud() { |
| return GetInternalRootController()->touch_observer_hud(); |
| } |
| |
| TouchObserverHUD* GetExternalTouchHud() { |
| return GetExternalRootController()->touch_observer_hud(); |
| } |
| |
| TouchObserverHUD* GetPrimaryTouchHud() { |
| return GetPrimaryRootController()->touch_observer_hud(); |
| } |
| |
| TouchObserverHUD* GetSecondaryTouchHud() { |
| return GetSecondaryRootController()->touch_observer_hud(); |
| } |
| |
| display::ManagedDisplayInfo CreateDisplayInfo(int64_t id, |
| const gfx::Rect& bounds) { |
| display::ManagedDisplayInfo info(id, base::StringPrintf("x-%" PRId64, id), |
| false); |
| info.SetBounds(bounds); |
| return info; |
| } |
| |
| aura::Window* GetRootWindowForTouchHud(TouchObserverHUD* hud) { |
| return hud->root_window_; |
| } |
| |
| views::Widget* GetWidgetForTouchHud(TouchObserverHUD* hud) { |
| return hud->widget_; |
| } |
| |
| int64_t internal_display_id_; |
| int64_t external_display_id_; |
| int64_t mirrored_display_id_; |
| display::ManagedDisplayInfo internal_display_info_; |
| display::ManagedDisplayInfo external_display_info_; |
| display::ManagedDisplayInfo mirrored_display_info_; |
| |
| std::vector<display::ManagedDisplayInfo> display_info_list_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TouchObserverHUDTest); |
| }; |
| |
| // Checks if debug touch HUD is correctly initialized for a single display. |
| TEST_F(TouchObserverHUDTest, SingleDisplay) { |
| // Setup a single display setting. |
| SetupSingleDisplay(); |
| |
| // Check if touch HUD is set correctly and associated with appropriate |
| // display. |
| CheckInternalDisplay(); |
| } |
| |
| // Checks if debug touch HUDs are correctly initialized for two displays. |
| TEST_F(TouchObserverHUDTest, DualDisplays) { |
| // Setup a dual display setting. |
| SetupDualDisplays(); |
| |
| // Check if touch HUDs are set correctly and associated with appropriate |
| // displays. |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| } |
| |
| // Checks if debug touch HUDs are correctly handled when primary display is |
| // changed. |
| TEST_F(TouchObserverHUDTest, SwapPrimaryDisplay) { |
| // Setup a dual display setting. |
| SetupDualDisplays(); |
| |
| // Set the primary display to the external one. |
| SetExternalAsPrimary(); |
| |
| // Check if displays' touch HUDs are not swapped as root windows are. |
| EXPECT_EQ(external_display_id(), GetPrimaryDisplay().id()); |
| EXPECT_EQ(internal_display_id(), |
| display_manager()->GetSecondaryDisplay().id()); |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| |
| // Set the primary display back to the internal one. |
| SetInternalAsPrimary(); |
| |
| // Check if displays' touch HUDs are not swapped back as root windows are. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| EXPECT_EQ(external_display_id(), |
| display_manager()->GetSecondaryDisplay().id()); |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| } |
| |
| // Checks if debug touch HUDs are correctly handled when displays are mirrored. |
| TEST_F(TouchObserverHUDTest, MirrorDisplays) { |
| // Disable restoring mirror mode to prevent interference from previous |
| // display configuration. |
| display_manager()->set_disable_restoring_mirror_mode_for_test(true); |
| |
| // Setup a dual display setting. |
| SetupDualDisplays(); |
| |
| // Mirror displays. |
| MirrorDisplays(); |
| |
| // Check if the internal display is intact. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| CheckInternalDisplay(); |
| |
| // Unmirror displays. |
| UnmirrorDisplays(); |
| |
| // Check if external display is added back correctly. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| EXPECT_EQ(external_display_id(), |
| display_manager()->GetSecondaryDisplay().id()); |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| |
| display_manager()->set_disable_restoring_mirror_mode_for_test(false); |
| } |
| |
| // Checks if debug touch HUDs are correctly handled when displays are mirrored |
| // after setting the external display as the primary one. |
| TEST_F(TouchObserverHUDTest, SwapPrimaryThenMirrorDisplays) { |
| display_manager()->set_disable_restoring_mirror_mode_for_test(true); |
| |
| // Setup a dual display setting. |
| SetupDualDisplays(); |
| |
| // Set the primary display to the external one. |
| SetExternalAsPrimary(); |
| |
| // Mirror displays. |
| MirrorDisplays(); |
| |
| // Check if the internal display is set as the primary one. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| CheckInternalDisplay(); |
| |
| // Unmirror displays. |
| UnmirrorDisplays(); |
| |
| // Check if the external display is added back as the primary display and |
| // touch HUDs are set correctly. |
| EXPECT_EQ(external_display_id(), GetPrimaryDisplay().id()); |
| EXPECT_EQ(internal_display_id(), |
| display_manager()->GetSecondaryDisplay().id()); |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| |
| display_manager()->set_disable_restoring_mirror_mode_for_test(false); |
| } |
| |
| // Checks if debug touch HUDs are correctly handled when the external display, |
| // which is the secondary one, is removed. |
| TEST_F(TouchObserverHUDTest, RemoveSecondaryDisplay) { |
| // Setup a dual display setting. |
| SetupDualDisplays(); |
| |
| // Remove external display which is the secondary one. |
| RemoveExternalDisplay(); |
| |
| // Check if the internal display is intact. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| CheckInternalDisplay(); |
| |
| // Add external display back. |
| AddExternalDisplay(); |
| |
| // Check if displays' touch HUDs are set correctly. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| EXPECT_EQ(external_display_id(), |
| display_manager()->GetSecondaryDisplay().id()); |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| } |
| |
| // Checks if debug touch HUDs are correctly handled when the external display, |
| // which is set as the primary display, is removed. |
| TEST_F(TouchObserverHUDTest, RemovePrimaryDisplay) { |
| // Setup a dual display setting. |
| SetupDualDisplays(); |
| |
| // Set the primary display to the external one. |
| SetExternalAsPrimary(); |
| |
| // Remove the external display which is the primary display. |
| RemoveExternalDisplay(); |
| |
| // Check if the internal display is set as the primary one. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| CheckInternalDisplay(); |
| |
| // Add the external display back. |
| AddExternalDisplay(); |
| |
| // Check if the external display is set as primary and touch HUDs are set |
| // correctly. |
| EXPECT_EQ(external_display_id(), GetPrimaryDisplay().id()); |
| EXPECT_EQ(internal_display_id(), |
| display_manager()->GetSecondaryDisplay().id()); |
| CheckInternalDisplay(); |
| CheckExternalDisplay(); |
| } |
| |
| // Checks if debug touch HUDs are correctly handled when all displays are |
| // removed. |
| TEST_F(TouchObserverHUDTest, Headless) { |
| // Setup a single display setting. |
| SetupSingleDisplay(); |
| |
| // Remove the only display which is the internal one. |
| RemoveInternalDisplay(); |
| |
| // Add the internal display back. |
| AddInternalDisplay(); |
| |
| // Check if the display's touch HUD is set correctly. |
| EXPECT_EQ(internal_display_id(), GetPrimaryDisplay().id()); |
| CheckInternalDisplay(); |
| } |
| |
| } // namespace ash |