diff options
Diffstat (limited to 'chrome/browser/notifications/notification_browsertest.cc')
-rw-r--r-- | chrome/browser/notifications/notification_browsertest.cc | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/chrome/browser/notifications/notification_browsertest.cc b/chrome/browser/notifications/notification_browsertest.cc index 21da5dd..3a693da 100644 --- a/chrome/browser/notifications/notification_browsertest.cc +++ b/chrome/browser/notifications/notification_browsertest.cc @@ -5,7 +5,6 @@ #include "chrome/browser/infobars/infobar_tab_helper.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/test/base/in_process_browser_test.h" #include "chrome/test/base/ui_test_utils.h" #include "content/public/browser/web_contents.h" @@ -37,8 +36,8 @@ IN_PROC_BROWSER_TEST_F(NotificationsPermissionTest, TestUserGestureInfobar) { &result)); EXPECT_TRUE(result); - EXPECT_EQ(1U, chrome::GetTabContentsAt(browser(), 0)->infobar_tab_helper()-> - GetInfoBarCount()); + EXPECT_EQ(1U, InfoBarTabHelper::FromWebContents( + chrome::GetWebContentsAt(browser(), 0))->GetInfoBarCount()); } // If this flakes, use http://crbug.com/62311. @@ -52,6 +51,6 @@ IN_PROC_BROWSER_TEST_F(NotificationsPermissionTest, TestNoUserGestureInfobar) { test_server()->GetURL( "files/notifications/notifications_request_inline.html")); - EXPECT_EQ(0U, chrome::GetTabContentsAt(browser(), 0)->infobar_tab_helper()-> - GetInfoBarCount()); + EXPECT_EQ(0U, InfoBarTabHelper::FromWebContents( + chrome::GetWebContentsAt(browser(), 0))->GetInfoBarCount()); } |