Improve TableView unit tests and fix bugs that were uncovered.

ui/views/controls/table/table_view_unittest.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 "ui/views/controls/table/table_view.h"
 
 #include <stddef.h>
 
 #include "base/macros.h"
+#include "base/run_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/events/event_utils.h"
+#include "ui/events/test/event_generator.h"
 #include "ui/views/controls/table/table_grouper.h"
 #include "ui/views/controls/table/table_header.h"
 #include "ui/views/controls/table/table_view_observer.h"
+#include "ui/views/test/views_test_base.h"
 
 // Put the tests in the views namespace to make it easier to declare them as
 // friend classes.
 namespace views {
 
 class TableViewTestHelper {
  public:
   explicit TableViewTestHelper(TableView* table) : table_(table) {}
 
   std::string GetPaintRegion(const gfx::Rect& bounds) {
@@ skipping 169 matching lines @@
     // Overriden so key processing works.
     return true;
   }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestTableView);
 };
 
 }  // namespace
 
-class TableViewTest : public testing::Test {
+class TableViewTest : public ViewsTestBase {
  public:
   TableViewTest() : table_(NULL) {}
 
   void SetUp() override {
+    ViewsTestBase::SetUp();
+
     model_.reset(new TestTableModel2);
     std::vector<ui::TableColumn> columns(2);
     columns[0].title = base::ASCIIToUTF16("Title Column 0");
     columns[0].sortable = true;
     columns[1].title = base::ASCIIToUTF16("Title Column 1");
     columns[1].id = 1;
     columns[1].sortable = true;
     table_ = new TestTableView(model_.get(), columns);
-    parent_.reset(table_->CreateParentIfNecessary());
-    parent_->SetBounds(0, 0, 10000, 10000);
-    parent_->Layout();
+    View* parent = table_->CreateParentIfNecessary();
+    parent->SetBounds(0, 0, 10000, 10000);
+    parent->Layout();
     helper_.reset(new TableViewTestHelper(table_));
+
+    Widget::InitParams params = CreateParams(Widget::InitParams::TYPE_WINDOW);
+    params.ownership = views::Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
+    params.bounds = gfx::Rect(0, 0, 650, 650);
+    widget_.reset(new Widget);
+    widget_->Init(params);
+    widget_->GetContentsView()->AddChildView(parent);
+    widget_->Show();
+  }
+
+  void TearDown() override {
+    widget_.reset();
+    ViewsTestBase::TearDown();
   }
 
   void ClickOnRow(int row, int flags) {
-    const int y = row * table_->row_height();
-    const ui::MouseEvent pressed(ui::ET_MOUSE_PRESSED, gfx::Point(0, y),
-                                 gfx::Point(0, y), ui::EventTimeForNow(),
-                                 ui::EF_LEFT_MOUSE_BUTTON | flags,
-                                 ui::EF_LEFT_MOUSE_BUTTON);
-    table_->OnMousePressed(pressed);
+    ui::test::EventGenerator generator(widget_->GetNativeWindow());
+    generator.set_flags(flags);
+    generator.set_current_location(GetPointForRow(row));
+    generator.PressLeftButton();
   }
 
   void TapOnRow(int row) {
-    const int y = row * table_->row_height();
-    const ui::GestureEventDetails event_details(ui::ET_GESTURE_TAP);
-    ui::GestureEvent tap(0, y, 0, base::TimeTicks(), event_details);
-    table_->OnGestureEvent(&tap);
+    ui::test::EventGenerator generator(widget_->GetNativeWindow());
+    generator.GestureTapAt(GetPointForRow(row));
   }
 
   // Returns the state of the selection model as a string. The format is:
   // 'active=X anchor=X selection=X X X...'.
   std::string SelectionStateAsString() const {
     const ui::ListSelectionModel& model(table_->selection_model());
     std::string result = "active=" + base::IntToString(model.active()) +
         " anchor=" + base::IntToString(model.anchor()) +
         " selection=";
     const ui::ListSelectionModel::SelectedIndices& selection(
         model.selected_indices());
     for (size_t i = 0; i < selection.size(); ++i) {
       if (i != 0)
         result += " ";
       result += base::IntToString(selection[i]);
     }
     return result;
   }
 
   void PressKey(ui::KeyboardCode code) {
-    ui::KeyEvent event(ui::ET_KEY_PRESSED, code, ui::EF_NONE);
-    table_->OnKeyPressed(event);
+    ui::test::EventGenerator generator(widget_->GetNativeWindow());
+    generator.PressKey(code, ui::EF_NONE);
   }
 
  protected:
   std::unique_ptr<TestTableModel2> model_;
 
   // Owned by |parent_|.
   TableView* table_;
 
   std::unique_ptr<TableViewTestHelper> helper_;
 
  private:
-  std::unique_ptr<View> parent_;
+  gfx::Point GetPointForRow(int row) {
+    const int y = row * table_->row_height();
+    return table_->GetBoundsInScreen().origin() + gfx::Vector2d(5, y);
+  }
+
+  std::unique_ptr<Widget> widget_;
 
   DISALLOW_COPY_AND_ASSIGN(TableViewTest);
 };
 
 // Verifies GetPaintRegion.
 TEST_F(TableViewTest, GetPaintRegion) {
   // Two columns should be visible.
   EXPECT_EQ(2u, helper_->visible_col_count());
 
   EXPECT_EQ("rows=0 4 cols=0 2", helper_->GetPaintRegion(table_->bounds()));
@@ skipping 476 matching lines @@
   ranges.push_back(2);
   grouper.SetRanges(ranges);
   table_->SetGrouper(&grouper);
 
   TableViewObserverImpl observer;
   table_->set_observer(&observer);
 
   // Initially no selection.
   EXPECT_EQ("active=-1 anchor=-1 selection=", SelectionStateAsString());
 
-  PressKey(ui::VKEY_DOWN);
+  // Focus selects the first row.
+  table_->RequestFocus();
+  EXPECT_EQ(0, observer.GetChangedCountAndClear());
+  // ... after a delay.
+  base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1, observer.GetChangedCountAndClear());
   EXPECT_EQ("active=0 anchor=0 selection=0 1", SelectionStateAsString());
 
   PressKey(ui::VKEY_DOWN);
   EXPECT_EQ(1, observer.GetChangedCountAndClear());
   EXPECT_EQ("active=1 anchor=1 selection=0 1", SelectionStateAsString());
 
   PressKey(ui::VKEY_DOWN);
   EXPECT_EQ(1, observer.GetChangedCountAndClear());
   EXPECT_EQ("active=2 anchor=2 selection=2", SelectionStateAsString());
@@ skipping 106 matching lines @@
   std::vector<int> ranges;
   ranges.push_back(2);
   ranges.push_back(1);
   ranges.push_back(2);
   grouper.SetRanges(ranges);
   table_->SetGrouper(&grouper);
 
   TableViewObserverImpl observer;
   table_->set_observer(&observer);
 
-  // Initially no selection.
-  EXPECT_EQ("active=-1 anchor=-1 selection=", SelectionStateAsString());
+  // Focus selects the first row.
+  table_->RequestFocus();
+  EXPECT_EQ(0, observer.GetChangedCountAndClear());
+  // ... after a delay.
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, observer.GetChangedCountAndClear());
+  EXPECT_EQ("active=0 anchor=0 selection=0 1", SelectionStateAsString());
 
   PressKey(ui::VKEY_HOME);
