git-svn-id: svn://svn.chromium.org/chrome/trunk/src@904 0039d316-1c4b-4281-b951-d872f2087c98

3 files changed, 152 insertions, 19 deletions

diff --git a/chrome/browser/automation/automation_provider.cc b/chrome/browser/automation/automation_provider.cc
index 818c321..f23e7d1 100644
--- a/chrome/browser/automation/automation_provider.cc
+++ b/chrome/browser/automation/automation_provider.cc
@@ -1272,7 +1272,8 @@ void AutomationProvider::WindowSimulateClick(const IPC::Message& message,
 void AutomationProvider::WindowSimulateDrag(const IPC::Message& message,
                                             int handle,
                                             std::vector<POINT> drag_path,
-                                            int flags) {
+                                            int flags,
+                                            bool press_escape_en_route) {
   bool succeeded = false;
   if (browser_tracker_->ContainsHandle(handle) && (drag_path.size() > 1)) {
     succeeded = true;
@@ -1313,6 +1314,17 @@ void AutomationProvider::WindowSimulateDrag(const IPC::Message& message,
     }
     POINT end = drag_path[drag_path.size() - 1];
     SetCursorPos(end.x, end.y);
+
+    if (press_escape_en_route) {
+      // Press Escape.
+      ui_controls::SendKeyPress(VK_ESCAPE,
+                               ((flags & ChromeViews::Event::EF_CONTROL_DOWN)
+                                == ChromeViews::Event::EF_CONTROL_DOWN),
+                               ((flags & ChromeViews::Event::EF_SHIFT_DOWN) ==
+                                ChromeViews::Event::EF_SHIFT_DOWN),
+                               ((flags & ChromeViews::Event::EF_ALT_DOWN) ==
+                                ChromeViews::Event::EF_ALT_DOWN));
+    }
     SendMessage(top_level_hwnd, up_message, wparam_flags,
                 MAKELPARAM(end.x, end.y));
 
diff --git a/chrome/browser/automation/automation_provider.h b/chrome/browser/automation/automation_provider.h
index 35827c8..20f34a1 100644
--- a/chrome/browser/automation/automation_provider.h
+++ b/chrome/browser/automation/automation_provider.h
@@ -144,7 +144,8 @@ class AutomationProvider : public base::RefCounted<AutomationProvider>,
   void WindowSimulateDrag(const IPC::Message& message,
                           int handle,
                           std::vector<POINT> drag_path,
-                          int flags);
+                          int flags,
+                          bool press_escape_en_route);
   void WindowSimulateClick(const IPC::Message& message,
                           int handle,
                           POINT click,
diff --git a/chrome/browser/tabs/tab_dragging_test.cc b/chrome/browser/tabs/tab_dragging_test.cc
index ecda371..a976d42 100644
--- a/chrome/browser/tabs/tab_dragging_test.cc
+++ b/chrome/browser/tabs/tab_dragging_test.cc
@@ -51,8 +51,8 @@ protected:
   }
 };
 
-// Automated UI test to open three tabs in a new window, and drag Tab-1 into
-// the position of Tab-2.
+// Automated UI test to open three tabs in a new window, and drag Tab_1 into
+// the position of Tab_2.
 TEST_F(TabDraggingTest, Tab1Tab2) {
   scoped_ptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
   ASSERT_TRUE(browser.get());
@@ -63,20 +63,21 @@ TEST_F(TabDraggingTest, Tab1Tab2) {
   // Get initial tab count.
   int initial_tab_count = 0;
   ASSERT_TRUE(browser->GetTabCount(&initial_tab_count));
+  ASSERT_TRUE(1 == initial_tab_count);
 
-  // Get Tab-1 which comes with the browser window.
+  // Get Tab_1 which comes with the browser window.
   scoped_ptr<TabProxy> tab1(browser->GetTab(0));
   ASSERT_TRUE(tab1.get());
   GURL tab1_url;
   ASSERT_TRUE(tab1->GetCurrentURL(&tab1_url));
 
-  // Add Tab-2.
+  // Add Tab_2.
   GURL tab2_url("about:");
   ASSERT_TRUE(browser->AppendTab(tab2_url));
   scoped_ptr<TabProxy> tab2(browser->GetTab(1));
   ASSERT_TRUE(tab2.get());
   
-  // Add Tab-3.
+  // Add Tab_3.
   GURL tab3_url("about:plugins");
   ASSERT_TRUE(browser->AppendTab(tab3_url));
   scoped_ptr<TabProxy> tab3(browser->GetTab(2));
@@ -119,15 +120,15 @@ TEST_F(TabDraggingTest, Tab1Tab2) {
   EXPECT_LT(0, urlbar_bounds.width());
   EXPECT_LT(0, urlbar_bounds.height());
 
-  // TEST: Move Tab-1 to the position of Tab-2
+  // TEST: Move Tab_1 to the position of Tab_2
   //   ____________   ____________   ____________
   //  /            \ /            \ /            \
-  // |    Tab-1     |     Tab-2    |    Tab-3     |
+  // |    Tab_1     |     Tab_2    |    Tab_3     |
   //  ---- ---- ---- ---- ---- ---- ---- ---- ----
   //         x---- ---->
   //              ____________  
   //             /     X      \ 
-  //            |    Tab-1     |
+  //            |    Tab_1     |
   //             ---- ---- ----
 
   POINT start;
@@ -137,7 +138,8 @@ TEST_F(TabDraggingTest, Tab1Tab2) {
   end.x = start.x + 2*bounds1.width()/3;
   end.y = start.y;
   ASSERT_TRUE(browser->SimulateDrag(start, end,
-                                    ChromeViews::Event::EF_LEFT_BUTTON_DOWN));
+                                    ChromeViews::Event::EF_LEFT_BUTTON_DOWN,
+                                    false));
 
   // Now check for expected results.
   tab1.reset(browser->GetTab(0));
@@ -154,7 +156,7 @@ TEST_F(TabDraggingTest, Tab1Tab2) {
   EXPECT_EQ(tab2_url.spec(), tab1_new_url.spec());
 }
 
-// Drag Tab-1 into the position of Tab-3. 
+// Drag Tab_1 into the position of Tab_3. 
 TEST_F(TabDraggingTest, Tab1Tab3) {
   scoped_ptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
   ASSERT_TRUE(browser.get());
@@ -165,20 +167,21 @@ TEST_F(TabDraggingTest, Tab1Tab3) {
   // Get initial tab count.
   int initial_tab_count = 0;
   ASSERT_TRUE(browser->GetTabCount(&initial_tab_count));
+  ASSERT_TRUE(1 == initial_tab_count);
 
-  // Get Tab-1 which comes with the browser window.
+  // Get Tab_1 which comes with the browser window.
   scoped_ptr<TabProxy> tab1(browser->GetTab(0));
   ASSERT_TRUE(tab1.get());
   GURL tab1_url;
   ASSERT_TRUE(tab1->GetCurrentURL(&tab1_url));
 
-  // Add Tab-2.
+  // Add Tab_2.
   GURL tab2_url("about:");
   ASSERT_TRUE(browser->AppendTab(tab2_url));
   scoped_ptr<TabProxy> tab2(browser->GetTab(1));
   ASSERT_TRUE(tab2.get());
   
-  // Add Tab-3.
+  // Add Tab_3.
   GURL tab3_url("about:plugins");
   ASSERT_TRUE(browser->AppendTab(tab3_url));
   scoped_ptr<TabProxy> tab3(browser->GetTab(2));
@@ -221,15 +224,15 @@ TEST_F(TabDraggingTest, Tab1Tab3) {
   EXPECT_LT(0, urlbar_bounds.width());
   EXPECT_LT(0, urlbar_bounds.height());
 
-  // TEST: Move Tab-1 to the middle position of Tab-3
+  // TEST: Move Tab_1 to the middle position of Tab_3
   //   ____________   ____________   ____________
   //  /            \ /            \ /            \
-  // |    Tab-1     |     Tab-2    |    Tab-3     |
+  // |    Tab_1     |     Tab_2    |    Tab_3     |
   //  ---- ---- ---- ---- ---- ---- ---- ---- ----
   //         x---- ---- ---- ---- ---- ---->
   //                                  ____________  
   //                                 /     X      \ 
-  //                                |    Tab-1     |
+  //                                |    Tab_1     |
   //                                 ---- ---- ----
 
   POINT start;
@@ -239,7 +242,8 @@ TEST_F(TabDraggingTest, Tab1Tab3) {
   end.x = start.x + bounds1.width()/2 + bounds2.width() + bounds3.width()/2;
   end.y = start.y;
   ASSERT_TRUE(browser->SimulateDrag(start, end,
-                                    ChromeViews::Event::EF_LEFT_BUTTON_DOWN));
+                                    ChromeViews::Event::EF_LEFT_BUTTON_DOWN,
+                                    false));
 
   // Now check for expected results.
   tab1.reset(browser->GetTab(0));
@@ -261,3 +265,119 @@ TEST_F(TabDraggingTest, Tab1Tab3) {
   EXPECT_EQ(tab2_new_url.spec(), tab3_url.spec());
   EXPECT_EQ(tab3_new_url.spec(), tab1_url.spec());
 }
+
+// Drag Tab_1 into the position of Tab_3, and press ESCAPE before releasing the
+// left mouse button.
+TEST_F(TabDraggingTest, Tab1Tab3Escape) {
+  scoped_ptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
+  ASSERT_TRUE(browser.get());
+  scoped_ptr<WindowProxy> window(
+      automation()->GetWindowForBrowser(browser.get()));
+  ASSERT_TRUE(window.get());
+
+  // Get initial tab count.
+  int initial_tab_count = 0;
+  ASSERT_TRUE(browser->GetTabCount(&initial_tab_count));
+  ASSERT_TRUE(1 == initial_tab_count);
+
+  // Get Tab_1 which comes with the browser window.
+  scoped_ptr<TabProxy> tab1(browser->GetTab(0));
+  ASSERT_TRUE(tab1.get());
+  GURL tab1_url;
+  ASSERT_TRUE(tab1->GetCurrentURL(&tab1_url));
+
+  // Add Tab_2.
+  GURL tab2_url("about:");
+  ASSERT_TRUE(browser->AppendTab(tab2_url));
+  scoped_ptr<TabProxy> tab2(browser->GetTab(1));
+  ASSERT_TRUE(tab2.get());
+  
+  // Add Tab_3.
+  GURL tab3_url("about:plugins");
+  ASSERT_TRUE(browser->AppendTab(tab3_url));
+  scoped_ptr<TabProxy> tab3(browser->GetTab(2));
+  ASSERT_TRUE(tab3.get());
+
+  // Make sure 3 tabs are open
+  int final_tab_count = 0;
+  ASSERT_TRUE(browser->WaitForTabCountToChange(initial_tab_count,
+                                               &final_tab_count,
+                                               10000));
+  ASSERT_TRUE(final_tab_count == initial_tab_count + 2);
+
+  // Get bounds for the tabs.
+  gfx::Rect bounds1;
+  ASSERT_TRUE(window->GetViewBounds(VIEW_ID_TAB_0, &bounds1, false));
+  EXPECT_LT(0, bounds1.x());
+  EXPECT_LT(0, bounds1.width());
+  EXPECT_LT(0, bounds1.height());
+
+  gfx::Rect bounds2;
+  ASSERT_TRUE(window->GetViewBounds(VIEW_ID_TAB_1, &bounds2, false));
+  EXPECT_LT(0, bounds2.width());
+  EXPECT_LT(0, bounds2.height());
+  EXPECT_LT(bounds1.x(), bounds2.x());
+  EXPECT_EQ(bounds2.y(), bounds1.y());
+
+  gfx::Rect bounds3;
+  ASSERT_TRUE(window->GetViewBounds(VIEW_ID_TAB_2, &bounds3, false));
+  EXPECT_LT(0, bounds3.width());
+  EXPECT_LT(0, bounds3.height());
+  EXPECT_LT(bounds2.x(), bounds3.x());
+  EXPECT_EQ(bounds3.y(), bounds2.y());
+
+  // Get url Bar bounds.
+  gfx::Rect urlbar_bounds;
+  ASSERT_TRUE(window->GetViewBounds(VIEW_ID_LOCATION_BAR, &urlbar_bounds,
+                                    false));
+  EXPECT_LT(0, urlbar_bounds.x());
+  EXPECT_LT(0, urlbar_bounds.y());
+  EXPECT_LT(0, urlbar_bounds.width());
+  EXPECT_LT(0, urlbar_bounds.height());
+
+  // TEST: Move Tab_1 to the middle position of Tab_3
+  //   ____________   ____________   ____________
+  //  /            \ /            \ /            \
+  // |    Tab_1     |     Tab_2    |    Tab_3     |
+  //  ---- ---- ---- ---- ---- ---- ---- ---- ----
+  //         x---- ---- ---- ---- ---- ----> + ESCAPE
+  //                                  ____________  
+  //                                 /     X      \ 
+  //                                |    Tab_1     |
+  //                                 ---- ---- ----
+
+  POINT start;
+  POINT end;  
+  start.x = bounds1.x() + bounds1.width()/2;
+  start.y = bounds1.y() + bounds1.height()/2;
+  end.x = start.x + bounds1.width()/2 + bounds2.width() + bounds3.width()/2;
+  end.y = start.y;
+
+  // Simulate drag with 'true' as the last parameter. This will interrupt 
+  // in-flight with Escape.
+  ASSERT_TRUE(browser->SimulateDrag(start, end,
+                                    ChromeViews::Event::EF_LEFT_BUTTON_DOWN,
+                                    true));
+
+  // Now check for expected results.
+  tab1.reset(browser->GetTab(0));
+  ASSERT_TRUE(tab1.get());
+  GURL tab1_new_url;
+  ASSERT_TRUE(tab1->GetCurrentURL(&tab1_new_url));
+
+  tab2.reset(browser->GetTab(1));
+  ASSERT_TRUE(tab2.get());
+  GURL tab2_new_url;
+  ASSERT_TRUE(tab2->GetCurrentURL(&tab2_new_url));
+
+  tab3.reset(browser->GetTab(2));
+  ASSERT_TRUE(tab3.get());
+  GURL tab3_new_url;
+  ASSERT_TRUE(tab3->GetCurrentURL(&tab3_new_url));
+  
+  // The tabs should be in their original positions.
+  EXPECT_EQ(tab1_new_url.spec(), tab1_url.spec());
+  EXPECT_EQ(tab2_new_url.spec(), tab2_url.spec());
+  EXPECT_EQ(tab3_new_url.spec(), tab3_url.spec());
+}
+