active_field_trials_unittest.cc 2.7 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576
  1. // Copyright 2014 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 "components/variations/active_field_trials.h"
  5. #include <stddef.h>
  6. #include "base/strings/string_piece.h"
  7. #include "components/variations/hashing.h"
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. namespace variations {
  10. TEST(ActiveFieldTrialsTest, GetFieldTrialActiveGroups) {
  11. typedef std::set<ActiveGroupId, ActiveGroupIdCompare> ActiveGroupIdSet;
  12. std::string trial_one("trial one");
  13. std::string group_one("group one");
  14. std::string trial_two("trial two");
  15. std::string group_two("group two");
  16. base::FieldTrial::ActiveGroups active_groups;
  17. base::FieldTrial::ActiveGroup active_group;
  18. active_group.trial_name = trial_one;
  19. active_group.group_name = group_one;
  20. active_groups.push_back(active_group);
  21. active_group.trial_name = trial_two;
  22. active_group.group_name = group_two;
  23. active_groups.push_back(active_group);
  24. // Create our expected groups of IDs.
  25. ActiveGroupIdSet expected_groups;
  26. ActiveGroupId name_group_id;
  27. name_group_id.name = HashName(trial_one);
  28. name_group_id.group = HashName(group_one);
  29. expected_groups.insert(name_group_id);
  30. name_group_id.name = HashName(trial_two);
  31. name_group_id.group = HashName(group_two);
  32. expected_groups.insert(name_group_id);
  33. std::vector<ActiveGroupId> active_group_ids;
  34. testing::TestGetFieldTrialActiveGroupIds(base::StringPiece(), active_groups,
  35. &active_group_ids);
  36. EXPECT_EQ(2U, active_group_ids.size());
  37. for (size_t i = 0; i < active_group_ids.size(); ++i) {
  38. auto expected_group = expected_groups.find(active_group_ids[i]);
  39. EXPECT_FALSE(expected_group == expected_groups.end());
  40. expected_groups.erase(expected_group);
  41. }
  42. EXPECT_EQ(0U, expected_groups.size());
  43. }
  44. TEST(ActiveFieldTrialsTest, GetFieldTrialActiveGroupsWithSuffix) {
  45. std::string trial_one("trial one");
  46. std::string group_one("group one");
  47. std::string suffix("some_suffix");
  48. base::FieldTrial::ActiveGroups active_groups;
  49. base::FieldTrial::ActiveGroup active_group;
  50. active_group.trial_name = trial_one;
  51. active_group.group_name = group_one;
  52. active_groups.push_back(active_group);
  53. std::vector<ActiveGroupId> active_group_ids;
  54. testing::TestGetFieldTrialActiveGroupIds(suffix, active_groups,
  55. &active_group_ids);
  56. EXPECT_EQ(1U, active_group_ids.size());
  57. uint32_t expected_name = HashName("trial onesome_suffix");
  58. uint32_t expected_group = HashName("group onesome_suffix");
  59. EXPECT_EQ(expected_name, active_group_ids[0].name);
  60. EXPECT_EQ(expected_group, active_group_ids[0].group);
  61. }
  62. } // namespace variations