123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172 |
- // Copyright 2017 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "base/command_line.h"
- #include "build/build_config.h"
- #include "chrome/browser/browser_process.h"
- #include "chrome/test/base/in_process_browser_test.h"
- #include "chrome/test/base/testing_browser_process.h"
- #include "components/embedder_support/origin_trials/pref_names.h"
- #include "components/embedder_support/switches.h"
- #include "components/prefs/scoped_user_pref_update.h"
- #include "content/public/test/browser_test.h"
- #include "testing/gtest/include/gtest/gtest.h"
- namespace {
- struct DisabledItemsTestData {
- const std::vector<std::string> input_list;
- const std::string expected_switch;
- };
- static const char kNewPublicKey[] = "new public key";
- const DisabledItemsTestData kDisabledFeaturesTests[] = {
- // One feature
- {{"A"}, "A"},
- // Two features
- {{"A", "B"}, "A|B"},
- // Three features
- {{"A", "B", "C"}, "A|B|C"},
- // Spaces in feature name
- {{"A", "B C"}, "A|B C"},
- };
- const DisabledItemsTestData kDisabledTokensTests[] = {
- // One token
- {{"t1"}, "t1"},
- // Two tokens
- {{"t1", "t2"}, "t1|t2"},
- // Three tokens
- {{"t1", "t2", "t3"}, "t1|t2|t3"},
- };
- class ChromeOriginTrialsTest : public InProcessBrowserTest {
- public:
- ChromeOriginTrialsTest(const ChromeOriginTrialsTest&) = delete;
- ChromeOriginTrialsTest& operator=(const ChromeOriginTrialsTest&) = delete;
- protected:
- ChromeOriginTrialsTest() {}
- std::string GetCommandLineSwitch(const base::StringPiece& switch_name) {
- base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
- EXPECT_TRUE(command_line->HasSwitch(switch_name));
- return command_line->GetSwitchValueASCII(switch_name);
- }
- void AddDisabledFeaturesToPrefs(const std::vector<std::string>& features) {
- base::Value disabled_feature_list(base::Value::Type::LIST);
- for (const std::string& feature : features) {
- disabled_feature_list.Append(feature);
- }
- ListPrefUpdate update(
- local_state(), embedder_support::prefs::kOriginTrialDisabledFeatures);
- *update = std::move(disabled_feature_list);
- }
- void AddDisabledTokensToPrefs(const std::vector<std::string>& tokens) {
- base::Value disabled_token_list(base::Value::Type::LIST);
- for (const std::string& token : tokens) {
- disabled_token_list.Append(token);
- }
- ListPrefUpdate update(local_state(),
- embedder_support::prefs::kOriginTrialDisabledTokens);
- *update = std::move(disabled_token_list);
- }
- PrefService* local_state() { return g_browser_process->local_state(); }
- };
- // Tests to verify that the command line is not set, when no prefs exist for
- // the various updates.
- IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, NoPublicKeySet) {
- base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
- EXPECT_FALSE(
- command_line->HasSwitch(embedder_support::kOriginTrialPublicKey));
- }
- IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, NoDisabledFeatures) {
- base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
- EXPECT_FALSE(
- command_line->HasSwitch(embedder_support::kOriginTrialDisabledFeatures));
- }
- IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, NoDisabledTokens) {
- base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
- EXPECT_FALSE(
- command_line->HasSwitch(embedder_support::kOriginTrialDisabledTokens));
- }
- // Tests to verify that the public key is correctly read from prefs and
- // added to the command line
- IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, PRE_PublicKeySetOnCommandLine) {
- local_state()->Set(embedder_support::prefs::kOriginTrialPublicKey,
- base::Value(kNewPublicKey));
- ASSERT_EQ(kNewPublicKey, local_state()->GetString(
- embedder_support::prefs::kOriginTrialPublicKey));
- }
- IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, PublicKeySetOnCommandLine) {
- ASSERT_EQ(kNewPublicKey, local_state()->GetString(
- embedder_support::prefs::kOriginTrialPublicKey));
- std::string actual =
- GetCommandLineSwitch(embedder_support::kOriginTrialPublicKey);
- EXPECT_EQ(kNewPublicKey, actual);
- }
- // Tests to verify that disabled features are correctly read from prefs and
- // added to the command line
- class ChromeOriginTrialsDisabledFeaturesTest
- : public ChromeOriginTrialsTest,
- public ::testing::WithParamInterface<DisabledItemsTestData> {};
- IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledFeaturesTest,
- PRE_DisabledFeaturesSetOnCommandLine) {
- AddDisabledFeaturesToPrefs(GetParam().input_list);
- ASSERT_TRUE(local_state()->HasPrefPath(
- embedder_support::prefs::kOriginTrialDisabledFeatures));
- }
- IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledFeaturesTest,
- DisabledFeaturesSetOnCommandLine) {
- ASSERT_TRUE(local_state()->HasPrefPath(
- embedder_support::prefs::kOriginTrialDisabledFeatures));
- std::string actual =
- GetCommandLineSwitch(embedder_support::kOriginTrialDisabledFeatures);
- EXPECT_EQ(GetParam().expected_switch, actual);
- }
- INSTANTIATE_TEST_SUITE_P(All,
- ChromeOriginTrialsDisabledFeaturesTest,
- ::testing::ValuesIn(kDisabledFeaturesTests));
- // Tests to verify that disabled tokens are correctly read from prefs and
- // added to the command line
- class ChromeOriginTrialsDisabledTokensTest
- : public ChromeOriginTrialsTest,
- public ::testing::WithParamInterface<DisabledItemsTestData> {};
- IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledTokensTest,
- PRE_DisabledTokensSetOnCommandLine) {
- AddDisabledTokensToPrefs(GetParam().input_list);
- ASSERT_TRUE(local_state()->HasPrefPath(
- embedder_support::prefs::kOriginTrialDisabledTokens));
- }
- IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledTokensTest,
- DisabledTokensSetOnCommandLine) {
- ASSERT_TRUE(local_state()->HasPrefPath(
- embedder_support::prefs::kOriginTrialDisabledTokens));
- std::string actual =
- GetCommandLineSwitch(embedder_support::kOriginTrialDisabledTokens);
- EXPECT_EQ(GetParam().expected_switch, actual);
- }
- INSTANTIATE_TEST_SUITE_P(All,
- ChromeOriginTrialsDisabledTokensTest,
- ::testing::ValuesIn(kDisabledTokensTests));
- } // namespace
|