Add simpler API for changing parameters in tests.

components/variations/variations_associated_data.cc

 // Copyright 2013 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_associated_data.h"
 
 #include <map>
 #include <utility>
 #include <vector>
 
 #include "base/feature_list.h"
 #include "base/macros.h"
 #include "base/memory/singleton.h"
 
 namespace variations {
 
 namespace {
 
+const char kGroupTesting[] = "Testing";
+
 // The internal singleton accessor for the map, used to keep it thread-safe.
 class GroupMapAccessor {
  public:
   typedef std::map<ActiveGroupId, VariationID, ActiveGroupIdCompare>
       GroupToIDMap;
 
   // Retrieve the singleton.
   static GroupMapAccessor* GetInstance() {
     return base::Singleton<GroupMapAccessor>::get();
   }
@@ skipping 216 matching lines @@
   if (!trial)
     return std::string();
 
   return GetVariationParamValue(trial->trial_name(), param_name);
 }
 
 // Functions below are exposed for testing explicitly behind this namespace.
 // They simply wrap existing functions in this file.
 namespace testing {
 
+VariationParamsManager::VariationParamsManager(
+    const std::string& trial_name,
+    const std::map<std::string, std::string>& params) {
+  SetVariationParams(trial_name, params);
+}
+
+VariationParamsManager::~VariationParamsManager() {
+  ClearAllVariationIDs();
+  ClearAllVariationParams();
+  field_trial_list_.reset();
+}
+
+void VariationParamsManager::SetVariationParams(
+    const std::string& trial_name,
+    const std::map<std::string, std::string>& params) {
+  field_trial_list_.reset(new base::FieldTrialList(nullptr));
+  variations::AssociateVariationParams(trial_name, kGroupTesting, params);
+  base::FieldTrialList::CreateFieldTrial(trial_name, kGroupTesting);
+}
+
 void ClearAllVariationIDs() {
   GroupMapAccessor::GetInstance()->ClearAllMapsForTesting();
 }
 
 void ClearAllVariationParams() {
   VariationsParamAssociator::GetInstance()->ClearAllParamsForTesting();
 }
 
 }  // namespace testing
 
 }  // namespace variations