Mac: Content blocked icons.

chrome/browser/cocoa/autocomplete_text_field_cell_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 "base/scoped_nsobject.h"
 #import "chrome/browser/cocoa/autocomplete_text_field_cell.h"
 #import "chrome/browser/cocoa/cocoa_test_helper.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 266 matching lines @@
 
 // Test that the security icon is at the right side of the cell.
 TEST_F(AutocompleteTextFieldCellTest, SecurityImageFrame) {
   AutocompleteTextFieldCell* cell =
       static_cast<AutocompleteTextFieldCell*>([view_ cell]);
   const NSRect bounds([view_ bounds]);
   security_image_view_.SetImageShown(
       LocationBarViewMac::SecurityImageView::LOCK);
 
   security_image_view_.SetVisible(false);
-  NSRect iconRect = [cell securityImageFrameForFrame:bounds];
-  EXPECT_TRUE(NSIsEmptyRect(iconRect));
-
-  // Save the starting frame for after clear.
-  const NSRect originalIconRect(iconRect);
+  EXPECT_EQ(0u, [[cell layedOutIcons:bounds] count]);
 
   security_image_view_.SetVisible(true);
-  iconRect = [cell securityImageFrameForFrame:bounds];
+  NSArray* icons = [cell layedOutIcons:bounds];
+  ASSERT_EQ(1u, [icons count]);
+  NSRect iconRect = [[icons objectAtIndex:0] rect];
 
   EXPECT_FALSE(NSIsEmptyRect(iconRect));
   EXPECT_TRUE(NSContainsRect(bounds, iconRect));
 
   // Make sure we are right of the |drawingRect|.
   NSRect drawingRect = [cell drawingRectForBounds:bounds];
   EXPECT_LE(NSMaxX(drawingRect), NSMinX(iconRect));
 
   // Make sure we're right of the |textFrame|.
   NSRect textFrame = [cell textFrameForFrame:bounds];
   EXPECT_LE(NSMaxX(textFrame), NSMinX(iconRect));
 
   // Now add a label.
   NSFont* font = [NSFont controlContentFontOfSize:12.0];
   NSColor* color = [NSColor blackColor];
   security_image_view_.SetLabel(@"Label", font, color);
-  iconRect = [cell securityImageFrameForFrame:bounds];
+  icons = [cell layedOutIcons:bounds];
+  ASSERT_EQ(1u, [icons count]);
+  iconRect = [[icons objectAtIndex:0] rect];
 
   EXPECT_FALSE(NSIsEmptyRect(iconRect));
   EXPECT_TRUE(NSContainsRect(bounds, iconRect));
 
   // Make sure we are right of the |drawingRect|.
   drawingRect = [cell drawingRectForBounds:bounds];
   EXPECT_LE(NSMaxX(drawingRect), NSMinX(iconRect));
 
   // Make sure we're right of the |textFrame|.
   textFrame = [cell textFrameForFrame:bounds];
   EXPECT_LE(NSMaxX(textFrame), NSMinX(iconRect));
 
   // Make sure we clear correctly.
   security_image_view_.SetVisible(false);
-  iconRect = [cell securityImageFrameForFrame:bounds];
-  EXPECT_TRUE(NSEqualRects(iconRect, originalIconRect));
-  EXPECT_TRUE(NSIsEmptyRect(iconRect));
+  EXPECT_EQ(0u, [[cell layedOutIcons:bounds] count]);
 }
 
 // Test Page Action counts.
 TEST_F(AutocompleteTextFieldCellTest, PageActionCount) {
   AutocompleteTextFieldCell* cell =
       static_cast<AutocompleteTextFieldCell*>([view_ cell]);
 
   TestPageActionImageView page_action_view;
   TestPageActionImageView page_action_view2;
 
@@ skipping 36 matching lines @@
   page_action_view2.SetVisible(false);
   EXPECT_TRUE(NSIsEmptyRect([cell pageActionFrameForIndex:0 inFrame:bounds]));
   EXPECT_TRUE(NSIsEmptyRect([cell pageActionFrameForIndex:1 inFrame:bounds]));
 
   // Test preview page actions, as used by the extension installed bubble.
   TestPageActionImageView preview_view;
   list.Add(&preview_view);
   EXPECT_TRUE(NSIsEmptyRect([cell pageActionFrameForIndex:2 inFrame:bounds]));
   preview_view.set_preview_enabled(true);
   EXPECT_FALSE(NSIsEmptyRect([cell pageActionFrameForIndex:2 inFrame:bounds]));
+  preview_view.set_preview_enabled(false);
 
   // One page action, no security icon.
   page_action_view.SetVisible(true);
   NSRect iconRect0 = [cell pageActionFrameForIndex:0 inFrame:bounds];
 
   EXPECT_FALSE(NSIsEmptyRect(iconRect0));
   EXPECT_TRUE(NSContainsRect(bounds, iconRect0));
 
   // Make sure we are right of the |drawingRect|.
   NSRect drawingRect = [cell drawingRectForBounds:bounds];
   EXPECT_LE(NSMaxX(drawingRect), NSMinX(iconRect0));
 
   // Make sure we're right of the |textFrame|.
   NSRect textFrame = [cell textFrameForFrame:bounds];
   EXPECT_LE(NSMaxX(textFrame), NSMinX(iconRect0));
 
   // Two page actions plus a security icon.
   page_action_view2.SetVisible(true);
   security_image_view_.SetVisible(true);
+  NSArray* icons = [cell layedOutIcons:bounds];
+  EXPECT_EQ(3u, [icons count]);
   iconRect0 = [cell pageActionFrameForIndex:0 inFrame:bounds];
   NSRect iconRect1 = [cell pageActionFrameForIndex:1 inFrame:bounds];
-  NSRect lockRect = [cell securityImageFrameForFrame:bounds];
+  NSRect lockRect = [[icons objectAtIndex:0] rect];
+
+  EXPECT_TRUE(NSEqualRects(iconRect0, [[icons objectAtIndex:1] rect]));
+  EXPECT_TRUE(NSEqualRects(iconRect1, [[icons objectAtIndex:2] rect]));
 
   // Make sure they're all in the expected order, and right of the |drawingRect|
   // and |textFrame|.
   drawingRect = [cell drawingRectForBounds:bounds];
   textFrame = [cell textFrameForFrame:bounds];
 
   EXPECT_FALSE(NSIsEmptyRect(iconRect0));
   EXPECT_TRUE(NSContainsRect(bounds, iconRect0));
   EXPECT_FALSE(NSIsEmptyRect(iconRect1));
   EXPECT_TRUE(NSContainsRect(bounds, iconRect1));
@@ skipping 73 matching lines @@
   EXPECT_TRUE([[[cell hintString] string] hasPrefix:kHintPrefix]);
   EXPECT_TRUE([[[cell hintString] string] hasSuffix:kHintSuffix]);
 
   // Narrow width suppresses everything but the image.
   [cell setKeywordHintPrefix:kHintPrefix image:image suffix:kHintSuffix
               availableWidth:kNarrowWidth];
   EXPECT_EQ([[cell hintString] length], 1U);
 }
 
 }  // namespace