Index: chrome/browser/extensions/page_action_browsertest.cc |
diff --git a/chrome/browser/extensions/page_action_browsertest.cc b/chrome/browser/extensions/page_action_browsertest.cc |
index 2b87b2664add05c99dabd9747ebdb0b4bc6f774d..b8f91f2dc944d808c0b48b31833c9c94cb1e9bfa 100644 |
--- a/chrome/browser/extensions/page_action_browsertest.cc |
+++ b/chrome/browser/extensions/page_action_browsertest.cc |
@@ -5,12 +5,12 @@ |
#include "base/strings/utf_string_conversions.h" |
#include "chrome/browser/extensions/extension_action.h" |
#include "chrome/browser/extensions/extension_action_manager.h" |
+#include "chrome/browser/extensions/extension_action_test_util.h" |
#include "chrome/browser/extensions/extension_browsertest.h" |
#include "chrome/browser/extensions/extension_service.h" |
#include "chrome/browser/extensions/extension_tab_util.h" |
#include "chrome/browser/ui/browser.h" |
#include "chrome/browser/ui/browser_window.h" |
-#include "chrome/browser/ui/location_bar/location_bar.h" |
#include "chrome/browser/ui/tabs/tab_strip_model.h" |
#include "chrome/test/base/ui_test_utils.h" |
#include "extensions/browser/extension_system.h" |
@@ -108,14 +108,14 @@ IN_PROC_BROWSER_TEST_F(ExtensionBrowserTest, UnloadPageAction) { |
// Navigation prompts the location bar to load page actions. |
GURL feed_url = test_server()->GetURL(kFeedPage); |
ui_test_utils::NavigateToURL(browser(), feed_url); |
- LocationBarTesting* location_bar = |
- browser()->window()->GetLocationBar()->GetLocationBarForTesting(); |
- EXPECT_EQ(1, location_bar->PageActionCount()); |
+ content::WebContents* tab = |
+ browser()->tab_strip_model()->GetActiveWebContents(); |
+ EXPECT_EQ(1u, extension_action_test_util::GetTotalPageActionCount(tab)); |
UnloadExtension(last_loaded_extension_id()); |
// Make sure the page action goes away when it's unloaded. |
- EXPECT_EQ(0, location_bar->PageActionCount()); |
+ EXPECT_EQ(0u, extension_action_test_util::GetTotalPageActionCount(tab)); |
} |
// Tests that we can load page actions in the Omnibox. |