MacViews: Tweak ComboboxTest.GetTextForRowTest to get it to pass.

ui/views/controls/combobox/combobox_unittest.cc

 // Copyright 2013 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/combobox/combobox.h"
 
 #include <set>
 
 #include "base/basictypes.h"
 #include "base/strings/utf_string_conversions.h"
@@ skipping 74 matching lines @@
 
   DISALLOW_COPY_AND_ASSIGN(TestCombobox);
 };
 
 // A concrete class is needed to test the combobox.
 class TestComboboxModel : public ui::ComboboxModel {
  public:
   TestComboboxModel() {}
   virtual ~TestComboboxModel() {}
 
+  static const int kItemCount = 10;
+
   // ui::ComboboxModel:
   virtual int GetItemCount() const OVERRIDE {
-    return 10;
+    return kItemCount;
   }
   virtual base::string16 GetItemAt(int index) OVERRIDE {
     if (IsItemSeparatorAt(index)) {
       NOTREACHED();
       return ASCIIToUTF16("SEPARATOR");
     }
     return ASCIIToUTF16(index % 2 == 0 ? "PEANUT BUTTER" : "JELLY");
   }
   virtual bool IsItemSeparatorAt(int index) OVERRIDE {
     return separators_.find(index) != separators_.end();
   }
 
+  virtual int GetDefaultIndex() const OVERRIDE {
+    // Return the first index that is not a separator.
+    for (int index = 0; index < kItemCount; ++index) {
+      if (separators_.find(index) == separators_.end())
+        return index;
+    }
+    NOTREACHED();
+    return 0;
+  }
+
   void SetSeparators(const std::set<int>& separators) {
     separators_ = separators;
   }
 
  private:
   std::set<int> separators_;
 
   DISALLOW_COPY_AND_ASSIGN(TestComboboxModel);
 };
 
