Index: chrome/browser/extensions/location_bar_controller_unittest.cc |
diff --git a/chrome/browser/extensions/location_bar_controller_unittest.cc b/chrome/browser/extensions/location_bar_controller_unittest.cc |
index 2cbc4da515fb82221f7479930dac508438894fa6..48c25866ba48987811301633d0e4f35589d8587b 100644 |
--- a/chrome/browser/extensions/location_bar_controller_unittest.cc |
+++ b/chrome/browser/extensions/location_bar_controller_unittest.cc |
@@ -8,6 +8,7 @@ |
#include "base/memory/scoped_ptr.h" |
#include "base/message_loop/message_loop.h" |
#include "chrome/browser/extensions/active_script_controller.h" |
+#include "chrome/browser/extensions/api/extension_action/extension_action_api.h" |
#include "chrome/browser/extensions/extension_action.h" |
#include "chrome/browser/extensions/extension_action_manager.h" |
#include "chrome/browser/extensions/extension_service.h" |
@@ -159,6 +160,17 @@ TEST_F(LocationBarControllerUnitTest, NavigationClearsState) { |
page_action.SetTitle(tab_id(), "Goodbye"); |
page_action.SetPopupUrl(tab_id(), extension->GetResourceURL("popup.html")); |
+ ExtensionActionAPI* extension_action_api = |
+ ExtensionActionAPI::Get(profile()); |
+ // By default, extensions shouldn't want to act on a page. |
+ EXPECT_FALSE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
+ // Showing the page action should indicate that an extension *does* want to |
+ // run on the page. |
+ page_action.SetIsVisible(tab_id(), true); |
+ EXPECT_TRUE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
+ |
EXPECT_EQ("Goodbye", page_action.GetTitle(tab_id())); |
EXPECT_EQ(extension->GetResourceURL("popup.html"), |
page_action.GetPopupUrl(tab_id())); |
@@ -169,12 +181,16 @@ TEST_F(LocationBarControllerUnitTest, NavigationClearsState) { |
EXPECT_EQ("Goodbye", page_action.GetTitle(tab_id())); |
EXPECT_EQ(extension->GetResourceURL("popup.html"), |
page_action.GetPopupUrl(tab_id())); |
+ EXPECT_TRUE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
// Should discard the settings, and go back to the defaults. |
NavigateAndCommit(GURL("http://www.yahoo.com")); |
EXPECT_EQ("Hello", page_action.GetTitle(tab_id())); |
EXPECT_EQ(GURL(), page_action.GetPopupUrl(tab_id())); |
+ EXPECT_FALSE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
} |
} // namespace |