Mac Picker UI: Screen Auto Selection

chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller_unittest.mm

 // 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.
 
 #import "chrome/browser/ui/cocoa/media_picker/desktop_media_picker_controller.h"
 
 #include "base/bind.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/media/desktop_media_list_observer.h"
@@ skipping 149 matching lines @@
   EXPECT_EQ(2U, [items count]);
   EXPECT_NSEQ(@"0", [[items objectAtIndex:0] imageTitle]);
   EXPECT_EQ(nil, [[items objectAtIndex:0] imageRepresentation]);
   EXPECT_NSEQ(@"1", [[items objectAtIndex:1] imageTitle]);
   EXPECT_TRUE([[items objectAtIndex:1] imageRepresentation] != nil);
 }
 
 TEST_F(DesktopMediaPickerControllerTest, ClickShareScreen) {
   [controller_ showWindow:nil];
   ChangeType(DesktopMediaID::TYPE_SCREEN);
+
+  EXPECT_FALSE([[controller_ shareButton] isEnabled]);
   AddScreen(0);
   screen_list_->SetSourceThumbnail(0);
+  // First screen will be automatically selected.
+  EXPECT_TRUE([[controller_ shareButton] isEnabled]);
+
   AddScreen(1);
   screen_list_->SetSourceThumbnail(1);
 
   EXPECT_EQ(2U, [[controller_ screenItems] count]);
-  EXPECT_FALSE([[controller_ shareButton] isEnabled]);
-
-  NSIndexSet* index_set = [NSIndexSet indexSetWithIndex:1];
-  [[controller_ screenBrowser] setSelectionIndexes:index_set
-                              byExtendingSelection:NO];
-  EXPECT_TRUE([[controller_ shareButton] isEnabled]);
 
   [[controller_ shareButton] performClick:nil];
   EXPECT_TRUE(WaitForCallback());
-  EXPECT_EQ(screen_list_->GetSource(1).id, source_reported_);
+  EXPECT_EQ(screen_list_->GetSource(0).id, source_reported_);
 }
 
 TEST_F(DesktopMediaPickerControllerTest, ClickShareWindow) {
   [controller_ showWindow:nil];
   ChangeType(DesktopMediaID::TYPE_WINDOW);
   AddWindow(0);
   window_list_->SetSourceThumbnail(0);
   AddWindow(1);
   window_list_->SetSourceThumbnail(1);
 
@@ skipping 204 matching lines @@
   tab_list_->RemoveSource(0);
   selected_index = [[browser selectedRowIndexes] firstIndex];
   EXPECT_EQ(1, [[items objectAtIndex:selected_index] sourceID].id);
 
   // Change source type back and forth, browser should memorize the selection.
   ChangeType(DesktopMediaID::TYPE_SCREEN);
   ChangeType(DesktopMediaID::TYPE_WEB_CONTENTS);
   selected_index = [[browser selectedRowIndexes] firstIndex];
   EXPECT_EQ(1, [[items objectAtIndex:selected_index] sourceID].id);
 }
+
+TEST_F(DesktopMediaPickerControllerTest, SingleScreenNoLabel) {
+  [controller_ showWindow:nil];
+  ChangeType(DesktopMediaID::TYPE_SCREEN);
+
+  NSArray* items = [controller_ screenItems];
+
+  AddScreen(0);
+  screen_list_->SetSourceThumbnail(0);
+  EXPECT_EQ(1U, [items count]);
+  EXPECT_EQ(nil, [[items objectAtIndex:0] imageTitle]);
+
+  AddScreen(1);
+  screen_list_->SetSourceThumbnail(1);
+  EXPECT_EQ(2U, [items count]);
+  EXPECT_NE(nil, [[items objectAtIndex:0] imageTitle]);
+  EXPECT_NE(nil, [[items objectAtIndex:1] imageTitle]);
+}