Index: chrome/browser/safe_browsing/safe_browsing_blocking_page_test.cc |
=================================================================== |
--- chrome/browser/safe_browsing/safe_browsing_blocking_page_test.cc (revision 145001) |
+++ chrome/browser/safe_browsing/safe_browsing_blocking_page_test.cc (working copy) |
@@ -15,6 +15,7 @@ |
#include "chrome/browser/safe_browsing/safe_browsing_blocking_page.h" |
#include "chrome/browser/safe_browsing/safe_browsing_service.h" |
#include "chrome/browser/ui/browser.h" |
+#include "chrome/browser/ui/browser_tabstrip.h" |
#include "chrome/browser/ui/tab_contents/tab_contents.h" |
#include "chrome/common/pref_names.h" |
#include "chrome/common/url_constants.h" |
@@ -261,7 +262,7 @@ |
} |
void SendCommand(const std::string& command) { |
- WebContents* contents = browser()->GetActiveWebContents(); |
+ WebContents* contents = chrome::GetActiveWebContents(browser()); |
// We use InterstitialPage::GetInterstitialPage(tab) instead of |
// tab->GetInterstitialPage() because the tab doesn't have a pointer |
// to its interstital page until it gets a command from the renderer |
@@ -276,7 +277,7 @@ |
} |
void DontProceedThroughInterstitial() { |
- WebContents* contents = browser()->GetActiveWebContents(); |
+ WebContents* contents = chrome::GetActiveWebContents(browser()); |
InterstitialPage* interstitial_page = InterstitialPage::GetInterstitialPage( |
contents); |
ASSERT_TRUE(interstitial_page); |
@@ -284,7 +285,7 @@ |
} |
void ProceedThroughInterstitial() { |
- WebContents* contents = browser()->GetActiveWebContents(); |
+ WebContents* contents = chrome::GetActiveWebContents(browser()); |
InterstitialPage* interstitial_page = InterstitialPage::GetInterstitialPage( |
contents); |
ASSERT_TRUE(interstitial_page); |
@@ -292,7 +293,7 @@ |
} |
void AssertNoInterstitial(bool wait_for_delete) { |
- WebContents* contents = browser()->GetActiveWebContents(); |
+ WebContents* contents = chrome::GetActiveWebContents(browser()); |
if (contents->ShowingInterstitialPage() && wait_for_delete) { |
// We'll get notified when the interstitial is deleted. |
@@ -308,14 +309,14 @@ |
} |
bool YesInterstitial() { |
- WebContents* contents = browser()->GetActiveWebContents(); |
+ WebContents* contents = chrome::GetActiveWebContents(browser()); |
InterstitialPage* interstitial_page = InterstitialPage::GetInterstitialPage( |
contents); |
return interstitial_page != NULL; |
} |
void WaitForInterstitial() { |
- WebContents* contents = browser()->GetActiveWebContents(); |
+ WebContents* contents = chrome::GetActiveWebContents(browser()); |
ui_test_utils::WindowedNotificationObserver interstitial_observer( |
content::NOTIFICATION_INTERSTITIAL_ATTACHED, |
content::Source<WebContents>(contents)); |
@@ -357,13 +358,13 @@ |
ui_test_utils::BROWSER_TEST_WAIT_FOR_TAB); |
WaitForInterstitial(); |
// Cancel the redirect request while interstitial page is open. |
- browser()->ActivateTabAt(0, true); |
+ chrome::ActivateTabAt(browser(), 0, true); |
ui_test_utils::NavigateToURLWithDisposition( |
browser(), |
GURL("javascript:stopWin()"), |
CURRENT_TAB, |
ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION); |
- browser()->ActivateTabAt(1, true); |
+ chrome::ActivateTabAt(browser(), 1, true); |
// Simulate the user clicking "proceed", there should be no crash. |
SendCommand("\"proceed\""); |
} |
@@ -409,7 +410,7 @@ |
AssertNoInterstitial(false); // Assert the interstitial is gone |
EXPECT_EQ( |
GURL(chrome::kAboutBlankURL), // Back to "about:blank" |
- browser()->GetActiveWebContents()->GetURL()); |
+ chrome::GetActiveWebContents(browser())->GetURL()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, MalwareProceed) { |
@@ -422,11 +423,11 @@ |
ui_test_utils::WindowedNotificationObserver observer( |
content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
content::Source<NavigationController>( |
- &browser()->GetActiveWebContents()->GetController())); |
+ &chrome::GetActiveWebContents(browser())->GetController())); |
SendCommand("\"proceed\""); // Simulate the user clicking "proceed" |
observer.Wait(); |
AssertNoInterstitial(true); // Assert the interstitial is gone. |
- EXPECT_EQ(url, browser()->GetActiveWebContents()->GetURL()); |
+ EXPECT_EQ(url, chrome::GetActiveWebContents(browser())->GetURL()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, PhishingDontProceed) { |
@@ -439,7 +440,7 @@ |
AssertNoInterstitial(false); // Assert the interstitial is gone |
EXPECT_EQ( |
GURL(chrome::kAboutBlankURL), // We are back to "about:blank". |
- browser()->GetActiveWebContents()->GetURL()); |
+ chrome::GetActiveWebContents(browser())->GetURL()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, PhishingProceed) { |
@@ -453,11 +454,11 @@ |
ui_test_utils::WindowedNotificationObserver observer( |
content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
content::Source<NavigationController>( |
- &browser()->GetActiveWebContents()->GetController())); |
+ &chrome::GetActiveWebContents(browser())->GetController())); |
SendCommand("\"proceed\""); // Simulate the user clicking "proceed". |
observer.Wait(); |
AssertNoInterstitial(true); // Assert the interstitial is gone |
- EXPECT_EQ(url, browser()->GetActiveWebContents()->GetURL()); |
+ EXPECT_EQ(url, chrome::GetActiveWebContents(browser())->GetURL()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, PhishingReportError) { |
@@ -471,7 +472,7 @@ |
ui_test_utils::WindowedNotificationObserver observer( |
content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
content::Source<NavigationController>( |
- &browser()->GetActiveWebContents()->GetController())); |
+ &chrome::GetActiveWebContents(browser())->GetController())); |
SendCommand("\"reportError\""); // Simulate the user clicking "report error" |
observer.Wait(); |
AssertNoInterstitial(false); // Assert the interstitial is gone |
@@ -479,7 +480,7 @@ |
// We are in the error reporting page. |
EXPECT_EQ( |
"/safebrowsing/report_error/", |
- browser()->GetActiveWebContents()->GetURL().path()); |
+ chrome::GetActiveWebContents(browser())->GetURL().path()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, |
@@ -494,7 +495,7 @@ |
ui_test_utils::WindowedNotificationObserver observer( |
content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
content::Source<NavigationController>( |
- &browser()->GetActiveWebContents()->GetController())); |
+ &chrome::GetActiveWebContents(browser())->GetController())); |
SendCommand("\"learnMore\""); // Simulate the user clicking "learn more" |
observer.Wait(); |
AssertNoInterstitial(false); // Assert the interstitial is gone |
@@ -502,7 +503,7 @@ |
// We are in the help page. |
EXPECT_EQ( |
"/support/bin/answer.py", |
- browser()->GetActiveWebContents()->GetURL().path()); |
+ chrome::GetActiveWebContents(browser())->GetURL().path()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, MalwareIframeDontProceed) { |
@@ -515,14 +516,14 @@ |
ui_test_utils::WindowedNotificationObserver observer( |
content::NOTIFICATION_NAV_ENTRY_COMMITTED, |
content::Source<NavigationController>( |
- &browser()->GetActiveWebContents()->GetController())); |
+ &chrome::GetActiveWebContents(browser())->GetController())); |
SendCommand("\"takeMeBack\""); // Simulate the user clicking "back" |
observer.Wait(); |
AssertNoInterstitial(false); // Assert the interstitial is gone |
EXPECT_EQ( |
GURL(chrome::kAboutBlankURL), // Back to "about:blank" |
- browser()->GetActiveWebContents()->GetURL()); |
+ chrome::GetActiveWebContents(browser())->GetURL()); |
} |
// Crashy, http://crbug.com/68834. |
@@ -537,7 +538,7 @@ |
SendCommand("\"proceed\""); // Simulate the user clicking "proceed" |
AssertNoInterstitial(true); // Assert the interstitial is gone |
- EXPECT_EQ(url, browser()->GetActiveWebContents()->GetURL()); |
+ EXPECT_EQ(url, chrome::GetActiveWebContents(browser())->GetURL()); |
} |
IN_PROC_BROWSER_TEST_F(SafeBrowsingBlockingPageTest, |
@@ -558,6 +559,6 @@ |
SendCommand("\"proceed\""); // Simulate the user clicking "back" |
AssertNoInterstitial(true); // Assert the interstitial is gone |
- EXPECT_EQ(url, browser()->GetActiveWebContents()->GetURL()); |
+ EXPECT_EQ(url, chrome::GetActiveWebContents(browser())->GetURL()); |
AssertReportSent(); |
} |