Index: chrome/browser/extensions/active_script_controller_unittest.cc |
diff --git a/chrome/browser/extensions/active_script_controller_unittest.cc b/chrome/browser/extensions/active_script_controller_unittest.cc |
index a2a95739d12fdd90e521509aae4ca85dcdd0bde6..20acc0de48ec388726b6ae47b1df1e4e0dfe5fbc 100644 |
--- a/chrome/browser/extensions/active_script_controller_unittest.cc |
+++ b/chrome/browser/extensions/active_script_controller_unittest.cc |
@@ -7,6 +7,7 @@ |
#include "base/values.h" |
#include "chrome/browser/extensions/active_script_controller.h" |
#include "chrome/browser/extensions/active_tab_permission_granter.h" |
+#include "chrome/browser/extensions/api/extension_action/extension_action_api.h" |
#include "chrome/browser/extensions/extension_util.h" |
#include "chrome/browser/extensions/permissions_updater.h" |
#include "chrome/browser/extensions/tab_helper.h" |
@@ -182,6 +183,11 @@ TEST_F(ActiveScriptControllerUnitTest, RequestPermissionAndExecute) { |
ASSERT_EQ(0u, GetExecutionCountForExtension(extension->id())); |
ASSERT_FALSE(controller()->WantsToRun(extension)); |
+ ExtensionActionAPI* extension_action_api = |
+ ExtensionActionAPI::Get(profile()); |
+ ASSERT_FALSE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
+ |
// Since the extension requests all_hosts, we should require user consent. |
EXPECT_TRUE(RequiresUserConsent(extension)); |
@@ -189,6 +195,8 @@ TEST_F(ActiveScriptControllerUnitTest, RequestPermissionAndExecute) { |
// executed. |
RequestInjection(extension); |
EXPECT_TRUE(controller()->WantsToRun(extension)); |
+ EXPECT_TRUE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
EXPECT_EQ(0u, GetExecutionCountForExtension(extension->id())); |
// Click to accept the extension executing. |
@@ -197,6 +205,8 @@ TEST_F(ActiveScriptControllerUnitTest, RequestPermissionAndExecute) { |
// The extension should execute, and the extension shouldn't want to run. |
EXPECT_EQ(1u, GetExecutionCountForExtension(extension->id())); |
EXPECT_FALSE(controller()->WantsToRun(extension)); |
+ EXPECT_FALSE(extension_action_api->ExtensionWantsToRun(extension, |
+ web_contents())); |
// Since we already executed on the given page, we shouldn't need permission |
// for a second time. |