[Mac] Fix for Page Info Bubble

chrome/browser/ui/cocoa/page_info/page_info_bubble_controller_unittest.mm

 // Copyright 2014 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/page_info/page_info_bubble_controller.h"
 
 #include <stddef.h>
 
 #include "base/i18n/rtl.h"
 #include "base/macros.h"
 #include "base/strings/utf_string_conversions.h"
-#include "chrome/browser/ui/cocoa/test/cocoa_test_helper.h"
-#include "chrome/test/base/testing_profile.h"
-#include "content/public/test/test_browser_thread_bundle.h"
+#include "chrome/browser/ui/browser_window.h"
+#include "chrome/browser/ui/cocoa/browser_window_controller.h"
+#include "chrome/browser/ui/cocoa/location_bar/location_bar_view_mac.h"
+#import "chrome/browser/ui/cocoa/location_bar/location_icon_decoration.h"
+#import "chrome/browser/ui/cocoa/test/cocoa_profile_test.h"
 #include "content/public/test/test_web_contents_factory.h"
 #include "net/test/test_certificate_data.h"
 #include "testing/gtest_mac.h"
 
 @interface PageInfoBubbleController (ExposedForTesting)
 - (NSView*)permissionsView;
 - (NSButton*)resetDecisionsButton;
 - (NSButton*)connectionHelpButton;
 @end
 
@@ skipping 55 matching lines @@
 const content_settings::SettingSource kTestSettingSources[] = {
     content_settings::SETTING_SOURCE_USER,
     content_settings::SETTING_SOURCE_USER,
     content_settings::SETTING_SOURCE_USER,
     content_settings::SETTING_SOURCE_USER,
     content_settings::SETTING_SOURCE_USER,
     content_settings::SETTING_SOURCE_POLICY,
     content_settings::SETTING_SOURCE_POLICY,
     content_settings::SETTING_SOURCE_EXTENSION};
 
-class PageInfoBubbleControllerTest : public CocoaTest {
+class PageInfoBubbleControllerTest : public CocoaProfileTest {
  public:
   PageInfoBubbleControllerTest() { controller_ = nil; }
 
   void TearDown() override {
     [controller_ close];
-    CocoaTest::TearDown();
+    CocoaProfileTest::TearDown();
   }
 
  protected:
   PageInfoUIBridge* bridge_;  // Weak, owned by controller.
 
   enum MatchType { TEXT_EQUAL = 0, TEXT_NOT_EQUAL };
 
   // Creates a new page info bubble, with the given default width.
   // If |default_width| is 0, the *default* default width will be used.
   void CreateBubbleWithWidth(CGFloat default_width) {
     bridge_ = new PageInfoUIBridge(nullptr);
 
     // The controller cleans up after itself when the window closes.
     controller_ = [PageInfoBubbleControllerForTesting alloc];
     [controller_ setDefaultWindowWidth:default_width];
-    [controller_
-        initWithParentWindow:test_window()
-            pageInfoUIBridge:bridge_
-                 webContents:web_contents_factory_.CreateWebContents(&profile_)
-                         url:GURL("https://www.google.com")];
+    [controller_ initWithParentWindow:browser()->window()->GetNativeWindow()
+                     pageInfoUIBridge:bridge_
+                          webContents:web_contents_factory_.CreateWebContents(
+                                          browser()->profile())
+                                  url:GURL("https://www.google.com")];
     window_ = [controller_ window];
     [controller_ showWindow:nil];
   }
 
   void CreateBubble() { CreateBubbleWithWidth(0.0); }
 
   // Return a pointer to the first NSTextField found that either matches, or
   // doesn't match, the given text.
   NSTextField* FindTextField(MatchType match_type, NSString* text) {
     // The window's only immediate child is an invisible view that has a flipped
@@ skipping 57 matching lines @@
   int NumSettingsNotSetByUser() const {
     int num_non_user_settings = 0;
     for (size_t i = 0; i < arraysize(kTestSettingSources); ++i) {
       num_non_user_settings +=
           (kTestSettingSources[i] != content_settings::SETTING_SOURCE_USER) ? 1
                                                                             : 0;
     }
     return num_non_user_settings;
   }
 
-  content::TestBrowserThreadBundle thread_bundle_;
-  TestingProfile profile_;
   content::TestWebContentsFactory web_contents_factory_;
 
   PageInfoBubbleControllerForTesting* controller_;  // Weak, owns self.
   NSWindow* window_;  // Weak, owned by controller.
 };
 
 TEST_F(PageInfoBubbleControllerTest, ConnectionHelpButton) {
   PageInfoUI::IdentityInfo info;
   info.site_identity = std::string("example.com");
   info.identity_status = PageInfo::SITE_IDENTITY_STATUS_UNKNOWN;
@@ skipping 105 matching lines @@
         EXPECT_GT(NSMinX([icon frame]), 0);
       }
       if ([view isKindOfClass:[NSPopUpButton class]]) {
         NSPopUpButton* button = static_cast<NSPopUpButton*>(view);
         EXPECT_LT(NSMaxX([button frame]), window_width);
       }
     }
   }
 }
 
+// Tests the page icon decoration's active state.
+TEST_F(PageInfoBubbleControllerTest, PageIconDecorationActiveState) {
+  NSWindow* window = browser()->window()->GetNativeWindow();
+  BrowserWindowController* controller =
+      [BrowserWindowController browserWindowControllerForWindow:window];
+  LocationBarDecoration* decoration =
+      [controller locationBarBridge]->GetPageInfoDecoration();
+
+  CreateBubble();
+  EXPECT_TRUE([[controller_ window] isVisible]);
+  EXPECT_TRUE(decoration->active());
+
+  [controller_ close];
+  EXPECT_FALSE(decoration->active());
+}
+
 }  // namespace