[Mac] Location icon in omnibox as drag source.

chrome/browser/cocoa/autocomplete_text_field_unittest.mm

 // Copyright (c) 2010 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 <Cocoa/Cocoa.h>
 
 #include "app/resource_bundle.h"
 #import "base/cocoa_protocols_mac.h"
 #include "base/scoped_nsobject.h"
 #import "chrome/browser/cocoa/autocomplete_text_field.h"
 #import "chrome/browser/cocoa/autocomplete_text_field_cell.h"
 #import "chrome/browser/cocoa/autocomplete_text_field_editor.h"
 #import "chrome/browser/cocoa/autocomplete_text_field_unittest_helper.h"
 #import "chrome/browser/cocoa/cocoa_test_helper.h"
 #include "grit/theme_resources.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 
 using ::testing::InSequence;
 using ::testing::Return;
 using ::testing::StrictMock;
 
 namespace {
 class MockLocationIconView : public LocationBarViewMac::LocationIconView {
  public:
   MockLocationIconView(LocationBarViewMac* owner)
       : LocationBarViewMac::LocationIconView(owner) {}
 
+  // |LocationBarViewMac::LocationIconView::GetDragPasteboard()|
+  // expects things to be more initialized than they will be, and
+  // will crash if it is called.
+  virtual NSPasteboard* GetDragPasteboard() {
+    return [NSPasteboard pasteboardWithUniqueName];
+  }
+
   // We can't use gmock's MOCK_METHOD macro, because it doesn't like the
   // NSRect argument to OnMousePressed.
   virtual void OnMousePressed(NSRect bounds) {
     mouse_was_pressed_ = true;
   }
   bool mouse_was_pressed_;
 };
 
 class MockPageActionImageView : public LocationBarViewMac::PageActionImageView {
  public:
@@ skipping 547 matching lines @@
 
   MockLocationIconView location_icon_view(NULL);
   [cell setLocationIconView:&location_icon_view];
   location_icon_view.SetImage(
       ResourceBundle::GetSharedInstance().GetNSImageNamed(
           IDR_OMNIBOX_HTTPS_VALID));
   location_icon_view.SetVisible(true);
 
   NSRect iconFrame([cell locationIconFrameForFrame:[field_ bounds]]);
   NSPoint location(NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame)));
-  NSEvent* event(Event(field_, location, NSLeftMouseDown, 1));
+  NSEvent* downEvent(Event(field_, location, NSLeftMouseDown, 1));
+  NSEvent* upEvent(Event(field_, location, NSLeftMouseUp, 1));
 
-  [field_ mouseDown:event];
+  // Since location icon can be dragged, the mouse-press is sent on
+  // mouse-up.
+  [NSApp postEvent:upEvent atStart:YES];
+  [field_ mouseDown:downEvent];
   EXPECT_TRUE(location_icon_view.mouse_was_pressed_);
+
+  // TODO(shess): Test that mouse drags are initiated if the next
+  // event is a drag, or if the mouse-up takes too long to arrive.
 }
 
 // Clicking a Page Action icon should call its OnMousePressed.
 TEST_F(AutocompleteTextFieldTest, PageActionMouseDown) {
   AutocompleteTextFieldCell* cell = [field_ autocompleteTextFieldCell];
 
-  MockLocationIconView location_icon_view(NULL);
-  location_icon_view.SetImage(
-      ResourceBundle::GetSharedInstance().GetNSImageNamed(
-          IDR_OMNIBOX_HTTPS_VALID));
-  [cell setLocationIconView:&location_icon_view];
-
   MockPageActionImageView page_action_view;
   NSImage* image = [NSImage imageNamed:@"NSApplicationIcon"];
   page_action_view.SetImage(image);
 
   MockPageActionImageView page_action_view2;
   page_action_view2.SetImage(image);
 
   TestPageActionViewList list;
   list.Add(&page_action_view);
   list.Add(&page_action_view2);
   [cell setPageActionViewList:&list];
 
-  // One page action, no lock.
-  location_icon_view.SetVisible(false);
+  // One page action.
   page_action_view.SetVisible(true);
   page_action_view2.SetVisible(false);
   NSRect iconFrame([cell pageActionFrameForIndex:0 inFrame:[field_ bounds]]);
   NSPoint location(NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame)));
   NSEvent* event(Event(field_, location, NSLeftMouseDown, 1));
 
   [field_ mouseDown:event];
   EXPECT_TRUE(page_action_view.MouseWasPressed());
 
   // Two page actions, no lock.
   page_action_view2.SetVisible(true);
   iconFrame = [cell pageActionFrameForIndex:0 inFrame:[field_ bounds]];
   location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
   event = Event(field_, location, NSLeftMouseDown, 1);
 
   [field_ mouseDown:event];
   EXPECT_TRUE(page_action_view.MouseWasPressed());
 
   iconFrame = [cell pageActionFrameForIndex:1 inFrame:[field_ bounds]];
   location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
   event = Event(field_, location, NSLeftMouseDown, 1);
 
   [field_ mouseDown:event];
   EXPECT_TRUE(page_action_view.MouseWasPressed());
 
-  // Two page actions plus lock.
-  location_icon_view.SetVisible(true);
+  // Two page actions.
   iconFrame = [cell pageActionFrameForIndex:0 inFrame:[field_ bounds]];
   location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
   event = Event(field_, location, NSLeftMouseDown, 1);
 
   [field_ mouseDown:event];
   EXPECT_TRUE(page_action_view.MouseWasPressed());
 
   iconFrame = [cell pageActionFrameForIndex:1 inFrame:[field_ bounds]];
   location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
   event = Event(field_, location, NSLeftMouseDown, 1);
 
   [field_ mouseDown:event];
   EXPECT_TRUE(page_action_view.MouseWasPressed());
-
-  iconFrame = [cell locationIconFrameForFrame:[field_ bounds]];
-  location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
-  event = Event(field_, location, NSLeftMouseDown, 1);
-
-  [field_ mouseDown:event];
-  EXPECT_TRUE(location_icon_view.mouse_was_pressed_);
 }
 
 // Test that page action menus are properly returned.
 // TODO(shess): Really, this should test that things are forwarded to
 // the cell, and the cell tests should test that the right things are
 // selected.  It's easier to mock the event here, though.  This code's
 // event-mockers might be worth promoting to |test_event_utils.h| or
 // |cocoa_test_helper.h|.
 TEST_F(AutocompleteTextFieldTest, PageActionMenu) {
   AutocompleteTextFieldCell* cell = [field_ autocompleteTextFieldCell];
@@ skipping 172 matching lines @@
   scoped_nsobject<AutocompleteTextField> pin([field_ retain]);
   [field_ removeFromSuperview];
   [test_window() resignKeyWindow];
 
   [[test_window() contentView] addSubview:field_];
   EXPECT_CALL(field_observer_, OnDidResignKey());
   [test_window() resignKeyWindow];
 }
 
 }  // namespace