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 "components/variations/metrics_util.h" 8#include "testing/gtest/include/gtest/gtest.h" 9 10namespace variations { 11 12TEST(VariationsUtilTest, GetFieldTrialActiveGroups) { 13 typedef std::set<ActiveGroupId, ActiveGroupIdCompare> ActiveGroupIdSet; 14 std::string trial_one("trial one"); 15 std::string group_one("group one"); 16 std::string trial_two("trial two"); 17 std::string group_two("group two"); 18 19 base::FieldTrial::ActiveGroups active_groups; 20 base::FieldTrial::ActiveGroup active_group; 21 active_group.trial_name = trial_one; 22 active_group.group_name = group_one; 23 active_groups.push_back(active_group); 24 25 active_group.trial_name = trial_two; 26 active_group.group_name = group_two; 27 active_groups.push_back(active_group); 28 29 // Create our expected groups of IDs. 30 ActiveGroupIdSet expected_groups; 31 ActiveGroupId name_group_id; 32 name_group_id.name = metrics::HashName(trial_one); 33 name_group_id.group = metrics::HashName(group_one); 34 expected_groups.insert(name_group_id); 35 name_group_id.name = metrics::HashName(trial_two); 36 name_group_id.group = metrics::HashName(group_two); 37 expected_groups.insert(name_group_id); 38 39 std::vector<ActiveGroupId> active_group_ids; 40 testing::TestGetFieldTrialActiveGroupIds(active_groups, &active_group_ids); 41 EXPECT_EQ(2U, active_group_ids.size()); 42 for (size_t i = 0; i < active_group_ids.size(); ++i) { 43 ActiveGroupIdSet::iterator expected_group = 44 expected_groups.find(active_group_ids[i]); 45 EXPECT_FALSE(expected_group == expected_groups.end()); 46 expected_groups.erase(expected_group); 47 } 48 EXPECT_EQ(0U, expected_groups.size()); 49} 50 51} // namespace variations 52