// Copyright 2016 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "components/variations/variations_params_manager.h" #include #include #include "base/base_switches.h" #include "base/feature_list.h" #include "base/metrics/field_trial.h" #include "base/test/scoped_feature_list.h" #include "components/variations/field_trial_config/field_trial_util.h" #include "components/variations/variations_associated_data.h" #include "components/variations/variations_switches.h" namespace variations { namespace testing { namespace { // The fixed testing group created in the provided trail when setting up params. const char kGroupTesting[] = "Testing"; base::FieldTrial* CreateFieldTrialWithParams( const std::string& trial_name, const std::map& param_values) { AssociateVariationParams(trial_name, kGroupTesting, param_values); return base::FieldTrialList::CreateFieldTrial(trial_name, kGroupTesting); } } // namespace VariationParamsManager::VariationParamsManager() : scoped_feature_list_(new base::test::ScopedFeatureList()) { scoped_feature_list_->InitWithEmptyFeatureAndFieldTrialLists(); } VariationParamsManager::VariationParamsManager( const std::string& trial_name, const std::map& param_values) : VariationParamsManager() { SetVariationParams(trial_name, param_values); } VariationParamsManager::VariationParamsManager( const std::string& trial_name, const std::map& param_values, const std::set& associated_features) : VariationParamsManager() { SetVariationParamsWithFeatureAssociations(trial_name, param_values, associated_features); } VariationParamsManager::~VariationParamsManager() { ClearAllVariationIDs(); ClearAllVariationParams(); } void VariationParamsManager::SetVariationParams( const std::string& trial_name, const std::map& param_values) { CreateFieldTrialWithParams(trial_name, param_values); } void VariationParamsManager::SetVariationParamsWithFeatureAssociations( const std::string& trial_name, const std::map& param_values, const std::set& associated_features) { scoped_feature_list_ = std::make_unique(); base::FieldTrial* field_trial = CreateFieldTrialWithParams(trial_name, param_values); std::unique_ptr feature_list(new base::FeatureList); for (const std::string& feature_name : associated_features) { feature_list->RegisterFieldTrialOverride( feature_name, base::FeatureList::OverrideState::OVERRIDE_ENABLE_FEATURE, field_trial); } scoped_feature_list_->InitWithFeatureList(std::move(feature_list)); } void VariationParamsManager::ClearAllVariationIDs() { testing::ClearAllVariationIDs(); } void VariationParamsManager::ClearAllVariationParams() { testing::ClearAllVariationParams(); // When the scoped feature list is destroyed, it puts back the original // feature list that was there when InitWithFeatureList() was called. scoped_feature_list_ = std::make_unique(); scoped_feature_list_->InitWithEmptyFeatureAndFieldTrialLists(); } // static void VariationParamsManager::AppendVariationParams( const std::string& trial_name, const std::string& trial_group_name, const std::map& param_values, base::CommandLine* command_line) { // Register the trial group. command_line->AppendSwitchASCII( ::switches::kForceFieldTrials, EscapeValue(trial_name) + "/" + EscapeValue(trial_group_name)); // Associate |param_values| with the trial group. std::string params_arg = EscapeValue(trial_name) + "." + EscapeValue(trial_group_name) + ":"; bool first = true; for (const auto& param : param_values) { // Separate each |param|. if (!first) params_arg += "/"; first = false; // Append each |param|. const std::string& name = param.first; const std::string& value = param.second; params_arg += EscapeValue(name) + "/" + EscapeValue(value); } command_line->AppendSwitchASCII(variations::switches::kForceFieldTrialParams, params_arg); } } // namespace testing } // namespace variations