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