@@ skipping 73 matching lines @@
 class ComboboxTest : public ViewsTestBase {
  public:
   ComboboxTest() : widget_(NULL), combobox_(NULL) {}
 
   virtual void TearDown() OVERRIDE {
     if (widget_)
       widget_->Close();
     ViewsTestBase::TearDown();
   }
 
-  void InitCombobox() {
+  void InitCombobox(const std::set<int>* separators) {
     model_.reset(new TestComboboxModel());
 
+    if (separators)
+      model_->SetSeparators(*separators);
+
     ASSERT_FALSE(combobox_);
     combobox_ = new TestCombobox(model_.get());
     combobox_->set_id(1);
 
     widget_ = new Widget;
     Widget::InitParams params = CreateParams(Widget::InitParams::TYPE_POPUP);
     params.bounds = gfx::Rect(200, 200, 200, 200);
     widget_->Init(params);
     View* container = new View();
     widget_->SetContentsView(container);
@@ skipping 39 matching lines @@
   Widget* widget_;
 
   // |combobox_| will be allocated InitCombobox() and then owned by |widget_|.
   TestCombobox* combobox_;
 
   // Combobox does not take ownership of the model, hence it needs to be scoped.
   scoped_ptr<TestComboboxModel> model_;
 };
 
 TEST_F(ComboboxTest, KeyTest) {
-  InitCombobox();
+  InitCombobox(NULL);
   SendKeyEvent(ui::VKEY_END);
   EXPECT_EQ(combobox_->selected_index() + 1, model_->GetItemCount());
   SendKeyEvent(ui::VKEY_HOME);
   EXPECT_EQ(combobox_->selected_index(), 0);
   SendKeyEvent(ui::VKEY_DOWN);
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(combobox_->selected_index(), 2);
   SendKeyEvent(ui::VKEY_RIGHT);
   EXPECT_EQ(combobox_->selected_index(), 2);
   SendKeyEvent(ui::VKEY_LEFT);
@@ skipping 21 matching lines @@
   widget_->Init(params);
   View* container = new View();
   widget_->SetContentsView(container);
   container->AddChildView(combobox_);
   EXPECT_FALSE(combobox_->enabled());
 }
 
 // Verifies that we don't select a separator line in combobox when navigating
 // through keyboard.
 TEST_F(ComboboxTest, SkipSeparatorSimple) {
-  InitCombobox();
   std::set<int> separators;
   separators.insert(2);
-  model_->SetSeparators(separators);
+  InitCombobox(&separators);
   EXPECT_EQ(0, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(1, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(3, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_UP);
   EXPECT_EQ(1, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_HOME);
   EXPECT_EQ(0, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_PRIOR);
   EXPECT_EQ(0, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_END);
   EXPECT_EQ(9, combobox_->selected_index());
 }
 
 // Verifies that we never select the separator that is in the beginning of the
 // combobox list when navigating through keyboard.
 TEST_F(ComboboxTest, SkipSeparatorBeginning) {
-  InitCombobox();
   std::set<int> separators;
   separators.insert(0);
-  model_->SetSeparators(separators);
-  EXPECT_EQ(0, combobox_->selected_index());
-  SendKeyEvent(ui::VKEY_DOWN);
+  InitCombobox(&separators);
   EXPECT_EQ(1, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(2, combobox_->selected_index());
+  SendKeyEvent(ui::VKEY_DOWN);
+  EXPECT_EQ(3, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_UP);
-  EXPECT_EQ(1, combobox_->selected_index());
+  EXPECT_EQ(2, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_HOME);
   EXPECT_EQ(1, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_PRIOR);
   EXPECT_EQ(1, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_END);
   EXPECT_EQ(9, combobox_->selected_index());
 }
 
 // Verifies that we never select the separator that is in the end of the
 // combobox list when navigating through keyboard.
 TEST_F(ComboboxTest, SkipSeparatorEnd) {
-  InitCombobox();
   std::set<int> separators;
-  separators.insert(model_->GetItemCount() - 1);
-  model_->SetSeparators(separators);
+  separators.insert(TestComboboxModel::kItemCount - 1);
+  InitCombobox(&separators);
   combobox_->SetSelectedIndex(8);
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(8, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_UP);
   EXPECT_EQ(7, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_END);
   EXPECT_EQ(8, combobox_->selected_index());
 }
 
 // Verifies that we never select any of the adjacent separators (multiple
 // consecutive) that appear in the beginning of the combobox list when
 // navigating through keyboard.
 TEST_F(ComboboxTest, SkipMultipleSeparatorsAtBeginning) {
-  InitCombobox();
   std::set<int> separators;
   separators.insert(0);
   separators.insert(1);
   separators.insert(2);
-  model_->SetSeparators(separators);
-  EXPECT_EQ(0, combobox_->selected_index());
+  InitCombobox(&separators);
+  EXPECT_EQ(3, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_DOWN);
-  EXPECT_EQ(3, combobox_->selected_index());
+  EXPECT_EQ(4, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_UP);
   EXPECT_EQ(3, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_NEXT);
   EXPECT_EQ(9, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_HOME);
   EXPECT_EQ(3, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_END);
   EXPECT_EQ(9, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_PRIOR);
   EXPECT_EQ(3, combobox_->selected_index());
 }
 
 // Verifies that we never select any of the adjacent separators (multiple
 // consecutive) that appear in the middle of the combobox list when navigating
 // through keyboard.
 TEST_F(ComboboxTest, SkipMultipleAdjacentSeparatorsAtMiddle) {
-  InitCombobox();
   std::set<int> separators;
   separators.insert(4);
   separators.insert(5);
   separators.insert(6);
-  model_->SetSeparators(separators);
+  InitCombobox(&separators);
   combobox_->SetSelectedIndex(3);
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(7, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_UP);
   EXPECT_EQ(3, combobox_->selected_index());
 }
 
 // Verifies that we never select any of the adjacent separators (multiple
 // consecutive) that appear in the end of the combobox list when navigating
 // through keyboard.
 TEST_F(ComboboxTest, SkipMultipleSeparatorsAtEnd) {
-  InitCombobox();
   std::set<int> separators;
   separators.insert(7);
   separators.insert(8);
   separators.insert(9);
-  model_->SetSeparators(separators);
+  InitCombobox(&separators);
   combobox_->SetSelectedIndex(6);
   SendKeyEvent(ui::VKEY_DOWN);
   EXPECT_EQ(6, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_UP);
   EXPECT_EQ(5, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_HOME);
   EXPECT_EQ(0, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_NEXT);
   EXPECT_EQ(6, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_PRIOR);
   EXPECT_EQ(0, combobox_->selected_index());
   SendKeyEvent(ui::VKEY_END);
   EXPECT_EQ(6, combobox_->selected_index());
 }
 
 TEST_F(ComboboxTest, GetTextForRowTest) {
-  InitCombobox();
   std::set<int> separators;
   separators.insert(0);
   separators.insert(1);
   separators.insert(9);
-  model_->SetSeparators(separators);
+  InitCombobox(&separators);
   for (int i = 0; i < combobox_->GetRowCount(); ++i) {
     if (separators.count(i) != 0) {
       EXPECT_TRUE(combobox_->GetTextForRow(i).empty()) << i;
     } else {
       EXPECT_EQ(ASCIIToUTF16(i % 2 == 0 ? "PEANUT BUTTER" : "JELLY"),
                 combobox_->GetTextForRow(i)) << i;
     }
   }
 }
 
 // Verifies selecting the first matching value (and returning whether found).
 TEST_F(ComboboxTest, SelectValue) {
-  InitCombobox();
+  InitCombobox(NULL);
   ASSERT_EQ(model_->GetDefaultIndex(), combobox_->selected_index());
   EXPECT_TRUE(combobox_->SelectValue(ASCIIToUTF16("PEANUT BUTTER")));
   EXPECT_EQ(0, combobox_->selected_index());
   EXPECT_TRUE(combobox_->SelectValue(ASCIIToUTF16("JELLY")));
   EXPECT_EQ(1, combobox_->selected_index());
   EXPECT_FALSE(combobox_->SelectValue(ASCIIToUTF16("BANANAS")));
   EXPECT_EQ(1, combobox_->selected_index());
 
   // With the action style, the selected index is always 0.
   combobox_->SetStyle(Combobox::STYLE_ACTION);
   EXPECT_FALSE(combobox_->SelectValue(ASCIIToUTF16("PEANUT BUTTER")));
   EXPECT_EQ(0, combobox_->selected_index());
   EXPECT_FALSE(combobox_->SelectValue(ASCIIToUTF16("JELLY")));
   EXPECT_EQ(0, combobox_->selected_index());
   EXPECT_FALSE(combobox_->SelectValue(ASCIIToUTF16("BANANAS")));
   EXPECT_EQ(0, combobox_->selected_index());
 }
 
 TEST_F(ComboboxTest, SelectIndexActionStyle) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   // With the action style, the selected index is always 0.
   combobox_->SetStyle(Combobox::STYLE_ACTION);
   combobox_->SetSelectedIndex(1);
   EXPECT_EQ(0, combobox_->selected_index());
   combobox_->SetSelectedIndex(2);
   EXPECT_EQ(0, combobox_->selected_index());
   combobox_->SetSelectedIndex(3);
   EXPECT_EQ(0, combobox_->selected_index());
 }
@@ skipping 12 matching lines @@
   // |combobox| will be deleted on change.
   combobox = new TestCombobox(&model);
   evil_listener.reset(new EvilListener());
   combobox->set_listener(evil_listener.get());
   combobox->SetStyle(Combobox::STYLE_ACTION);
   ASSERT_NO_FATAL_FAILURE(combobox->ExecuteCommand(2));
   EXPECT_TRUE(evil_listener->deleted());
 }
 
 TEST_F(ComboboxTest, Click) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   TestComboboxListener listener;
   combobox_->set_listener(&listener);
 
   combobox_->Layout();
 
   // Click the left side. The menu is shown.
   TestMenuRunnerHandler* test_menu_runner_handler = new TestMenuRunnerHandler();
   scoped_ptr<MenuRunnerHandler> menu_runner_handler(test_menu_runner_handler);
   test::MenuRunnerTestAPI test_api(
       combobox_->dropdown_list_menu_runner_.get());
   test_api.SetMenuRunnerHandler(menu_runner_handler.Pass());
   PerformClick(gfx::Point(combobox_->x() + 1,
                           combobox_->y() + combobox_->height() / 2));
   EXPECT_FALSE(listener.on_perform_action_called());
   EXPECT_TRUE(test_menu_runner_handler->executed());
 }
 
 TEST_F(ComboboxTest, ClickButDisabled) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   TestComboboxListener listener;
   combobox_->set_listener(&listener);
 
   combobox_->Layout();
   combobox_->SetEnabled(false);
 
   // Click the left side, but nothing happens since the combobox is disabled.
   TestMenuRunnerHandler* test_menu_runner_handler = new TestMenuRunnerHandler();
   scoped_ptr<MenuRunnerHandler> menu_runner_handler(test_menu_runner_handler);
   test::MenuRunnerTestAPI test_api(
       combobox_->dropdown_list_menu_runner_.get());
   test_api.SetMenuRunnerHandler(menu_runner_handler.Pass());
   PerformClick(gfx::Point(combobox_->x() + 1,
                           combobox_->y() + combobox_->height() / 2));
   EXPECT_FALSE(listener.on_perform_action_called());
   EXPECT_FALSE(test_menu_runner_handler->executed());
 }
 
 TEST_F(ComboboxTest, NotifyOnClickWithReturnKey) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   TestComboboxListener listener;
   combobox_->set_listener(&listener);
 
   // With STYLE_NORMAL, the click event is ignored.
   SendKeyEvent(ui::VKEY_RETURN);
   EXPECT_FALSE(listener.on_perform_action_called());
 
   // With STYLE_ACTION, the click event is notified.
   combobox_->SetStyle(Combobox::STYLE_ACTION);
   SendKeyEvent(ui::VKEY_RETURN);
   EXPECT_TRUE(listener.on_perform_action_called());
   EXPECT_EQ(0, listener.perform_action_index());
 }
 
 TEST_F(ComboboxTest, NotifyOnClickWithSpaceKey) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   TestComboboxListener listener;
   combobox_->set_listener(&listener);
 
   // With STYLE_NORMAL, the click event is ignored.
   SendKeyEvent(ui::VKEY_SPACE);
   EXPECT_FALSE(listener.on_perform_action_called());
   SendKeyEventWithType(ui::VKEY_SPACE, ui::ET_KEY_RELEASED);
   EXPECT_FALSE(listener.on_perform_action_called());
 
   // With STYLE_ACTION, the click event is notified after releasing.
   combobox_->SetStyle(Combobox::STYLE_ACTION);
   SendKeyEvent(ui::VKEY_SPACE);
   EXPECT_FALSE(listener.on_perform_action_called());
   SendKeyEventWithType(ui::VKEY_SPACE, ui::ET_KEY_RELEASED);
   EXPECT_TRUE(listener.on_perform_action_called());
   EXPECT_EQ(0, listener.perform_action_index());
 }
 
 TEST_F(ComboboxTest, NotifyOnClickWithMouse) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   TestComboboxListener listener;
   combobox_->set_listener(&listener);
 
   combobox_->SetStyle(Combobox::STYLE_ACTION);
   combobox_->Layout();
 
   // Click the right side (arrow button). The menu is shown.
   TestMenuRunnerHandler* test_menu_runner_handler = new TestMenuRunnerHandler();
   scoped_ptr<MenuRunnerHandler> menu_runner_handler(test_menu_runner_handler);
