chrome_origin_trials_browsertest.cc 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172
  1. // Copyright 2017 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "base/command_line.h"
  5. #include "build/build_config.h"
  6. #include "chrome/browser/browser_process.h"
  7. #include "chrome/test/base/in_process_browser_test.h"
  8. #include "chrome/test/base/testing_browser_process.h"
  9. #include "components/embedder_support/origin_trials/pref_names.h"
  10. #include "components/embedder_support/switches.h"
  11. #include "components/prefs/scoped_user_pref_update.h"
  12. #include "content/public/test/browser_test.h"
  13. #include "testing/gtest/include/gtest/gtest.h"
  14. namespace {
  15. struct DisabledItemsTestData {
  16. const std::vector<std::string> input_list;
  17. const std::string expected_switch;
  18. };
  19. static const char kNewPublicKey[] = "new public key";
  20. const DisabledItemsTestData kDisabledFeaturesTests[] = {
  21. // One feature
  22. {{"A"}, "A"},
  23. // Two features
  24. {{"A", "B"}, "A|B"},
  25. // Three features
  26. {{"A", "B", "C"}, "A|B|C"},
  27. // Spaces in feature name
  28. {{"A", "B C"}, "A|B C"},
  29. };
  30. const DisabledItemsTestData kDisabledTokensTests[] = {
  31. // One token
  32. {{"t1"}, "t1"},
  33. // Two tokens
  34. {{"t1", "t2"}, "t1|t2"},
  35. // Three tokens
  36. {{"t1", "t2", "t3"}, "t1|t2|t3"},
  37. };
  38. class ChromeOriginTrialsTest : public InProcessBrowserTest {
  39. public:
  40. ChromeOriginTrialsTest(const ChromeOriginTrialsTest&) = delete;
  41. ChromeOriginTrialsTest& operator=(const ChromeOriginTrialsTest&) = delete;
  42. protected:
  43. ChromeOriginTrialsTest() {}
  44. std::string GetCommandLineSwitch(const base::StringPiece& switch_name) {
  45. base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
  46. EXPECT_TRUE(command_line->HasSwitch(switch_name));
  47. return command_line->GetSwitchValueASCII(switch_name);
  48. }
  49. void AddDisabledFeaturesToPrefs(const std::vector<std::string>& features) {
  50. base::Value disabled_feature_list(base::Value::Type::LIST);
  51. for (const std::string& feature : features) {
  52. disabled_feature_list.Append(feature);
  53. }
  54. ListPrefUpdate update(
  55. local_state(), embedder_support::prefs::kOriginTrialDisabledFeatures);
  56. *update = std::move(disabled_feature_list);
  57. }
  58. void AddDisabledTokensToPrefs(const std::vector<std::string>& tokens) {
  59. base::Value disabled_token_list(base::Value::Type::LIST);
  60. for (const std::string& token : tokens) {
  61. disabled_token_list.Append(token);
  62. }
  63. ListPrefUpdate update(local_state(),
  64. embedder_support::prefs::kOriginTrialDisabledTokens);
  65. *update = std::move(disabled_token_list);
  66. }
  67. PrefService* local_state() { return g_browser_process->local_state(); }
  68. };
  69. // Tests to verify that the command line is not set, when no prefs exist for
  70. // the various updates.
  71. IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, NoPublicKeySet) {
  72. base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
  73. EXPECT_FALSE(
  74. command_line->HasSwitch(embedder_support::kOriginTrialPublicKey));
  75. }
  76. IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, NoDisabledFeatures) {
  77. base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
  78. EXPECT_FALSE(
  79. command_line->HasSwitch(embedder_support::kOriginTrialDisabledFeatures));
  80. }
  81. IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, NoDisabledTokens) {
  82. base::CommandLine* command_line = base::CommandLine::ForCurrentProcess();
  83. EXPECT_FALSE(
  84. command_line->HasSwitch(embedder_support::kOriginTrialDisabledTokens));
  85. }
  86. // Tests to verify that the public key is correctly read from prefs and
  87. // added to the command line
  88. IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, PRE_PublicKeySetOnCommandLine) {
  89. local_state()->Set(embedder_support::prefs::kOriginTrialPublicKey,
  90. base::Value(kNewPublicKey));
  91. ASSERT_EQ(kNewPublicKey, local_state()->GetString(
  92. embedder_support::prefs::kOriginTrialPublicKey));
  93. }
  94. IN_PROC_BROWSER_TEST_F(ChromeOriginTrialsTest, PublicKeySetOnCommandLine) {
  95. ASSERT_EQ(kNewPublicKey, local_state()->GetString(
  96. embedder_support::prefs::kOriginTrialPublicKey));
  97. std::string actual =
  98. GetCommandLineSwitch(embedder_support::kOriginTrialPublicKey);
  99. EXPECT_EQ(kNewPublicKey, actual);
  100. }
  101. // Tests to verify that disabled features are correctly read from prefs and
  102. // added to the command line
  103. class ChromeOriginTrialsDisabledFeaturesTest
  104. : public ChromeOriginTrialsTest,
  105. public ::testing::WithParamInterface<DisabledItemsTestData> {};
  106. IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledFeaturesTest,
  107. PRE_DisabledFeaturesSetOnCommandLine) {
  108. AddDisabledFeaturesToPrefs(GetParam().input_list);
  109. ASSERT_TRUE(local_state()->HasPrefPath(
  110. embedder_support::prefs::kOriginTrialDisabledFeatures));
  111. }
  112. IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledFeaturesTest,
  113. DisabledFeaturesSetOnCommandLine) {
  114. ASSERT_TRUE(local_state()->HasPrefPath(
  115. embedder_support::prefs::kOriginTrialDisabledFeatures));
  116. std::string actual =
  117. GetCommandLineSwitch(embedder_support::kOriginTrialDisabledFeatures);
  118. EXPECT_EQ(GetParam().expected_switch, actual);
  119. }
  120. INSTANTIATE_TEST_SUITE_P(All,
  121. ChromeOriginTrialsDisabledFeaturesTest,
  122. ::testing::ValuesIn(kDisabledFeaturesTests));
  123. // Tests to verify that disabled tokens are correctly read from prefs and
  124. // added to the command line
  125. class ChromeOriginTrialsDisabledTokensTest
  126. : public ChromeOriginTrialsTest,
  127. public ::testing::WithParamInterface<DisabledItemsTestData> {};
  128. IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledTokensTest,
  129. PRE_DisabledTokensSetOnCommandLine) {
  130. AddDisabledTokensToPrefs(GetParam().input_list);
  131. ASSERT_TRUE(local_state()->HasPrefPath(
  132. embedder_support::prefs::kOriginTrialDisabledTokens));
  133. }
  134. IN_PROC_BROWSER_TEST_P(ChromeOriginTrialsDisabledTokensTest,
  135. DisabledTokensSetOnCommandLine) {
  136. ASSERT_TRUE(local_state()->HasPrefPath(
  137. embedder_support::prefs::kOriginTrialDisabledTokens));
  138. std::string actual =
  139. GetCommandLineSwitch(embedder_support::kOriginTrialDisabledTokens);
  140. EXPECT_EQ(GetParam().expected_switch, actual);
  141. }
  142. INSTANTIATE_TEST_SUITE_P(All,
  143. ChromeOriginTrialsDisabledTokensTest,
  144. ::testing::ValuesIn(kDisabledTokensTests));
  145. } // namespace