Add multiple selection support to chooser on desktops

chrome/browser/ui/cocoa/extensions/chooser_dialog_cocoa_controller_unittest.mm

 // Copyright 2016 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.
 
 #import "chrome/browser/ui/cocoa/extensions/chooser_dialog_cocoa_controller.h"
 
 #import <Cocoa/Cocoa.h>
 
 #include <memory>
 #include <vector>
@@ skipping 831 matching lines @@
   mock_chooser_controller_->OptionAdded(
       base::ASCIIToUTF16("b"), MockChooserController::kSignalStrengthLevel0Bar,
       MockChooserController::ConnectedPairedStatus::NONE);
   mock_chooser_controller_->OptionAdded(
       base::ASCIIToUTF16("c"), MockChooserController::kSignalStrengthLevel1Bar,
       MockChooserController::ConnectedPairedStatus::NONE);
 
   // Select option 0 and press "Connect" button.
   [table_view_ selectRowIndexes:[NSIndexSet indexSetWithIndex:0]
            byExtendingSelection:NO];
-  EXPECT_CALL(*mock_chooser_controller_, Select(0)).Times(1);
+  std::vector<size_t> indices(1);
+  indices[0] = 0;
+  EXPECT_CALL(*mock_chooser_controller_, Select(indices)).Times(1);
   EXPECT_CALL(*mock_chooser_controller_, Cancel()).Times(0);
   [connect_button_ performClick:chooser_dialog_controller_];
 
   // Select option 2 and press "Connect" button.
   [table_view_ selectRowIndexes:[NSIndexSet indexSetWithIndex:2]
            byExtendingSelection:NO];