@@ skipping 13 matching lines @@
       new test::MenuRunnerTestAPI(combobox_->dropdown_list_menu_runner_.get()));
   test_api->SetMenuRunnerHandler(menu_runner_handler.Pass());
   PerformClick(gfx::Point(combobox_->x() + 1,
                           combobox_->y() + combobox_->height() / 2));
   EXPECT_TRUE(listener.on_perform_action_called());
   EXPECT_FALSE(test_menu_runner_handler->executed());
   EXPECT_EQ(0, listener.perform_action_index());
 }
 
 TEST_F(ComboboxTest, ConsumingPressKeyEvents) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   EXPECT_FALSE(combobox_->OnKeyPressed(
       ui::KeyEvent(ui::ET_KEY_PRESSED, ui::VKEY_RETURN, 0, false)));
   EXPECT_FALSE(combobox_->OnKeyPressed(
       ui::KeyEvent(ui::ET_KEY_PRESSED, ui::VKEY_SPACE, 0, false)));
 
   // When the combobox's style is STYLE_ACTION, pressing events of a space key
   // or an enter key will be consumed.
   combobox_->SetStyle(Combobox::STYLE_ACTION);
   EXPECT_TRUE(combobox_->OnKeyPressed(
@@ skipping 30 matching lines @@
   combobox.SetStyle(Combobox::STYLE_NORMAL);
   EXPECT_EQ(long_item_width, combobox.content_size_.width());
 
   // When the style is STYLE_ACTION, the width will fit with the first items'
   // width.
   combobox.SetStyle(Combobox::STYLE_ACTION);
   EXPECT_EQ(short_item_width, combobox.content_size_.width());
 }
 
 TEST_F(ComboboxTest, TypingPrefixNotifiesListener) {
-  InitCombobox();
+  InitCombobox(NULL);
 
   TestComboboxListener listener;
   combobox_->set_listener(&listener);
 
   // Type the first character of the second menu item ("JELLY").
   combobox_->GetTextInputClient()->InsertChar('J', ui::EF_NONE);
   EXPECT_EQ(1, listener.actions_performed());
   EXPECT_EQ(1, listener.perform_action_index());
 
   // Type the second character of "JELLY", item shouldn't change and
   // OnPerformAction() shouldn't be re-called.
   combobox_->GetTextInputClient()->InsertChar('E', ui::EF_NONE);
   EXPECT_EQ(1, listener.actions_performed());
   EXPECT_EQ(1, listener.perform_action_index());
 
   // Clears the typed text.
   combobox_->OnBlur();
 
   // Type the first character of "PEANUT BUTTER", which should change the
   // selected index and perform an action.
   combobox_->GetTextInputClient()->InsertChar('P', ui::EF_NONE);
   EXPECT_EQ(2, listener.actions_performed());
   EXPECT_EQ(2, listener.perform_action_index());
 }
 
 }  // namespace views