-  EXPECT_EQ(1, observer.GetChangedCountAndClear());
+  EXPECT_EQ(0, observer.GetChangedCountAndClear());
   EXPECT_EQ("active=0 anchor=0 selection=0 1", SelectionStateAsString());
 
   PressKey(ui::VKEY_END);
   EXPECT_EQ(1, observer.GetChangedCountAndClear());
   EXPECT_EQ("active=4 anchor=4 selection=3 4", SelectionStateAsString());
 
+  PressKey(ui::VKEY_HOME);
+  EXPECT_EQ(1, observer.GetChangedCountAndClear());
+  EXPECT_EQ("active=0 anchor=0 selection=0 1", SelectionStateAsString());
+
   table_->set_observer(NULL);
 }
 
 // Verifies multiple selection gestures work (control-click, shift-click ...).
 TEST_F(TableViewTest, Multiselection) {
   // Configure the grouper so that there are three groups:
   // A 0
   //   1
   // B 5
   // C 2
@@ skipping 102 matching lines @@
   new_selection.set_active(0);
   new_selection.set_anchor(0);
   helper_->SetSelectionModel(new_selection);
   model_->RemoveRow(0);
   helper_->OnFocus();
 }
 
 // Tests that focusing the table will activate the first row, but only if
 // there's no active row.
 TEST_F(TableViewTest, InitialFocusActivatesFirstRow) {
+  // Focus selects the first row if there's nothing selected.
   EXPECT_EQ(-1, table_->selection_model().active());
   helper_->OnFocus();
+  base::RunLoop().RunUntilIdle();
   EXPECT_EQ(0, table_->selection_model().active());
 
+  // Focus does not affect selection if something's already selected.
   ui::ListSelectionModel new_selection;
   new_selection.set_active(1);
   helper_->SetSelectionModel(new_selection);
   EXPECT_EQ(1, table_->selection_model().active());
   helper_->OnFocus();
+  base::RunLoop().RunUntilIdle();
+  EXPECT_EQ(1, table_->selection_model().active());
+
+  // Focus does not affect selection if something gets selected synchronously
+  // after focus is taken.
+  new_selection.Clear();
+  helper_->SetSelectionModel(new_selection);
+  EXPECT_EQ(-1, table_->selection_model().active());
+  helper_->OnFocus();
+  new_selection.set_active(1);
+  helper_->SetSelectionModel(new_selection);
+  EXPECT_EQ(1, table_->selection_model().active());
+  base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1, table_->selection_model().active());
 
   // Remove all rows; focusing should not activate a non existent first row.
   while (model_->RowCount())
     model_->RemoveRow(0);
 
   new_selection.set_active(-1);
   helper_->SetSelectionModel(new_selection);
   helper_->OnFocus();
   EXPECT_EQ(-1, table_->selection_model().active());
 }
 
 }  // namespace views