Mac: Content blocked icons.

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>
 
 #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 "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 MockSecurityImageView : public LocationBarViewMac::SecurityImageView {
  public:
   MockSecurityImageView(LocationBarViewMac* owner,
                         Profile* profile,
                         ToolbarModel* model)
       : LocationBarViewMac::SecurityImageView(owner, profile, model) {}
 
-  MOCK_METHOD0(OnMousePressed, bool());
+  // 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:
   MockPageActionImageView() {}
   virtual ~MockPageActionImageView() {}
 
   // We can't use gmock's MOCK_METHOD macro, because it doesn't like the
   // NSRect argument to OnMousePressed.
-  virtual bool OnMousePressed(NSRect bounds) {
+  virtual void OnMousePressed(NSRect bounds) {
     mouse_was_pressed_ = true;
-    return true;
   }
 
   bool MouseWasPressed() { return mouse_was_pressed_; }
 
  private:
   bool mouse_was_pressed_;
 };
 
 // TODO(shess): Consider lifting this to
 // autocomplete_text_field_unittest_helper.h to share this with the
@@ skipping 525 matching lines @@
   MockSecurityImageView security_image_view(NULL, NULL, NULL);
   [cell setSecurityImageView:&security_image_view];
   security_image_view.SetImageShown(
       LocationBarViewMac::SecurityImageView::LOCK);
   security_image_view.SetVisible(true);
 
   NSRect iconFrame([cell securityImageFrameForFrame:[field_ bounds]]);
   NSPoint location(NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame)));
   NSEvent* event(Event(field_, location, NSLeftMouseDown, 1));
 
-  EXPECT_CALL(security_image_view, OnMousePressed());
   [field_ mouseDown:event];
+  EXPECT_TRUE(security_image_view.mouse_was_pressed_);
 }
 
 // Clicking a Page Action icon should call its OnMousePressed.
 TEST_F(AutocompleteTextFieldObserverTest, PageActionMouseDown) {
   AutocompleteTextFieldCell* cell = [field_ autocompleteTextFieldCell];
 
   MockSecurityImageView security_image_view(NULL, NULL, NULL);
   security_image_view.SetImageShown(
       LocationBarViewMac::SecurityImageView::LOCK);
   [cell setSecurityImageView:&security_image_view];
@@ skipping 50 matching lines @@
   location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
   event = Event(field_, location, NSLeftMouseDown, 1);
 
   [field_ mouseDown:event];
   EXPECT_TRUE(page_action_view.MouseWasPressed());
 
   iconFrame = [cell securityImageFrameForFrame:[field_ bounds]];
   location = NSMakePoint(NSMidX(iconFrame), NSMidY(iconFrame));
   event = Event(field_, location, NSLeftMouseDown, 1);
 
-  EXPECT_CALL(security_image_view, OnMousePressed());
   [field_ mouseDown:event];
+  EXPECT_TRUE(security_image_view.mouse_was_pressed_);
 }
 
 // Verify that -setAttributedStringValue: works as expected when
 // focussed or when not focussed.  Our code mostly depends on about
 // whether -stringValue works right.
 TEST_F(AutocompleteTextFieldTest, SetAttributedStringBaseline) {
   EXPECT_EQ(nil, [field_ currentEditor]);
 
   // So that we can set rich text.
   [field_ setAllowsEditingTextAttributes:YES];
@@ skipping 123 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