Add more test to DisplayControllerTest.

ash/display/display_controller_unittest.cc

Index: ash/display/display_controller_unittest.cc
diff --git a/ash/display/display_controller_unittest.cc b/ash/display/display_controller_unittest.cc
index 03428a922892e5cc9a2cedcdd53c6d339f0e81d5..c6dee63d62944baf727471b0f32a2aa6852dc68a 100644
--- a/ash/display/display_controller_unittest.cc
+++ b/ash/display/display_controller_unittest.cc
@@ -107,6 +107,8 @@ TEST_F(DisplayControllerTest, MAYBE_BoundsUpdated) {
   EXPECT_EQ("0,700 1000x1000", GetSecondaryDisplay().bounds().ToString());
 }
 
+// Verifies if the mouse pointer correctly moves to another display when there
+// are two displays.
 TEST_F(DisplayControllerTest, WarpMouse) {
   UpdateDisplay("500x500,500x500");
 
@@ -163,6 +165,38 @@ TEST_F(DisplayControllerTest, WarpMouse) {
   EXPECT_FALSE(is_warped);
 }
 
+// Verifies if the mouse pointer correctly moves to another display even when
+// two displays are not the same size.
+TEST_F(DisplayControllerTest, WarpMouseDifferentSizeDisplays) {
+  UpdateDisplay("500x500,600x600");  // the second one is larger.
+
+  ash::internal::DisplayController* controller =
+      Shell::GetInstance()->display_controller();
+  EXPECT_EQ(internal::DisplayController::RIGHT,
+            controller->secondary_display_layout());
+
+  Shell::RootWindowList root_windows = Shell::GetAllRootWindows();
+  aura::Env::GetInstance()->SetLastMouseLocation(*root_windows[1],
+                                                 gfx::Point(123, 123));
+
+  // Touch the left edge of the secondary root window. Pointer should NOT warp
+  // because 1px left of (0, 500) is outside the primary root window.
+  bool is_warped = controller->WarpMouseCursorIfNecessary(root_windows[1],
+                                                          gfx::Point(0, 500));
+  EXPECT_FALSE(is_warped);
+  EXPECT_EQ("623,123",  // by 2px.
+            aura::Env::GetInstance()->last_mouse_location().ToString());
+
+  // Touch the left edge of the secondary root window. Pointer should warp
+  // because 1px left of (0, 499) is inside the primary root window.
+  is_warped = controller->WarpMouseCursorIfNecessary(root_windows[1],
+                                                     gfx::Point(0, 499));
+  EXPECT_TRUE(is_warped);
+  EXPECT_EQ("498,499",  // by 2px.
+            aura::Env::GetInstance()->last_mouse_location().ToString());
+}
+
+// Verifies if DisplayController::dont_warp_mouse() works as expected.
 TEST_F(DisplayControllerTest, SetUnsetDontWarpMousedFlag) {
   UpdateDisplay("500x500,500x500");