Fix moving the mouse between two displays after the secondary display becomes primary.

ash/display/extended_mouse_warp_controller_unittest.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/display/extended_mouse_warp_controller.h"
 
 #include "ash/display/display_layout_store.h"
 #include "ash/display/display_manager.h"
 #include "ash/display/mouse_cursor_event_filter.h"
 #include "ash/shell.h"
@@ skipping 197 matching lines @@
             mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
 
   event_filter()->HideSharedEdgeIndicator();
 }
 
 // Verify indicators show up as expected with 3+ displays.
 TEST_F(ExtendedMouseWarpControllerTest, IndicatorBoundsTestThreeDisplays) {
   if (!SupportsMultipleDisplays())
     return;
 
+  auto run_test = [this] {
+    aura::Window::Windows root_windows = Shell::GetAllRootWindows();
+
+    // Left most display
+    event_filter()->ShowSharedEdgeIndicator(root_windows[0]);
+    ASSERT_EQ(2U, mouse_warp_controller()->warp_regions_.size());
+    EXPECT_EQ(gfx::Rect(359, 16, 1, 344),
+              mouse_warp_controller()->warp_regions_[0]->a_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(360, 0, 1, 360),
+              mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(1060, 16, 1, 684),
+              mouse_warp_controller()->warp_regions_[1]->a_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(1059, 0, 1, 700),
+              mouse_warp_controller()->warp_regions_[1]->b_indicator_bounds);
+
+    // Middle display
+    event_filter()->ShowSharedEdgeIndicator(root_windows[1]);
+    ASSERT_EQ(2U, mouse_warp_controller()->warp_regions_.size());
+    EXPECT_EQ(gfx::Rect(360, 16, 1, 344),
+              mouse_warp_controller()->warp_regions_[0]->a_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(359, 0, 1, 360),
+              mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(1059, 16, 1, 684),
+              mouse_warp_controller()->warp_regions_[1]->a_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(1060, 0, 1, 700),
+              mouse_warp_controller()->warp_regions_[1]->b_indicator_bounds);
+
+    // Right most display
+    event_filter()->ShowSharedEdgeIndicator(root_windows[2]);
+    ASSERT_EQ(2U, mouse_warp_controller()->warp_regions_.size());
+    EXPECT_EQ(gfx::Rect(360, 16, 1, 344),
+              mouse_warp_controller()->warp_regions_[0]->a_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(359, 0, 1, 360),
+              mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(1060, 16, 1, 684),
+              mouse_warp_controller()->warp_regions_[1]->a_indicator_bounds);
+    EXPECT_EQ(gfx::Rect(1059, 0, 1, 700),
+              mouse_warp_controller()->warp_regions_[1]->b_indicator_bounds);
+
+    event_filter()->HideSharedEdgeIndicator();
+  };
+
   UpdateDisplay("360x360,700x700,1000x1000");
-  aura::Window::Windows root_windows = Shell::GetAllRootWindows();
+  run_test();
 
-  // Left most display
-  event_filter()->ShowSharedEdgeIndicator(root_windows[0]);
-  ASSERT_EQ(2U, mouse_warp_controller()->warp_regions_.size());
-  EXPECT_EQ(gfx::Rect(359, 16, 1, 344),
-            mouse_warp_controller()->warp_regions_[0]->a_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(360, 0, 1, 360),
-            mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(1060, 16, 1, 684),
-            mouse_warp_controller()->warp_regions_[1]->a_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(1059, 0, 1, 700),
-            mouse_warp_controller()->warp_regions_[1]->b_indicator_bounds);
-
-  // Middle display
-  event_filter()->ShowSharedEdgeIndicator(root_windows[1]);
-  ASSERT_EQ(2U, mouse_warp_controller()->warp_regions_.size());
-  EXPECT_EQ(gfx::Rect(360, 16, 1, 344),
-            mouse_warp_controller()->warp_regions_[0]->a_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(359, 0, 1, 360),
-            mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(1059, 16, 1, 684),
-            mouse_warp_controller()->warp_regions_[1]->a_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(1060, 0, 1, 700),
-            mouse_warp_controller()->warp_regions_[1]->b_indicator_bounds);
-
-  // Right most display
-  event_filter()->ShowSharedEdgeIndicator(root_windows[2]);
-  ASSERT_EQ(2U, mouse_warp_controller()->warp_regions_.size());
-  EXPECT_EQ(gfx::Rect(360, 16, 1, 344),
-            mouse_warp_controller()->warp_regions_[0]->a_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(359, 0, 1, 360),
-            mouse_warp_controller()->warp_regions_[0]->b_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(1060, 16, 1, 684),
-            mouse_warp_controller()->warp_regions_[1]->a_indicator_bounds);
-  EXPECT_EQ(gfx::Rect(1059, 0, 1, 700),
-            mouse_warp_controller()->warp_regions_[1]->b_indicator_bounds);
-
-  event_filter()->HideSharedEdgeIndicator();
+  UpdateDisplay("360x360,700x700,1000x1000");
+  Shell::GetInstance()->window_tree_host_manager()->SwapPrimaryDisplayForTest();
+  run_test();
 }
 
 }  // namespace ash