Index: chrome/browser/ui/search/search_delegate_unittest.cc |
diff --git a/chrome/browser/ui/search/search_delegate_unittest.cc b/chrome/browser/ui/search/search_delegate_unittest.cc |
index 07d6a1c3b89333457a486f21fdc610810f7f29c7..f5b18b07e27d42852f2ba8ec35b09dd83a5e0a05 100644 |
--- a/chrome/browser/ui/search/search_delegate_unittest.cc |
+++ b/chrome/browser/ui/search/search_delegate_unittest.cc |
@@ -4,10 +4,10 @@ |
#include "base/command_line.h" |
#include "chrome/browser/ui/browser.h" |
-#include "chrome/browser/ui/browser_tabstrip.h" |
#include "chrome/browser/ui/search/search.h" |
#include "chrome/browser/ui/search/search_model.h" |
#include "chrome/browser/ui/search/search_tab_helper.h" |
+#include "chrome/browser/ui/tabs/tab_strip_model.h" |
#include "chrome/common/chrome_switches.h" |
#include "chrome/test/base/browser_with_test_window_test.h" |
@@ -31,7 +31,8 @@ TEST_F(SearchDelegateTest, SearchModel) { |
// Propagate change from tab's search model to browser's search model. |
AddTab(browser(), GURL("http://foo/0")); |
- content::WebContents* web_contents = chrome::GetWebContentsAt(browser(), 0); |
+ content::WebContents* web_contents = |
+ browser()->tab_strip_model()->GetWebContentsAt(0); |
chrome::search::SearchTabHelper::FromWebContents(web_contents)->model()-> |
SetMode(Mode(Mode::MODE_NTP, Mode::ORIGIN_NTP, false)); |
EXPECT_TRUE(browser()->search_model()->mode().is_ntp()); |
@@ -39,14 +40,14 @@ TEST_F(SearchDelegateTest, SearchModel) { |
// Add second tab, make it active, and make sure its mode changes |
// propagate to the browser's search model. |
AddTab(browser(), GURL("http://foo/1")); |
- chrome::ActivateTabAt(browser(), 1, true); |
- web_contents = chrome::GetWebContentsAt(browser(), 1); |
+ browser()->tab_strip_model()->ActivateTabAt(1, true); |
+ web_contents = browser()->tab_strip_model()->GetWebContentsAt(1); |
chrome::search::SearchTabHelper::FromWebContents(web_contents)->model()-> |
SetMode(Mode(Mode::MODE_SEARCH_RESULTS, Mode::ORIGIN_DEFAULT, false)); |
EXPECT_TRUE(browser()->search_model()->mode().is_search()); |
// The first tab is not active so changes should not propagate. |
- web_contents = chrome::GetWebContentsAt(browser(), 0); |
+ web_contents = browser()->tab_strip_model()->GetWebContentsAt(0); |
chrome::search::SearchTabHelper::FromWebContents(web_contents)->model()-> |
SetMode(Mode(Mode::MODE_NTP, Mode::ORIGIN_NTP, false)); |
EXPECT_TRUE(browser()->search_model()->mode().is_search()); |