-  EXPECT_CALL(*mock_chooser_controller_, Select(2)).Times(1);
+  indices[0] = 2;
+  EXPECT_CALL(*mock_chooser_controller_, Select(indices)).Times(1);
   EXPECT_CALL(*mock_chooser_controller_, Cancel()).Times(0);
   [connect_button_ performClick:chooser_dialog_controller_];
 }
 
 TEST_F(ChooserDialogCocoaControllerTest, SelectAnOptionAndPressCancelButton) {
   CreateChooserDialog();
 
   mock_chooser_controller_->OptionAdded(
       base::ASCIIToUTF16("a"),
       MockChooserController::kNoSignalStrengthLevelImage,
@@ skipping 215 matching lines @@
   EXPECT_TRUE(spinner_.hidden);
   EXPECT_TRUE(scanning_message_.hidden);
   EXPECT_FALSE(word_connector_.hidden);
   EXPECT_FALSE(rescan_button_.hidden);
   EXPECT_NSEQ(l10n_util::GetNSString(IDS_BLUETOOTH_DEVICE_CHOOSER_RE_SCAN),
               rescan_button_.title);
   EXPECT_TRUE(connect_button_.enabled);
   EXPECT_TRUE(cancel_button_.enabled);
 }
 
+TEST_F(ChooserDialogCocoaControllerTest, MultipleSelection) {
+  CreateChooserDialog();
+  [table_view_ setAllowsMultipleSelection:YES];
+
+  mock_chooser_controller_->OptionAdded(
+      base::ASCIIToUTF16("a"),
+      MockChooserController::kNoSignalStrengthLevelImage,
+      MockChooserController::ConnectedPairedStatus::CONNECTED |
+          MockChooserController::ConnectedPairedStatus::PAIRED);
+  mock_chooser_controller_->OptionAdded(
+      base::ASCIIToUTF16("b"), MockChooserController::kSignalStrengthLevel0Bar,
+      MockChooserController::ConnectedPairedStatus::NONE);
+  mock_chooser_controller_->OptionAdded(
+      base::ASCIIToUTF16("c"), MockChooserController::kSignalStrengthLevel1Bar,
+      MockChooserController::ConnectedPairedStatus::NONE);
+  mock_chooser_controller_->OptionAdded(
+      base::ASCIIToUTF16("d"), MockChooserController::kSignalStrengthLevel2Bar,
+      MockChooserController::ConnectedPairedStatus::NONE);
+  mock_chooser_controller_->OptionAdded(
+      base::ASCIIToUTF16("e"), MockChooserController::kSignalStrengthLevel3Bar,
+      MockChooserController::ConnectedPairedStatus::NONE);
+  mock_chooser_controller_->OptionAdded(
+      base::ASCIIToUTF16("f"), MockChooserController::kSignalStrengthLevel4Bar,
+      MockChooserController::ConnectedPairedStatus::NONE);
+
+  NSMutableIndexSet* selected_rows = [NSMutableIndexSet indexSet];
+  // Select options "a", "c", "d".
+  [selected_rows addIndex:0];
+  [selected_rows addIndex:2];
+  [selected_rows addIndex:3];
+  [table_view_ selectRowIndexes:selected_rows byExtendingSelection:NO];
+
+  // The options are: [a] b [c] [d] e f. The option with [] is selected.
+  EXPECT_NSEQ(selected_rows, [table_view_ selectedRowIndexes]);
+  ExpectRowImageIsConnectedImage(0, SK_ColorWHITE);
+  ExpectSignalStrengthLevelImageIs(
+      1, MockChooserController::kSignalStrengthLevel0Bar,
+      MockChooserController::kImageColorUnselected);
+  ExpectSignalStrengthLevelImageIs(
+      2, MockChooserController::kSignalStrengthLevel1Bar,
+      MockChooserController::kImageColorSelected);
+  ExpectSignalStrengthLevelImageIs(
+      3, MockChooserController::kSignalStrengthLevel2Bar,
+      MockChooserController::kImageColorSelected);
+  ExpectSignalStrengthLevelImageIs(
+      4, MockChooserController::kSignalStrengthLevel3Bar,
+      MockChooserController::kImageColorUnselected);
+  ExpectSignalStrengthLevelImageIs(
+      5, MockChooserController::kSignalStrengthLevel4Bar,
+      MockChooserController::kImageColorUnselected);
+  ExpectRowTextColorIs(0, [NSColor whiteColor]);
+  ExpectRowTextColorIs(1, [NSColor blackColor]);
+  ExpectRowTextColorIs(2, [NSColor whiteColor]);
+  ExpectRowTextColorIs(3, [NSColor whiteColor]);
+  ExpectRowTextColorIs(4, [NSColor blackColor]);
+  ExpectRowTextColorIs(5, [NSColor blackColor]);
+  ExpectPairedTextColorIs(
+      0, skia::SkColorToCalibratedNSColor(gfx::kGoogleGreen300));
+  EXPECT_TRUE(connect_button_.enabled);
+
+  // Remove option "b".
+  // The options are: [a] [c] [d] e f.
+  mock_chooser_controller_->OptionRemoved(base::ASCIIToUTF16("b"));
+  EXPECT_TRUE([table_view_ isRowSelected:0]);
+  EXPECT_TRUE([table_view_ isRowSelected:1]);
+  EXPECT_TRUE([table_view_ isRowSelected:2]);
+  EXPECT_FALSE([table_view_ isRowSelected:3]);
+  EXPECT_FALSE([table_view_ isRowSelected:4]);
+  EXPECT_TRUE(connect_button_.enabled);
+
+  // Remove option "c".
+  // The options are: [a] [d] e f.
+  mock_chooser_controller_->OptionRemoved(base::ASCIIToUTF16("c"));
+  EXPECT_TRUE([table_view_ isRowSelected:0]);
+  EXPECT_TRUE([table_view_ isRowSelected:1]);
+  EXPECT_FALSE([table_view_ isRowSelected:2]);
+  EXPECT_FALSE([table_view_ isRowSelected:3]);
+  EXPECT_TRUE(connect_button_.enabled);
+
+  // Remove option "a".
+  // The options are: [d] e f.
+  mock_chooser_controller_->OptionRemoved(base::ASCIIToUTF16("a"));
+  EXPECT_TRUE([table_view_ isRowSelected:0]);
+  EXPECT_FALSE([table_view_ isRowSelected:1]);
+  EXPECT_FALSE([table_view_ isRowSelected:2]);
+  EXPECT_TRUE(connect_button_.enabled);
+
+  // Remove option "d".
+  // The options are: e f.
+  mock_chooser_controller_->OptionRemoved(base::ASCIIToUTF16("d"));
+  EXPECT_FALSE([table_view_ isRowSelected:0]);
+  EXPECT_FALSE([table_view_ isRowSelected:1]);
+  EXPECT_FALSE(connect_button_.enabled);
+}
+
 TEST_F(ChooserDialogCocoaControllerTest, PressAdapterOffHelpButton) {
   CreateChooserDialog();
 
   EXPECT_CALL(*mock_chooser_controller_, OpenAdapterOffHelpUrl()).Times(1);
   [adapter_off_help_button_ performClick:chooser_dialog_controller_];
 }
 
 TEST_F(ChooserDialogCocoaControllerTest, PressRescanButton) {
   CreateChooserDialog();
 
@@ skipping 17 matching lines @@
       MockChooserController::ConnectedPairedStatus::NONE);
 
   // Select option 0 and press "Get help" button.
   [table_view_ selectRowIndexes:[NSIndexSet indexSetWithIndex:0]
            byExtendingSelection:NO];
   EXPECT_CALL(*mock_chooser_controller_, Select(testing::_)).Times(0);
   EXPECT_CALL(*mock_chooser_controller_, Cancel()).Times(0);
   EXPECT_CALL(*mock_chooser_controller_, OpenHelpCenterUrl()).Times(1);
   [help_button_ performClick:chooser_dialog_controller_];
 }