Cancel drag if display configuration changes.

chrome/browser/ui/views/tabs/tab_drag_controller_interactive_uitest.cc

 // Copyright (c) 2012 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 "chrome/browser/ui/views/tabs/tab_drag_controller_interactive_uitest.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "base/run_loop.h"
@@ skipping 15 matching lines @@
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/web_contents.h"
 #include "ui/gfx/screen.h"
 #include "ui/ui_controls/ui_controls.h"
 #include "ui/views/view.h"
 #include "ui/views/widget/widget.h"
 
 #if defined(USE_ASH)
 #include "ash/display/display_controller.h"
+#include "ash/display/multi_display_manager.h"
 #include "ash/shell.h"
 #include "ui/aura/test/event_generator.h"
 #include "ui/aura/root_window.h"
 #endif
 
 namespace test {
 
 namespace {
 
 const char kTabDragControllerInteractiveUITestUserDataKey[] =
@@ skipping 887 matching lines @@
   // Add another tab.
   AddTabAndResetBrowser(browser());
   TabStrip* tab_strip = GetTabStripForBrowser(browser());
 
   // Create another browser.
   Browser* browser2 = CreateBrowser(browser()->profile());
   TabStrip* tab_strip2 = GetTabStripForBrowser(browser2);
   ResetIDs(browser2->tab_strip_model(), 100);
 
   // Move the second browser to the second display.
-  std::vector<aura::RootWindow*> roots(
-      ash::Shell::GetInstance()->display_controller()->GetAllRootWindows());
+  std::vector<aura::RootWindow*> roots(ash::Shell::GetAllRootWindows());
   ASSERT_EQ(2u, roots.size());
   aura::RootWindow* second_root = roots[1];
   gfx::Rect work_area = gfx::Screen::GetDisplayNearestWindow(
       second_root).work_area();
   browser2->window()->SetBounds(work_area);
   EXPECT_EQ(second_root,
             browser2->window()->GetNativeWindow()->GetRootWindow());
 
   // Move to the first tab and drag it enough so that it detaches, but not
   // enough that it attaches to browser2.
@@ skipping 12 matching lines @@
 
   // Release the mouse, stopping the drag session.
   ASSERT_TRUE(ReleaseInput());
   ASSERT_FALSE(tab_strip2->IsDragSessionActive());
   ASSERT_FALSE(tab_strip->IsDragSessionActive());
   ASSERT_FALSE(TabDragController::IsActive());
   EXPECT_EQ("0 100", IDString(browser2->tab_strip_model()));
   EXPECT_EQ("1", IDString(browser()->tab_strip_model()));
 }
 
+namespace {
+
+// Invoked from the nested message loop.
+void CancelDragTabToWindowInSeparateDisplayStep3(
+    DetachToBrowserTabDragControllerTest* test,
+    TabStrip* tab_strip) {
+  ASSERT_FALSE(tab_strip->IsDragSessionActive());
+  ASSERT_TRUE(TabDragController::IsActive());
+  ASSERT_EQ(2u, BrowserList::size());
+
+  // Switching display mode should cancel the drag operation.
+  ash::internal::MultiDisplayManager::CycleDisplay();
+}
+
+// Invoked from the nested message loop.
+void CancelDragTabToWindowInSeparateDisplayStep2(
+    DetachToBrowserTabDragControllerTest* test,
+    TabStrip* tab_strip,
+    gfx::Point final_destination) {
+  ASSERT_FALSE(tab_strip->IsDragSessionActive());
+  ASSERT_TRUE(TabDragController::IsActive());
+  ASSERT_EQ(2u, BrowserList::size());
+
+  Browser* new_browser = *(++BrowserList::begin());
+  EXPECT_EQ(ash::Shell::GetPrimaryRootWindow(),
+            new_browser->window()->GetNativeWindow()->GetRootWindow());
+
+  ASSERT_TRUE(test->DragInputToNotifyWhenDone(
+      final_destination.x(), final_destination.y(),
+      base::Bind(&CancelDragTabToWindowInSeparateDisplayStep3,
+                 test, tab_strip)));
+}
+
+}  // namespace
+
+// Drags from browser to a second display and releases input.
+IN_PROC_BROWSER_TEST_P(DetachToBrowserInSeparateDisplayTabDragControllerTest,
+                       CancelDragTabToWindowInSeparateDisplay) {
+  // Add another tab.
+  AddTabAndResetBrowser(browser());
+  TabStrip* tab_strip = GetTabStripForBrowser(browser());
+
+  EXPECT_EQ("0 1", IDString(browser()->tab_strip_model()));
+
+  // Move the second browser to the second display.
+  std::vector<aura::RootWindow*> roots(ash::Shell::GetAllRootWindows());
+  ASSERT_EQ(2u, roots.size());
+  gfx::Point final_destination =
+      gfx::Screen::GetDisplayNearestWindow(roots[1]).work_area().CenterPoint();
+
+  // Move to the first tab and drag it enough so that it detaches, but not
+  // enough to move to another display.
+  gfx::Point tab_0_center(GetCenterInScreenCoordinates(tab_strip->tab_at(0)));
+  ASSERT_TRUE(PressInput(tab_0_center));
+  ASSERT_TRUE(DragInputToNotifyWhenDone(
+                  tab_0_center.x(), tab_0_center.y() + GetDetachY(tab_strip),
+                  base::Bind(&CancelDragTabToWindowInSeparateDisplayStep2,
+                             this, tab_strip, final_destination)));
+  QuitWhenNotDragging();
+
+  // Drag operation must have been cancelled.
+  if (input_source() == INPUT_SOURCE_TOUCH) {
+    // A window capture is reset to NULL when merging two root
+    // windows, and that capture event completes the touch based
+    // drag operation, instead of cancelling it.
+    ASSERT_EQ(2u, BrowserList::size());
+    ASSERT_FALSE(tab_strip->IsDragSessionActive());
+    ASSERT_FALSE(TabDragController::IsActive());
+    EXPECT_EQ("1", IDString(browser()->tab_strip_model()));
+    Browser* new_browser = *(++BrowserList::begin());
+    EXPECT_EQ("0", IDString(new_browser->tab_strip_model()));
+  } else {
+    // Mouse operation doesn't get cancelled when the capture is
+    // lost, and removing the display correctly cancel the operation.
+    ASSERT_EQ(1u, BrowserList::size());
+    ASSERT_FALSE(tab_strip->IsDragSessionActive());
+    ASSERT_FALSE(TabDragController::IsActive());
+    EXPECT_EQ("0 1", IDString(browser()->tab_strip_model()));
+  }
+  // Release the mouse
+  ASSERT_TRUE(ReleaseInput());
+}
+
 #endif
 
 #if defined(USE_ASH)
 INSTANTIATE_TEST_CASE_P(TabDragging,
                         DetachToBrowserInSeparateDisplayTabDragControllerTest,
                         ::testing::Values("mouse", "touch"));
 INSTANTIATE_TEST_CASE_P(TabDragging,
                         DetachToBrowserTabDragControllerTest,
                         ::testing::Values("mouse", "touch"));
 #else
 INSTANTIATE_TEST_CASE_P(TabDragging,
                         DetachToBrowserTabDragControllerTest,
                         ::testing::Values("mouse"));
 #endif