| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/about_flags.h" | 5 #include "chrome/browser/about_flags.h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <map> |
| 8 #include <utility> | 8 #include <set> |
| 9 #include <string> |
| 9 | 10 |
| 10 #include "base/feature_list.h" | 11 #include "base/feature_list.h" |
| 11 #include "base/files/file_path.h" | 12 #include "base/files/file_path.h" |
| 12 #include "base/format_macros.h" | 13 #include "base/format_macros.h" |
| 13 #include "base/path_service.h" | 14 #include "base/path_service.h" |
| 14 #include "base/prefs/pref_registry_simple.h" | |
| 15 #include "base/prefs/testing_pref_service.h" | |
| 16 #include "base/strings/string_number_conversions.h" | 15 #include "base/strings/string_number_conversions.h" |
| 17 #include "base/strings/string_split.h" | |
| 18 #include "base/strings/stringprintf.h" | 16 #include "base/strings/stringprintf.h" |
| 19 #include "base/strings/utf_string_conversions.h" | 17 #include "base/strings/utf_string_conversions.h" |
| 20 #include "base/values.h" | 18 #include "base/values.h" |
| 21 #include "chrome/common/chrome_switches.h" | 19 #include "components/flags_ui/feature_entry.h" |
| 22 #include "chrome/grit/chromium_strings.h" | |
| 23 #include "components/flags_ui/flags_ui_pref_names.h" | |
| 24 #include "components/flags_ui/pref_service_flags_storage.h" | |
| 25 #include "content/public/common/content_switches.h" | |
| 26 #include "testing/gtest/include/gtest/gtest.h" | 20 #include "testing/gtest/include/gtest/gtest.h" |
| 27 #include "third_party/libxml/chromium/libxml_utils.h" | 21 #include "third_party/libxml/chromium/libxml_utils.h" |
| 28 | 22 |
| 29 using flags_ui::FeatureEntry; | |
| 30 | |
| 31 namespace about_flags { | 23 namespace about_flags { |
| 32 | 24 |
| 33 namespace { | 25 namespace { |
| 34 | 26 |
| 35 const char kFlags1[] = "flag1"; | |
| 36 const char kFlags2[] = "flag2"; | |
| 37 const char kFlags3[] = "flag3"; | |
| 38 const char kFlags4[] = "flag4"; | |
| 39 const char kFlags5[] = "flag5"; | |
| 40 const char kFlags6[] = "flag6"; | |
| 41 const char kFlags7[] = "flag7"; | |
| 42 | |
| 43 const char kSwitch1[] = "switch"; | |
| 44 const char kSwitch2[] = "switch2"; | |
| 45 const char kSwitch3[] = "switch3"; | |
| 46 const char kSwitch6[] = "switch6"; | |
| 47 const char kValueForSwitch2[] = "value_for_switch2"; | |
| 48 | |
| 49 const char kMultiSwitch1[] = "multi_switch1"; | |
| 50 const char kMultiSwitch2[] = "multi_switch2"; | |
| 51 const char kValueForMultiSwitch2[] = "value_for_multi_switch2"; | |
| 52 | |
| 53 const char kEnableDisableValue1[] = "value1"; | |
| 54 const char kEnableDisableValue2[] = "value2"; | |
| 55 | |
| 56 typedef base::HistogramBase::Sample Sample; | 27 typedef base::HistogramBase::Sample Sample; |
| 57 typedef std::map<std::string, Sample> SwitchToIdMap; | 28 typedef std::map<std::string, Sample> SwitchToIdMap; |
| 58 | 29 |
| 59 // This is a helper function to the ReadEnumFromHistogramsXml(). | 30 // This is a helper function to the ReadEnumFromHistogramsXml(). |
| 60 // Extracts single enum (with integer values) from histograms.xml. | 31 // Extracts single enum (with integer values) from histograms.xml. |
| 61 // Expects |reader| to point at given enum. | 32 // Expects |reader| to point at given enum. |
| 62 // Returns map { value => label }. | 33 // Returns map { value => label }. |
| 63 // Returns empty map on error. | 34 // Returns empty map on error. |
| 64 std::map<Sample, std::string> ParseEnumFromHistogramsXml( | 35 std::map<Sample, std::string> ParseEnumFromHistogramsXml( |
| 65 const std::string& enum_name, | 36 const std::string& enum_name, |
| (...skipping 126 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 192 return path; | 163 return path; |
| 193 #endif | 164 #endif |
| 194 } | 165 } |
| 195 | 166 |
| 196 // Get all associated switches corresponding to defined about_flags.cc entries. | 167 // Get all associated switches corresponding to defined about_flags.cc entries. |
| 197 // Does not include information about FEATURE_VALUE entries. | 168 // Does not include information about FEATURE_VALUE entries. |
| 198 std::set<std::string> GetAllSwitchesForTesting() { | 169 std::set<std::string> GetAllSwitchesForTesting() { |
| 199 std::set<std::string> result; | 170 std::set<std::string> result; |
| 200 | 171 |
| 201 size_t num_entries = 0; | 172 size_t num_entries = 0; |
| 202 const FeatureEntry* entries = | 173 const flags_ui::FeatureEntry* entries = |
| 203 testing::GetFeatureEntries(&num_entries); | 174 testing::GetFeatureEntries(&num_entries); |
| 204 | 175 |
| 205 for (size_t i = 0; i < num_entries; ++i) { | 176 for (size_t i = 0; i < num_entries; ++i) { |
| 206 const FeatureEntry& entry = entries[i]; | 177 const flags_ui::FeatureEntry& entry = entries[i]; |
| 207 switch (entry.type) { | 178 switch (entry.type) { |
| 208 case FeatureEntry::SINGLE_VALUE: | 179 case flags_ui::FeatureEntry::SINGLE_VALUE: |
| 209 case FeatureEntry::SINGLE_DISABLE_VALUE: | 180 case flags_ui::FeatureEntry::SINGLE_DISABLE_VALUE: |
| 210 result.insert(entry.command_line_switch); | 181 result.insert(entry.command_line_switch); |
| 211 break; | 182 break; |
| 212 case FeatureEntry::MULTI_VALUE: | 183 case flags_ui::FeatureEntry::MULTI_VALUE: |
| 213 for (int j = 0; j < entry.num_choices; ++j) { | 184 for (int j = 0; j < entry.num_choices; ++j) { |
| 214 result.insert(entry.choices[j].command_line_switch); | 185 result.insert(entry.choices[j].command_line_switch); |
| 215 } | 186 } |
| 216 break; | 187 break; |
| 217 case FeatureEntry::ENABLE_DISABLE_VALUE: | 188 case flags_ui::FeatureEntry::ENABLE_DISABLE_VALUE: |
| 218 result.insert(entry.command_line_switch); | 189 result.insert(entry.command_line_switch); |
| 219 result.insert(entry.disable_command_line_switch); | 190 result.insert(entry.disable_command_line_switch); |
| 220 break; | 191 break; |
| 221 case FeatureEntry::FEATURE_VALUE: | 192 case flags_ui::FeatureEntry::FEATURE_VALUE: |
| 222 break; | 193 break; |
| 223 } | 194 } |
| 224 } | 195 } |
| 225 return result; | 196 return result; |
| 226 } | 197 } |
| 227 | 198 |
| 228 } // anonymous namespace | 199 } // anonymous namespace |
| 229 | 200 |
| 230 const FeatureEntry::Choice kMultiChoices[] = { | |
| 231 { IDS_PRODUCT_NAME, "", "" }, | |
| 232 { IDS_PRODUCT_NAME, kMultiSwitch1, "" }, | |
| 233 { IDS_PRODUCT_NAME, kMultiSwitch2, kValueForMultiSwitch2 }, | |
| 234 }; | |
| 235 | |
| 236 const base::Feature kTestFeature{"FeatureName", | |
| 237 base::FEATURE_ENABLED_BY_DEFAULT}; | |
| 238 | |
| 239 // The entries that are set for these tests. The 3rd entry is not supported on | |
| 240 // the current platform, all others are. | |
| 241 static FeatureEntry kEntries[] = { | |
| 242 {kFlags1, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, | |
| 243 0, // Ends up being mapped to the current platform. | |
| 244 FeatureEntry::SINGLE_VALUE, kSwitch1, "", nullptr, nullptr, nullptr, | |
| 245 nullptr, 0}, | |
| 246 {kFlags2, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, | |
| 247 0, // Ends up being mapped to the current platform. | |
| 248 FeatureEntry::SINGLE_VALUE, kSwitch2, kValueForSwitch2, nullptr, nullptr, | |
| 249 nullptr, nullptr, 0}, | |
| 250 {kFlags3, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, | |
| 251 0, // This ends up enabling for an OS other than the current. | |
| 252 FeatureEntry::SINGLE_VALUE, kSwitch3, "", nullptr, nullptr, nullptr, | |
| 253 nullptr, 0}, | |
| 254 {kFlags4, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, | |
| 255 0, // Ends up being mapped to the current platform. | |
| 256 FeatureEntry::MULTI_VALUE, "", "", "", "", nullptr, kMultiChoices, | |
| 257 arraysize(kMultiChoices)}, | |
| 258 {kFlags5, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, | |
| 259 0, // Ends up being mapped to the current platform. | |
| 260 FeatureEntry::ENABLE_DISABLE_VALUE, kSwitch1, kEnableDisableValue1, | |
| 261 kSwitch2, kEnableDisableValue2, nullptr, nullptr, 3}, | |
| 262 {kFlags6, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, 0, | |
| 263 FeatureEntry::SINGLE_DISABLE_VALUE, kSwitch6, "", nullptr, nullptr, | |
| 264 nullptr, nullptr, 0}, | |
| 265 {kFlags7, IDS_PRODUCT_NAME, IDS_PRODUCT_NAME, | |
| 266 0, // Ends up being mapped to the current platform. | |
| 267 FeatureEntry::FEATURE_VALUE, nullptr, nullptr, nullptr, nullptr, | |
| 268 &kTestFeature, nullptr, 3}, | |
| 269 }; | |
| 270 | |
| 271 class AboutFlagsTest : public ::testing::Test { | |
| 272 protected: | |
| 273 AboutFlagsTest() : flags_storage_(&prefs_) { | |
| 274 prefs_.registry()->RegisterListPref( | |
| 275 flags_ui::prefs::kEnabledLabsExperiments); | |
| 276 testing::ClearState(); | |
| 277 } | |
| 278 | |
| 279 void SetUp() override { | |
| 280 for (size_t i = 0; i < arraysize(kEntries); ++i) | |
| 281 kEntries[i].supported_platforms = GetCurrentPlatform(); | |
| 282 | |
| 283 int os_other_than_current = 1; | |
| 284 while (os_other_than_current == GetCurrentPlatform()) | |
| 285 os_other_than_current <<= 1; | |
| 286 kEntries[2].supported_platforms = os_other_than_current; | |
| 287 | |
| 288 testing::SetFeatureEntries(kEntries, arraysize(kEntries)); | |
| 289 } | |
| 290 | |
| 291 void TearDown() override { testing::SetFeatureEntries(nullptr, 0); } | |
| 292 | |
| 293 TestingPrefServiceSimple prefs_; | |
| 294 flags_ui::PrefServiceFlagsStorage flags_storage_; | |
| 295 }; | |
| 296 | |
| 297 | |
| 298 TEST_F(AboutFlagsTest, NoChangeNoRestart) { | |
| 299 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 300 SetFeatureEntryEnabled(&flags_storage_, kFlags1, false); | |
| 301 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 302 | |
| 303 // kFlags6 is enabled by default, so enabling should not require a restart. | |
| 304 SetFeatureEntryEnabled(&flags_storage_, kFlags6, true); | |
| 305 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 306 } | |
| 307 | |
| 308 TEST_F(AboutFlagsTest, ChangeNeedsRestart) { | |
| 309 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 310 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 311 EXPECT_TRUE(IsRestartNeededToCommitChanges()); | |
| 312 } | |
| 313 | |
| 314 // Tests that disabling a default enabled entry requires a restart. | |
| 315 TEST_F(AboutFlagsTest, DisableChangeNeedsRestart) { | |
| 316 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 317 SetFeatureEntryEnabled(&flags_storage_, kFlags6, false); | |
| 318 EXPECT_TRUE(IsRestartNeededToCommitChanges()); | |
| 319 } | |
| 320 | |
| 321 TEST_F(AboutFlagsTest, MultiFlagChangeNeedsRestart) { | |
| 322 const FeatureEntry& entry = kEntries[3]; | |
| 323 ASSERT_EQ(kFlags4, entry.internal_name); | |
| 324 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 325 // Enable the 2nd choice of the multi-value. | |
| 326 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(2), true); | |
| 327 EXPECT_TRUE(IsRestartNeededToCommitChanges()); | |
| 328 testing::ClearState(); | |
| 329 EXPECT_FALSE(IsRestartNeededToCommitChanges()); | |
| 330 // Enable the default choice now. | |
| 331 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(0), true); | |
| 332 EXPECT_TRUE(IsRestartNeededToCommitChanges()); | |
| 333 } | |
| 334 | |
| 335 TEST_F(AboutFlagsTest, AddTwoFlagsRemoveOne) { | |
| 336 // Add two entries, check they're there. | |
| 337 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 338 SetFeatureEntryEnabled(&flags_storage_, kFlags2, true); | |
| 339 | |
| 340 const base::ListValue* entries_list = | |
| 341 prefs_.GetList(flags_ui::prefs::kEnabledLabsExperiments); | |
| 342 ASSERT_TRUE(entries_list != nullptr); | |
| 343 | |
| 344 ASSERT_EQ(2u, entries_list->GetSize()); | |
| 345 | |
| 346 std::string s0; | |
| 347 ASSERT_TRUE(entries_list->GetString(0, &s0)); | |
| 348 std::string s1; | |
| 349 ASSERT_TRUE(entries_list->GetString(1, &s1)); | |
| 350 | |
| 351 EXPECT_TRUE(s0 == kFlags1 || s1 == kFlags1); | |
| 352 EXPECT_TRUE(s0 == kFlags2 || s1 == kFlags2); | |
| 353 | |
| 354 // Remove one entry, check the other's still around. | |
| 355 SetFeatureEntryEnabled(&flags_storage_, kFlags2, false); | |
| 356 | |
| 357 entries_list = prefs_.GetList(flags_ui::prefs::kEnabledLabsExperiments); | |
| 358 ASSERT_TRUE(entries_list != nullptr); | |
| 359 ASSERT_EQ(1u, entries_list->GetSize()); | |
| 360 ASSERT_TRUE(entries_list->GetString(0, &s0)); | |
| 361 EXPECT_TRUE(s0 == kFlags1); | |
| 362 } | |
| 363 | |
| 364 TEST_F(AboutFlagsTest, AddTwoFlagsRemoveBoth) { | |
| 365 // Add two entries, check the pref exists. | |
| 366 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 367 SetFeatureEntryEnabled(&flags_storage_, kFlags2, true); | |
| 368 const base::ListValue* entries_list = | |
| 369 prefs_.GetList(flags_ui::prefs::kEnabledLabsExperiments); | |
| 370 ASSERT_TRUE(entries_list != nullptr); | |
| 371 | |
| 372 // Remove both, the pref should have been removed completely. | |
| 373 SetFeatureEntryEnabled(&flags_storage_, kFlags1, false); | |
| 374 SetFeatureEntryEnabled(&flags_storage_, kFlags2, false); | |
| 375 entries_list = prefs_.GetList(flags_ui::prefs::kEnabledLabsExperiments); | |
| 376 EXPECT_TRUE(entries_list == nullptr || entries_list->GetSize() == 0); | |
| 377 } | |
| 378 | |
| 379 TEST_F(AboutFlagsTest, ConvertFlagsToSwitches) { | |
| 380 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 381 | |
| 382 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 383 command_line.AppendSwitch("foo"); | |
| 384 | |
| 385 EXPECT_TRUE(command_line.HasSwitch("foo")); | |
| 386 EXPECT_FALSE(command_line.HasSwitch(kSwitch1)); | |
| 387 | |
| 388 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 389 | |
| 390 EXPECT_TRUE(command_line.HasSwitch("foo")); | |
| 391 EXPECT_TRUE(command_line.HasSwitch(kSwitch1)); | |
| 392 EXPECT_TRUE(command_line.HasSwitch(switches::kFlagSwitchesBegin)); | |
| 393 EXPECT_TRUE(command_line.HasSwitch(switches::kFlagSwitchesEnd)); | |
| 394 | |
| 395 base::CommandLine command_line2(base::CommandLine::NO_PROGRAM); | |
| 396 | |
| 397 ConvertFlagsToSwitches(&flags_storage_, &command_line2, kNoSentinels); | |
| 398 | |
| 399 EXPECT_TRUE(command_line2.HasSwitch(kSwitch1)); | |
| 400 EXPECT_FALSE(command_line2.HasSwitch(switches::kFlagSwitchesBegin)); | |
| 401 EXPECT_FALSE(command_line2.HasSwitch(switches::kFlagSwitchesEnd)); | |
| 402 } | |
| 403 | |
| 404 base::CommandLine::StringType CreateSwitch(const std::string& value) { | |
| 405 #if defined(OS_WIN) | |
| 406 return base::ASCIIToUTF16(value); | |
| 407 #else | |
| 408 return value; | |
| 409 #endif | |
| 410 } | |
| 411 | |
| 412 TEST_F(AboutFlagsTest, CompareSwitchesToCurrentCommandLine) { | |
| 413 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 414 | |
| 415 const std::string kDoubleDash("--"); | |
| 416 | |
| 417 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 418 command_line.AppendSwitch("foo"); | |
| 419 | |
| 420 base::CommandLine new_command_line(base::CommandLine::NO_PROGRAM); | |
| 421 ConvertFlagsToSwitches(&flags_storage_, &new_command_line, kAddSentinels); | |
| 422 | |
| 423 EXPECT_FALSE(AreSwitchesIdenticalToCurrentCommandLine(new_command_line, | |
| 424 command_line, nullptr)); | |
| 425 { | |
| 426 std::set<base::CommandLine::StringType> difference; | |
| 427 EXPECT_FALSE(AreSwitchesIdenticalToCurrentCommandLine( | |
| 428 new_command_line, command_line, &difference)); | |
| 429 EXPECT_EQ(1U, difference.size()); | |
| 430 EXPECT_EQ(1U, difference.count(CreateSwitch(kDoubleDash + kSwitch1))); | |
| 431 } | |
| 432 | |
| 433 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 434 | |
| 435 EXPECT_TRUE(AreSwitchesIdenticalToCurrentCommandLine(new_command_line, | |
| 436 command_line, nullptr)); | |
| 437 { | |
| 438 std::set<base::CommandLine::StringType> difference; | |
| 439 EXPECT_TRUE(AreSwitchesIdenticalToCurrentCommandLine( | |
| 440 new_command_line, command_line, &difference)); | |
| 441 EXPECT_TRUE(difference.empty()); | |
| 442 } | |
| 443 | |
| 444 // Now both have flags but different. | |
| 445 SetFeatureEntryEnabled(&flags_storage_, kFlags1, false); | |
| 446 SetFeatureEntryEnabled(&flags_storage_, kFlags2, true); | |
| 447 | |
| 448 base::CommandLine another_command_line(base::CommandLine::NO_PROGRAM); | |
| 449 ConvertFlagsToSwitches(&flags_storage_, &another_command_line, kAddSentinels); | |
| 450 | |
| 451 EXPECT_FALSE(AreSwitchesIdenticalToCurrentCommandLine( | |
| 452 new_command_line, another_command_line, nullptr)); | |
| 453 { | |
| 454 std::set<base::CommandLine::StringType> difference; | |
| 455 EXPECT_FALSE(AreSwitchesIdenticalToCurrentCommandLine( | |
| 456 new_command_line, another_command_line, &difference)); | |
| 457 EXPECT_EQ(2U, difference.size()); | |
| 458 EXPECT_EQ(1U, difference.count(CreateSwitch(kDoubleDash + kSwitch1))); | |
| 459 EXPECT_EQ(1U, | |
| 460 difference.count(CreateSwitch(kDoubleDash + kSwitch2 + "=" + | |
| 461 kValueForSwitch2))); | |
| 462 } | |
| 463 } | |
| 464 | |
| 465 TEST_F(AboutFlagsTest, RemoveFlagSwitches) { | |
| 466 std::map<std::string, base::CommandLine::StringType> switch_list; | |
| 467 switch_list[kSwitch1] = base::CommandLine::StringType(); | |
| 468 switch_list[switches::kFlagSwitchesBegin] = base::CommandLine::StringType(); | |
| 469 switch_list[switches::kFlagSwitchesEnd] = base::CommandLine::StringType(); | |
| 470 switch_list["foo"] = base::CommandLine::StringType(); | |
| 471 | |
| 472 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 473 | |
| 474 // This shouldn't do anything before ConvertFlagsToSwitches() wasn't called. | |
| 475 RemoveFlagsSwitches(&switch_list); | |
| 476 ASSERT_EQ(4u, switch_list.size()); | |
| 477 EXPECT_TRUE(ContainsKey(switch_list, kSwitch1)); | |
| 478 EXPECT_TRUE(ContainsKey(switch_list, switches::kFlagSwitchesBegin)); | |
| 479 EXPECT_TRUE(ContainsKey(switch_list, switches::kFlagSwitchesEnd)); | |
| 480 EXPECT_TRUE(ContainsKey(switch_list, "foo")); | |
| 481 | |
| 482 // Call ConvertFlagsToSwitches(), then RemoveFlagsSwitches() again. | |
| 483 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 484 command_line.AppendSwitch("foo"); | |
| 485 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 486 RemoveFlagsSwitches(&switch_list); | |
| 487 | |
| 488 // Now the about:flags-related switch should have been removed. | |
| 489 ASSERT_EQ(1u, switch_list.size()); | |
| 490 EXPECT_TRUE(ContainsKey(switch_list, "foo")); | |
| 491 } | |
| 492 | |
| 493 TEST_F(AboutFlagsTest, RemoveFlagSwitches_Features) { | |
| 494 struct { | |
| 495 int enabled_choice; // 0: default, 1: enabled, 2: disabled. | |
| 496 const char* existing_enable_features; | |
| 497 const char* existing_disable_features; | |
| 498 const char* expected_enable_features; | |
| 499 const char* expected_disable_features; | |
| 500 } cases[] = { | |
| 501 // Default value: Should not affect existing flags. | |
| 502 {0, nullptr, nullptr, nullptr, nullptr}, | |
| 503 {0, "A,B", "C", "A,B", "C"}, | |
| 504 // "Enable" option: should only affect enabled list. | |
| 505 {1, nullptr, nullptr, "FeatureName", nullptr}, | |
| 506 {1, "A,B", "C", "A,B,FeatureName", "C"}, | |
| 507 // "Disable" option: should only affect disabled list. | |
| 508 {2, nullptr, nullptr, nullptr, "FeatureName"}, | |
| 509 {2, "A,B", "C", "A,B", "C,FeatureName"}, | |
| 510 }; | |
| 511 | |
| 512 for (size_t i = 0; i < arraysize(cases); ++i) { | |
| 513 SCOPED_TRACE(base::StringPrintf( | |
| 514 "Test[%" PRIuS "]: %d [%s] [%s]", i, cases[i].enabled_choice, | |
| 515 cases[i].existing_enable_features ? cases[i].existing_enable_features | |
| 516 : "null", | |
| 517 cases[i].existing_disable_features ? cases[i].existing_disable_features | |
| 518 : "null")); | |
| 519 | |
| 520 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 521 if (cases[i].existing_enable_features) { | |
| 522 command_line.AppendSwitchASCII(switches::kEnableFeatures, | |
| 523 cases[i].existing_enable_features); | |
| 524 } | |
| 525 if (cases[i].existing_disable_features) { | |
| 526 command_line.AppendSwitchASCII(switches::kDisableFeatures, | |
| 527 cases[i].existing_disable_features); | |
| 528 } | |
| 529 | |
| 530 testing::ClearState(); | |
| 531 | |
| 532 const std::string entry_name = base::StringPrintf( | |
| 533 "%s%s%d", kFlags7, flags_ui::testing::kMultiSeparator, | |
| 534 cases[i].enabled_choice); | |
| 535 SetFeatureEntryEnabled(&flags_storage_, entry_name, true); | |
| 536 | |
| 537 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 538 auto switch_list = command_line.GetSwitches(); | |
| 539 EXPECT_EQ(cases[i].expected_enable_features != nullptr, | |
| 540 ContainsKey(switch_list, switches::kEnableFeatures)); | |
| 541 if (cases[i].expected_enable_features) | |
| 542 EXPECT_EQ(CreateSwitch(cases[i].expected_enable_features), | |
| 543 switch_list[switches::kEnableFeatures]); | |
| 544 | |
| 545 EXPECT_EQ(cases[i].expected_disable_features != nullptr, | |
| 546 ContainsKey(switch_list, switches::kDisableFeatures)); | |
| 547 if (cases[i].expected_disable_features) | |
| 548 EXPECT_EQ(CreateSwitch(cases[i].expected_disable_features), | |
| 549 switch_list[switches::kDisableFeatures]); | |
| 550 | |
| 551 // RemoveFlagsSwitches() should result in the original values for these | |
| 552 // switches. | |
| 553 switch_list = command_line.GetSwitches(); | |
| 554 RemoveFlagsSwitches(&switch_list); | |
| 555 EXPECT_EQ(cases[i].existing_enable_features != nullptr, | |
| 556 ContainsKey(switch_list, switches::kEnableFeatures)); | |
| 557 if (cases[i].existing_enable_features) | |
| 558 EXPECT_EQ(CreateSwitch(cases[i].existing_enable_features), | |
| 559 switch_list[switches::kEnableFeatures]); | |
| 560 EXPECT_EQ(cases[i].existing_disable_features != nullptr, | |
| 561 ContainsKey(switch_list, switches::kEnableFeatures)); | |
| 562 if (cases[i].existing_disable_features) | |
| 563 EXPECT_EQ(CreateSwitch(cases[i].existing_disable_features), | |
| 564 switch_list[switches::kDisableFeatures]); | |
| 565 } | |
| 566 } | |
| 567 | |
| 568 // Tests enabling entries that aren't supported on the current platform. | |
| 569 TEST_F(AboutFlagsTest, PersistAndPrune) { | |
| 570 // Enable entries 1 and 3. | |
| 571 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 572 SetFeatureEntryEnabled(&flags_storage_, kFlags3, true); | |
| 573 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 574 EXPECT_FALSE(command_line.HasSwitch(kSwitch1)); | |
| 575 EXPECT_FALSE(command_line.HasSwitch(kSwitch3)); | |
| 576 | |
| 577 // Convert the flags to switches. Entry 3 shouldn't be among the switches | |
| 578 // as it is not applicable to the current platform. | |
| 579 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 580 EXPECT_TRUE(command_line.HasSwitch(kSwitch1)); | |
| 581 EXPECT_FALSE(command_line.HasSwitch(kSwitch3)); | |
| 582 | |
| 583 // FeatureEntry 3 should show still be persisted in preferences though. | |
| 584 const base::ListValue* entries_list = | |
| 585 prefs_.GetList(flags_ui::prefs::kEnabledLabsExperiments); | |
| 586 ASSERT_TRUE(entries_list); | |
| 587 EXPECT_EQ(2U, entries_list->GetSize()); | |
| 588 std::string s0; | |
| 589 ASSERT_TRUE(entries_list->GetString(0, &s0)); | |
| 590 EXPECT_EQ(kFlags1, s0); | |
| 591 std::string s1; | |
| 592 ASSERT_TRUE(entries_list->GetString(1, &s1)); | |
| 593 EXPECT_EQ(kFlags3, s1); | |
| 594 } | |
| 595 | |
| 596 // Tests that switches which should have values get them in the command | |
| 597 // line. | |
| 598 TEST_F(AboutFlagsTest, CheckValues) { | |
| 599 // Enable entries 1 and 2. | |
| 600 SetFeatureEntryEnabled(&flags_storage_, kFlags1, true); | |
| 601 SetFeatureEntryEnabled(&flags_storage_, kFlags2, true); | |
| 602 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 603 EXPECT_FALSE(command_line.HasSwitch(kSwitch1)); | |
| 604 EXPECT_FALSE(command_line.HasSwitch(kSwitch2)); | |
| 605 | |
| 606 // Convert the flags to switches. | |
| 607 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 608 EXPECT_TRUE(command_line.HasSwitch(kSwitch1)); | |
| 609 EXPECT_EQ(std::string(), command_line.GetSwitchValueASCII(kSwitch1)); | |
| 610 EXPECT_TRUE(command_line.HasSwitch(kSwitch2)); | |
| 611 EXPECT_EQ(std::string(kValueForSwitch2), | |
| 612 command_line.GetSwitchValueASCII(kSwitch2)); | |
| 613 | |
| 614 // Confirm that there is no '=' in the command line for simple switches. | |
| 615 std::string switch1_with_equals = std::string("--") + | |
| 616 std::string(kSwitch1) + | |
| 617 std::string("="); | |
| 618 #if defined(OS_WIN) | |
| 619 EXPECT_EQ(base::string16::npos, | |
| 620 command_line.GetCommandLineString().find( | |
| 621 base::ASCIIToUTF16(switch1_with_equals))); | |
| 622 #else | |
| 623 EXPECT_EQ(std::string::npos, | |
| 624 command_line.GetCommandLineString().find(switch1_with_equals)); | |
| 625 #endif | |
| 626 | |
| 627 // And confirm there is a '=' for switches with values. | |
| 628 std::string switch2_with_equals = std::string("--") + | |
| 629 std::string(kSwitch2) + | |
| 630 std::string("="); | |
| 631 #if defined(OS_WIN) | |
| 632 EXPECT_NE(base::string16::npos, | |
| 633 command_line.GetCommandLineString().find( | |
| 634 base::ASCIIToUTF16(switch2_with_equals))); | |
| 635 #else | |
| 636 EXPECT_NE(std::string::npos, | |
| 637 command_line.GetCommandLineString().find(switch2_with_equals)); | |
| 638 #endif | |
| 639 | |
| 640 // And it should persist. | |
| 641 const base::ListValue* entries_list = | |
| 642 prefs_.GetList(flags_ui::prefs::kEnabledLabsExperiments); | |
| 643 ASSERT_TRUE(entries_list); | |
| 644 EXPECT_EQ(2U, entries_list->GetSize()); | |
| 645 std::string s0; | |
| 646 ASSERT_TRUE(entries_list->GetString(0, &s0)); | |
| 647 EXPECT_EQ(kFlags1, s0); | |
| 648 std::string s1; | |
| 649 ASSERT_TRUE(entries_list->GetString(1, &s1)); | |
| 650 EXPECT_EQ(kFlags2, s1); | |
| 651 } | |
| 652 | |
| 653 // Tests multi-value type entries. | |
| 654 TEST_F(AboutFlagsTest, MultiValues) { | |
| 655 const FeatureEntry& entry = kEntries[3]; | |
| 656 ASSERT_EQ(kFlags4, entry.internal_name); | |
| 657 | |
| 658 // Initially, the first "deactivated" option of the multi entry should | |
| 659 // be set. | |
| 660 { | |
| 661 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 662 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 663 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1)); | |
| 664 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch2)); | |
| 665 } | |
| 666 | |
| 667 // Enable the 2nd choice of the multi-value. | |
| 668 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(2), true); | |
| 669 { | |
| 670 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 671 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 672 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1)); | |
| 673 EXPECT_TRUE(command_line.HasSwitch(kMultiSwitch2)); | |
| 674 EXPECT_EQ(std::string(kValueForMultiSwitch2), | |
| 675 command_line.GetSwitchValueASCII(kMultiSwitch2)); | |
| 676 } | |
| 677 | |
| 678 // Disable the multi-value entry. | |
| 679 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(0), true); | |
| 680 { | |
| 681 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 682 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 683 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1)); | |
| 684 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch2)); | |
| 685 } | |
| 686 } | |
| 687 | |
| 688 // Tests that disable flags are added when an entry is disabled. | |
| 689 TEST_F(AboutFlagsTest, DisableFlagCommandLine) { | |
| 690 // Nothing selected. | |
| 691 { | |
| 692 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 693 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 694 EXPECT_FALSE(command_line.HasSwitch(kSwitch6)); | |
| 695 } | |
| 696 | |
| 697 // Disable the entry 6. | |
| 698 SetFeatureEntryEnabled(&flags_storage_, kFlags6, false); | |
| 699 { | |
| 700 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 701 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 702 EXPECT_TRUE(command_line.HasSwitch(kSwitch6)); | |
| 703 } | |
| 704 | |
| 705 // Enable entry 6. | |
| 706 SetFeatureEntryEnabled(&flags_storage_, kFlags6, true); | |
| 707 { | |
| 708 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 709 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 710 EXPECT_FALSE(command_line.HasSwitch(kSwitch6)); | |
| 711 } | |
| 712 } | |
| 713 | |
| 714 TEST_F(AboutFlagsTest, EnableDisableValues) { | |
| 715 const FeatureEntry& entry = kEntries[4]; | |
| 716 ASSERT_EQ(kFlags5, entry.internal_name); | |
| 717 | |
| 718 // Nothing selected. | |
| 719 { | |
| 720 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 721 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 722 EXPECT_FALSE(command_line.HasSwitch(kSwitch1)); | |
| 723 EXPECT_FALSE(command_line.HasSwitch(kSwitch2)); | |
| 724 } | |
| 725 | |
| 726 // "Enable" option selected. | |
| 727 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(1), true); | |
| 728 { | |
| 729 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 730 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 731 EXPECT_TRUE(command_line.HasSwitch(kSwitch1)); | |
| 732 EXPECT_FALSE(command_line.HasSwitch(kSwitch2)); | |
| 733 EXPECT_EQ(kEnableDisableValue1, command_line.GetSwitchValueASCII(kSwitch1)); | |
| 734 } | |
| 735 | |
| 736 // "Disable" option selected. | |
| 737 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(2), true); | |
| 738 { | |
| 739 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 740 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 741 EXPECT_FALSE(command_line.HasSwitch(kSwitch1)); | |
| 742 EXPECT_TRUE(command_line.HasSwitch(kSwitch2)); | |
| 743 EXPECT_EQ(kEnableDisableValue2, command_line.GetSwitchValueASCII(kSwitch2)); | |
| 744 } | |
| 745 | |
| 746 // "Default" option selected, same as nothing selected. | |
| 747 SetFeatureEntryEnabled(&flags_storage_, entry.NameForChoice(0), true); | |
| 748 { | |
| 749 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 750 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 751 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch1)); | |
| 752 EXPECT_FALSE(command_line.HasSwitch(kMultiSwitch2)); | |
| 753 } | |
| 754 } | |
| 755 | |
| 756 TEST_F(AboutFlagsTest, FeatureValues) { | |
| 757 const FeatureEntry& entry = kEntries[6]; | |
| 758 ASSERT_EQ(kFlags7, entry.internal_name); | |
| 759 | |
| 760 struct { | |
| 761 int enabled_choice; | |
| 762 const char* existing_enable_features; | |
| 763 const char* existing_disable_features; | |
| 764 const char* expected_enable_features; | |
| 765 const char* expected_disable_features; | |
| 766 } cases[] = { | |
| 767 // Nothing selected. | |
| 768 {-1, nullptr, nullptr, "", ""}, | |
| 769 // "Default" option selected, same as nothing selected. | |
| 770 {0, nullptr, nullptr, "", ""}, | |
| 771 // "Enable" option selected. | |
| 772 {1, nullptr, nullptr, "FeatureName", ""}, | |
| 773 // "Disable" option selected. | |
| 774 {2, nullptr, nullptr, "", "FeatureName"}, | |
| 775 // "Enable" option should get added to the existing list. | |
| 776 {1, "Foo,Bar", nullptr, "Foo,Bar,FeatureName", ""}, | |
| 777 // "Disable" option should get added to the existing list. | |
| 778 {2, nullptr, "Foo,Bar", "", "Foo,Bar,FeatureName"}, | |
| 779 }; | |
| 780 | |
| 781 for (size_t i = 0; i < arraysize(cases); ++i) { | |
| 782 SCOPED_TRACE(base::StringPrintf( | |
| 783 "Test[%" PRIuS "]: %d [%s] [%s]", i, cases[i].enabled_choice, | |
| 784 cases[i].existing_enable_features ? cases[i].existing_enable_features | |
| 785 : "null", | |
| 786 cases[i].existing_disable_features ? cases[i].existing_disable_features | |
| 787 : "null")); | |
| 788 | |
| 789 if (cases[i].enabled_choice != -1) { | |
| 790 SetFeatureEntryEnabled( | |
| 791 &flags_storage_, entry.NameForChoice(cases[i].enabled_choice), true); | |
| 792 } | |
| 793 | |
| 794 base::CommandLine command_line(base::CommandLine::NO_PROGRAM); | |
| 795 if (cases[i].existing_enable_features) { | |
| 796 command_line.AppendSwitchASCII(switches::kEnableFeatures, | |
| 797 cases[i].existing_enable_features); | |
| 798 } | |
| 799 if (cases[i].existing_disable_features) { | |
| 800 command_line.AppendSwitchASCII(switches::kDisableFeatures, | |
| 801 cases[i].existing_disable_features); | |
| 802 } | |
| 803 | |
| 804 ConvertFlagsToSwitches(&flags_storage_, &command_line, kAddSentinels); | |
| 805 EXPECT_EQ(cases[i].expected_enable_features, | |
| 806 command_line.GetSwitchValueASCII(switches::kEnableFeatures)); | |
| 807 EXPECT_EQ(cases[i].expected_disable_features, | |
| 808 command_line.GetSwitchValueASCII(switches::kDisableFeatures)); | |
| 809 } | |
| 810 } | |
| 811 | |
| 812 // Makes sure there are no separators in any of the entry names. | 201 // Makes sure there are no separators in any of the entry names. |
| 813 TEST_F(AboutFlagsTest, NoSeparators) { | 202 TEST(AboutFlagsTest, NoSeparators) { |
| 814 testing::SetFeatureEntries(nullptr, 0); | |
| 815 size_t count; | 203 size_t count; |
| 816 const FeatureEntry* entries = testing::GetFeatureEntries(&count); | 204 const flags_ui::FeatureEntry* entries = testing::GetFeatureEntries(&count); |
| 817 for (size_t i = 0; i < count; ++i) { | 205 for (size_t i = 0; i < count; ++i) { |
| 818 std::string name = entries[i].internal_name; | 206 std::string name = entries[i].internal_name; |
| 819 EXPECT_EQ(std::string::npos, name.find(flags_ui::testing::kMultiSeparator)) | 207 EXPECT_EQ(std::string::npos, name.find(flags_ui::testing::kMultiSeparator)) |
| 820 << i; | 208 << i; |
| 821 } | 209 } |
| 822 } | 210 } |
| 823 | 211 |
| 824 TEST_F(AboutFlagsTest, GetFlagFeatureEntries) { | |
| 825 base::ListValue supported_entries; | |
| 826 base::ListValue unsupported_entries; | |
| 827 GetFlagFeatureEntries(&flags_storage_, kGeneralAccessFlagsOnly, | |
| 828 &supported_entries, &unsupported_entries); | |
| 829 // All |kEntries| except for |kFlags3| should be supported. | |
| 830 EXPECT_EQ(6u, supported_entries.GetSize()); | |
| 831 EXPECT_EQ(1u, unsupported_entries.GetSize()); | |
| 832 EXPECT_EQ(arraysize(kEntries), | |
| 833 supported_entries.GetSize() + unsupported_entries.GetSize()); | |
| 834 } | |
| 835 | |
| 836 class AboutFlagsHistogramTest : public ::testing::Test { | 212 class AboutFlagsHistogramTest : public ::testing::Test { |
| 837 protected: | 213 protected: |
| 838 // This is a helper function to check that all IDs in enum LoginCustomFlags in | 214 // This is a helper function to check that all IDs in enum LoginCustomFlags in |
| 839 // histograms.xml are unique. | 215 // histograms.xml are unique. |
| 840 void SetSwitchToHistogramIdMapping(const std::string& switch_name, | 216 void SetSwitchToHistogramIdMapping(const std::string& switch_name, |
| 841 const Sample switch_histogram_id, | 217 const Sample switch_histogram_id, |
| 842 std::map<std::string, Sample>* out_map) { | 218 std::map<std::string, Sample>* out_map) { |
| 843 const std::pair<std::map<std::string, Sample>::iterator, bool> status = | 219 const std::pair<std::map<std::string, Sample>::iterator, bool> status = |
| 844 out_map->insert(std::make_pair(switch_name, switch_histogram_id)); | 220 out_map->insert(std::make_pair(switch_name, switch_histogram_id)); |
| 845 if (!status.second) { | 221 if (!status.second) { |
| (...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 922 EXPECT_TRUE(enum_entry != histograms_xml_switches_ids.end() && | 298 EXPECT_TRUE(enum_entry != histograms_xml_switches_ids.end() && |
| 923 enum_entry->first == flag) | 299 enum_entry->first == flag) |
| 924 << "histograms.xml enum LoginCustomFlags doesn't contain switch '" | 300 << "histograms.xml enum LoginCustomFlags doesn't contain switch '" |
| 925 << flag << "' (value=" << uma_id | 301 << flag << "' (value=" << uma_id |
| 926 << " expected). Consider adding entry:\n" | 302 << " expected). Consider adding entry:\n" |
| 927 << " " << GetHistogramEnumEntryText(flag, uma_id); | 303 << " " << GetHistogramEnumEntryText(flag, uma_id); |
| 928 } | 304 } |
| 929 } | 305 } |
| 930 | 306 |
| 931 } // namespace about_flags | 307 } // namespace about_flags |
| OLD | NEW |