Index: chrome/browser/extensions/permission_message_combinations_unittest.cc |
diff --git a/chrome/browser/extensions/permission_message_combinations_unittest.cc b/chrome/browser/extensions/permission_message_combinations_unittest.cc |
index 7581ff1d42618ca91b4167cdc2b85899086ceaf3..66046869b2ce1ff4d2d8d6dfb02d477139049d7b 100644 |
--- a/chrome/browser/extensions/permission_message_combinations_unittest.cc |
+++ b/chrome/browser/extensions/permission_message_combinations_unittest.cc |
@@ -8,6 +8,7 @@ |
#include "chrome/browser/extensions/test_extension_environment.h" |
#include "chrome/common/extensions/permissions/chrome_permission_message_provider.h" |
#include "extensions/common/extension.h" |
+#include "extensions/common/features/simple_feature.h" |
#include "extensions/common/permissions/permission_message_test_util.h" |
#include "extensions/common/permissions/permissions_data.h" |
#include "extensions/common/switches.h" |
@@ -16,12 +17,15 @@ |
namespace extensions { |
+const char kWhitelistedExtensionID[] = "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa"; |
+ |
// Tests that ChromePermissionMessageProvider produces the expected messages for |
// various combinations of app/extension permissions. |
class PermissionMessageCombinationsUnittest : public testing::Test { |
public: |
PermissionMessageCombinationsUnittest() |
- : message_provider_(new ChromePermissionMessageProvider()) {} |
+ : message_provider_(new ChromePermissionMessageProvider()), |
+ whitelisted_extension_id_(kWhitelistedExtensionID) {} |
~PermissionMessageCombinationsUnittest() override {} |
// Overridden from testing::Test: |
@@ -39,10 +43,8 @@ class PermissionMessageCombinationsUnittest : public testing::Test { |
std::replace(json_manifest_with_double_quotes.begin(), |
json_manifest_with_double_quotes.end(), '\'', '"'); |
app_ = env_.MakeExtension( |
- *base::test::ParseJson(json_manifest_with_double_quotes)); |
- // Add the app to any whitelists so we can test all permissions. |
- base::CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
- switches::kWhitelistedExtensionID, app_->id()); |
+ *base::test::ParseJson(json_manifest_with_double_quotes), |
+ kWhitelistedExtensionID); |
} |
// Checks whether the currently installed app or extension produces the given |
@@ -197,6 +199,9 @@ class PermissionMessageCombinationsUnittest : public testing::Test { |
extensions::TestExtensionEnvironment env_; |
scoped_ptr<ChromePermissionMessageProvider> message_provider_; |
scoped_refptr<const Extension> app_; |
+ // Whitelist a known extension id so we can test all permissions. This ID |
+ // will be used for each test app. |
+ extensions::SimpleFeature::ScopedWhitelistForTest whitelisted_extension_id_; |
DISALLOW_COPY_AND_ASSIGN(PermissionMessageCombinationsUnittest); |
}; |