| 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.
|
|
|