Add Copy URL option to Omnibox context menu when URL is replaced by Instant Extended.

chrome/browser/ui/cocoa/location_bar/autocomplete_text_field_editor_unittest.mm

 // Copyright (c) 2011 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/location_bar/autocomplete_text_field_editor.h"
 
 #include "base/memory/scoped_nsobject.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/string_util.h"
 #include "chrome/app/chrome_command_ids.h"  // IDC_*
@@ skipping 184 matching lines @@
 
 // Test that -copy: is correctly delegated to the observer.
 TEST_F(AutocompleteTextFieldEditorObserverTest, Copy) {
   EXPECT_CALL(field_observer_, CanCopy())
       .WillOnce(Return(true));
   EXPECT_CALL(field_observer_, CopyToPasteboard(A<NSPasteboard*>()))
       .Times(1);
   [editor_ copy:nil];
 }
 
+// Test that -copyURL: is correctly delegated to the observer.
+TEST_F(AutocompleteTextFieldEditorObserverTest, CopyURL) {
+  EXPECT_CALL(field_observer_, CanCopy()).WillOnce(Return(true));
+  EXPECT_CALL(field_observer_,
+              CopyURLToPasteboard(A<NSPasteboard*>())).Times(1);
+  [editor_ copyURL:nil];
+}
+
 // Test that -cut: is correctly delegated to the observer and clears
 // the text field.
 TEST_F(AutocompleteTextFieldEditorObserverTest, Cut) {
   // Sets a string in the field.
   NSString* test_string = @"astring";
   EXPECT_CALL(field_observer_, OnDidBeginEditing());
   EXPECT_CALL(field_observer_, OnBeforeChange());
   EXPECT_CALL(field_observer_, OnDidChange());
   EXPECT_CALL(field_observer_, SelectionRangeForProposedRange(A<NSRange>()))
       .WillRepeatedly(ReturnArg<0>());
@@ skipping 12 matching lines @@
 }
 
 // Test that -pasteAndGo: is correctly delegated to the observer.
 TEST_F(AutocompleteTextFieldEditorObserverTest, PasteAndGo) {
   EXPECT_CALL(field_observer_, OnPasteAndGo());
   [editor_ pasteAndGo:nil];
 }
 
 // Test that the menu is constructed correctly when CanPasteAndGo().
 TEST_F(AutocompleteTextFieldEditorObserverTest, CanPasteAndGoMenu) {
-  EXPECT_CALL(field_observer_, CanPasteAndGo())
-      .WillOnce(Return(true));
+  EXPECT_CALL(field_observer_, ShouldAddCopyURL()).WillOnce(Return(false));
+  EXPECT_CALL(field_observer_, CanPasteAndGo()).WillOnce(Return(true));
   EXPECT_CALL(field_observer_, GetPasteActionStringId())
       .WillOnce(Return(IDS_PASTE_AND_GO));
 
   NSMenu* menu = [editor_ menuForEvent:nil];
   NSArray* items = [menu itemArray];
   ASSERT_EQ([items count], 6U);
   // TODO(shess): Check the titles, too?
   NSUInteger i = 0;  // Use an index to make future changes easier.
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(cut:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(copy:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(paste:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(pasteAndGo:));
   EXPECT_TRUE([[items objectAtIndex:i++] isSeparatorItem]);
 
   EXPECT_EQ([[items objectAtIndex:i] action], @selector(commandDispatch:));
   EXPECT_EQ([[items objectAtIndex:i] tag], IDC_EDIT_SEARCH_ENGINES);
   i++;
 }
 
 // Test that the menu is constructed correctly when !CanPasteAndGo().
 TEST_F(AutocompleteTextFieldEditorObserverTest, CannotPasteAndGoMenu) {
-  EXPECT_CALL(field_observer_, CanPasteAndGo())
-      .WillOnce(Return(false));
+  EXPECT_CALL(field_observer_, ShouldAddCopyURL()).WillOnce(Return(false));
+  EXPECT_CALL(field_observer_, CanPasteAndGo()).WillOnce(Return(false));
 
   NSMenu* menu = [editor_ menuForEvent:nil];
   NSArray* items = [menu itemArray];
   ASSERT_EQ([items count], 5U);
   // TODO(shess): Check the titles, too?
   NSUInteger i = 0;  // Use an index to make future changes easier.
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(cut:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(copy:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(paste:));
   EXPECT_TRUE([[items objectAtIndex:i++] isSeparatorItem]);
@@ skipping 18 matching lines @@
   NSMenu* menu = [editor_ menuForEvent:nil];
   NSArray* items = [menu itemArray];
   ASSERT_EQ([items count], 3U);
   // TODO(shess): Check the titles, too?
   NSUInteger i = 0;  // Use an index to make future changes easier.
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(cut:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(copy:));
   EXPECT_EQ([[items objectAtIndex:i++] action], @selector(paste:));
 }
 
+// Test that the menu is constructed correctly when ShouldAddCopyURL() returns
+// true.
+TEST_F(AutocompleteTextFieldEditorObserverTest, ShouldAllowCopyURLMenu) {
+  EXPECT_CALL(field_observer_, ShouldAddCopyURL()).WillOnce(Return(true));
+  EXPECT_CALL(field_observer_, CanPasteAndGo()).WillOnce(Return(false));
+
+  NSMenu* menu = [editor_ menuForEvent:nil];
+  NSArray* items = [menu itemArray];
+  ASSERT_EQ([items count], 6U);
+  // TODO(shess): Check the titles, too?
+  NSUInteger i = 0;  // Use an index to make future changes easier.
+  EXPECT_EQ([[items objectAtIndex:i++] action], @selector(cut:));
+  EXPECT_EQ([[items objectAtIndex:i++] action], @selector(copy:));
+  EXPECT_EQ([[items objectAtIndex:i++] action], @selector(copyURL:));
+  EXPECT_EQ([[items objectAtIndex:i++] action], @selector(paste:));
+  EXPECT_TRUE([[items objectAtIndex:i++] isSeparatorItem]);
+
+  EXPECT_EQ([[items objectAtIndex:i] action], @selector(commandDispatch:));
+  EXPECT_EQ([[items objectAtIndex:i] tag], IDC_EDIT_SEARCH_ENGINES);
+}
+
+
 }  // namespace