Remove TabStripModel wrapper use.

chrome/browser/managed_mode/managed_mode_browsertest.cc

 // Copyright (c) 2012 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.
 
 #include "base/command_line.h"
 #include "base/utf_string_conversions.h"
 #include "base/values.h"
 #include "chrome/browser/api/infobars/confirm_infobar_delegate.h"
 #include "chrome/browser/infobars/infobar.h"
 #include "chrome/browser/infobars/infobar_tab_helper.h"
 #include "chrome/browser/managed_mode/managed_mode.h"
 #include "chrome/browser/managed_mode/managed_user_service.h"
 #include "chrome/browser/managed_mode/managed_user_service_factory.h"
 #include "chrome/browser/prefs/pref_service_syncable.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_navigator.h"
-#include "chrome/browser/ui/browser_tabstrip.h"
+#include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/interstitial_page.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/web_contents.h"
@@ skipping 58 matching lines @@
   void CheckShownPage(WebContents* tab, content::PageType page_type) {
     ASSERT_FALSE(tab->IsCrashed());
     NavigationEntry* entry = tab->GetController().GetActiveEntry();
     ASSERT_TRUE(entry);
     ASSERT_EQ(page_type, entry->GetPageType());
   }
 
   // Checks if the current number of shown infobars is equal to |expected|.
   void CheckNumberOfInfobars(unsigned int expected) {
     EXPECT_EQ(expected,
-        InfoBarService::FromWebContents(
-            chrome::GetActiveWebContents(browser()))->GetInfoBarCount());
+              InfoBarService::FromWebContents(
+                  browser()->tab_strip_model()->GetActiveWebContents())->
+                      GetInfoBarCount());
   }
 
   // Acts on the interstitial and infobar according to the values set to
   // |interstitial_action| and |infobar_action|.
   void ActOnInterstitialAndInfobar(WebContents* tab,
                                    InterstitialAction interstitial_action,
                                    InfobarAction infobar_action) {
     InterstitialPage* interstitial_page = tab->GetInterstitialPage();
     ASSERT_TRUE(interstitial_page);
 
@@ skipping 71 matching lines @@
   ManagedUserService* managed_user_service_;
 };
 
 // Navigates to a URL which is not in a manual list, clicks preview on the
 // interstitial and then allows the website. The website should now be in the
 // manual whitelist.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest, SimpleURLNotInAnyLists) {
   GURL test_url("http://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // Same as above just that the URL redirects to a second URL first. The initial
 // exact URL should be in the whitelist as well as the second hostname.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest, RedirectedURLsNotInAnyLists) {
   std::vector<std::string> url_list;
   url_list.push_back("www.a.com");
   std::string last_url("www.example.com/files/simple.html");
   GURL test_url(GetRedirectURL(url_list, last_url));
 
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(
       true, "http://.www.a.com/server-redirect"));
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // Navigates to a URL in the whitelist. No interstitial should be shown and
 // the browser should navigate to the page.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest, SimpleURLInWhitelist) {
   GURL test_url("http://www.example.com/files/simple.html");
   ListValue whitelist;
   whitelist.AppendString(test_url.host());
   managed_user_service_->AddToManualList(true, whitelist);
 
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsNotInterstitial(tab);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // Navigates to a URL which redirects to another URL, both in the whitelist.
 // No interstitial should be shown and the browser should navigate to the page.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        RedirectedURLsBothInWhitelist) {
   std::vector<std::string> url_list;
   url_list.push_back("www.a.com");
   std::string last_url("www.example.com/files/simple.html");
   GURL test_url(GetRedirectURL(url_list, last_url));
 
   // Add both hostnames to the whitelist, should navigate without interstitial.
   ListValue whitelist;
   whitelist.AppendString("www.a.com");
   whitelist.AppendString("www.example.com");
   managed_user_service_->AddToManualList(true, whitelist);
 
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsNotInterstitial(tab);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.a.com"));
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // Only one URL is in the whitelist and the second not, so it should redirect,
 // show an interstitial, then after clicking preview and clicking accept both
 // websites should be in the whitelist.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        RedirectedURLFirstInWhitelist) {
   std::vector<std::string> url_list;
   url_list.push_back("www.a.com");
   std::string last_url("www.example.com/files/simple.html");
   GURL test_url(GetRedirectURL(url_list, last_url));
 
   // Add the first URL to the whitelist.
   ListValue whitelist;
   whitelist.AppendString("www.a.com");
   managed_user_service_->AddToManualList(true, whitelist);
 
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   EXPECT_EQ(tab->GetURL().spec(), "http://" + last_url);
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.a.com"));
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // This test navigates to a URL which is not in the whitelist but redirects to
 // one that is. The expected behavior is that the user will get an interstitial
 // and after clicking preview the user will see an infobar stating that the URL
 // was already in the whitelist (and the first URL gets added as well).
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        RedirectedURLLastInWhitelist) {
   std::vector<std::string> url_list;
   url_list.push_back("www.a.com");
   std::string last_url("www.example.com/files/simple.html");
   GURL test_url(GetRedirectURL(url_list, last_url));
 
   // Add the last URL to the whitelist.
   ListValue whitelist;
   whitelist.AppendString("www.example.com");
   managed_user_service_->AddToManualList(true, whitelist);
 
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   EXPECT_EQ(tab->GetURL().host(), "www.a.com");
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED,
                                    INFOBAR_ALREADY_ADDED);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(
       true, "http://.www.a.com/server-redirect"));
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // Tests whether going back after being shown an interstitial works. No
 // websites should be added to the whitelist.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        SimpleURLNotInAnyListsGoBack) {
   GURL test_url("http://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_DONTPROCEED,
                                    INFOBAR_NOT_USED);
 
   EXPECT_EQ(tab->GetURL().spec(), "about:blank");
 
   EXPECT_FALSE(managed_user_service_->IsInManualList(true, "www.example.com"));
 }
 
 // Like SimpleURLNotInAnyLists just that it navigates to a page on the allowed
 // domain after clicking allow on the infobar. The navigation should complete
 // and no interstitial should be shown the second time.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        SimpleURLNotInAnyListsNavigateAgain) {
   GURL test_url("http://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 
   // Navigate to a different page on the same host.
   test_url = GURL("http://www.example.com/files/english_page.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
   // An interstitial should not show up.
   CheckShownPageIsNotInterstitial(tab);
   CheckNumberOfInfobars(0);
 }
 
 // Same as above just that it reloads the page instead of navigating to another
 // page on the website. Same expected behavior.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        SimpleURLNotInAnyListsReloadPageAfterAdd) {
   GURL test_url("http://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "www.example.com"));
 
   // Reload the page
   tab->GetController().Reload(false);
 
   // Expect that the page shows up and not an interstitial.
   CheckShownPageIsNotInterstitial(tab);
   CheckNumberOfInfobars(0);
   EXPECT_EQ(tab->GetURL().spec(), test_url.spec());
 }
 
 // Navigates to an HTTPS page not in any lists, similar to the simple test but
 // over HTTPS (expected behavior is that only the HTTPS version of the site
 // gets whitelisted, compared to the simple case where all protocols get
 // whitelisted).
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        SimpleHTTPSURLNotInAnyLists) {
   // Navigate to an HTTPS URL.
   GURL test_url("https://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   // Check that the https:// version is added in the whitelist.
   EXPECT_TRUE(managed_user_service_->IsInManualList(true, "https://www.example.com"));
 }
 
 // The test navigates to a page, the interstitial is shown and preview is
 // clicked but then the test navigates to other pages on the same domain before
 // clicking allow on the page. The "allow" infobar should still be there during
 // this time and the navigation should be allowed. When the test finally clicks
 // accept the webpage should be whitelisted.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        SimpleURLNotInAnyListNavigateAround) {
   GURL test_url("http://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
 
   InterstitialPage* interstitial_page = tab->GetInterstitialPage();
   ASSERT_TRUE(interstitial_page);
 
   content::WindowedNotificationObserver infobar_added(
       chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_ADDED,
       content::NotificationService::AllSources());
 
@@ skipping 44 matching lines @@
 // The test navigates to a page, the interstitial is shown and preview is
 // clicked but then the test navigates to a page on a different host, which
 // should trigger an interstitial again. Clicking preview on this interstitial
 // and accepting it should add the second website to the whitelist and not the
 // first one.
 IN_PROC_BROWSER_TEST_F(ManagedModeBlockModeTest,
                        NavigateDifferentHostAfterPreview) {
   GURL test_url("http://www.example.com/files/simple.html");
   ui_test_utils::NavigateToURL(browser(), test_url);
 
-  WebContents* tab = chrome::GetActiveWebContents(browser());
+  WebContents* tab = browser()->tab_strip_model()->GetActiveWebContents();
 
   CheckShownPageIsInterstitial(tab);
 
   InterstitialPage* interstitial_page = tab->GetInterstitialPage();
   ASSERT_TRUE(interstitial_page);
 
   content::WindowedNotificationObserver infobar_added(
       chrome::NOTIFICATION_TAB_CONTENTS_INFOBAR_ADDED,
       content::NotificationService::AllSources());
 
@@ skipping 15 matching lines @@
 
   CheckShownPageIsInterstitial(tab);
   CheckNumberOfInfobars(0);
 
   ActOnInterstitialAndInfobar(tab, INTERSTITIAL_PROCEED, INFOBAR_ACCEPT);
 
   EXPECT_FALSE(managed_user_service_->IsInManualList(true, "www.example.com"));
   EXPECT_TRUE(managed_user_service_->IsInManualList(true,
                                                     "www.new-example.com"));
 }