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