Expose API on VariationsService to set the restrict mode param.

chrome/browser/metrics/variations/variations_service.h

 // Copyright (c) 2012 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.
 
 #ifndef CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_
 #define CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_
 
 #include <string>
 
 #include "base/compiler_specific.h"
@@ skipping 84 matching lines @@
   // TODO(rkaplow): Handle this and the similar event in metrics_service by
   // observing an 'OnAppEnterForeground' event instead of requiring the frontend
   // code to notify each service individually.
   void OnAppEnterForeground();
 
 #if defined(OS_WIN)
   // Starts syncing Google Update Variation IDs with the registry.
   void StartGoogleUpdateRegistrySync();
 #endif
 
+  // Sets the value of the "restrict" URL param to the variations service that
+  // should be used for variation seed requests. This takes precedence over any
+  // value coming from policy prefs. This should be called prior to any calls
+  // to |StartRepeatedVariationsSeedFetch|.
+  void SetRestrictMode(const std::string& restrict_mode);
+
   // Exposed for testing.
   void SetCreateTrialsFromSeedCalledForTesting(bool called);
 
   // Returns the variations server URL, which can vary if a command-line flag is
   // set and/or the variations restrict pref is set in |local_prefs|. Declared
   // static for test purposes.
-  static GURL GetVariationsServerURL(PrefService* local_prefs);
+  static GURL GetVariationsServerURL(PrefService* local_prefs,
+                                     const std::string& restrict_mode_override);
 
   // Exposed for testing.
   static std::string GetDefaultVariationsServerURLForTesting();
 
   // Register Variations related prefs in Local State.
   static void RegisterPrefs(PrefRegistrySimple* registry);
 
   // Register Variations related prefs in the Profile prefs.
   static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry);
 
@@ skipping 79 matching lines @@
 
   // Contains the scheduler instance that handles timing for requests to the
   // server. Initially NULL and instantiated when the initial fetch is
   // requested.
   scoped_ptr<VariationsRequestScheduler> request_scheduler_;
 
   // Contains the current seed request. Will only have a value while a request
   // is pending, and will be reset by |OnURLFetchComplete|.
   scoped_ptr<net::URLFetcher> pending_seed_request_;
 
-  // The URL to use for querying the Variations server.
+  // The value of the "restrict" URL param to the variations server that has
+  // been specified via |SetRestrictMode|. If empty, the URL param will be set
+  // based on policy prefs.
+  std::string restrict_mode_;
+
+  // The URL to use for querying the variations server.
   GURL variations_server_url_;
 
   // Tracks whether |CreateTrialsFromSeed| has been called, to ensure that
   // it gets called prior to |StartRepeatedVariationsSeedFetch|.
   bool create_trials_from_seed_called_;
 
   // Tracks whether the initial request to the variations server had completed.
   bool initial_request_completed_;
 
   // Helper class used to tell this service if it's allowed to make network
@@ skipping 14 matching lines @@
 #endif
 
   base::WeakPtrFactory<VariationsService> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(VariationsService);
 };
 
 }  // namespace chrome_variations
 
 #endif  // CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_