Make VariationsService simulate received seeds.

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 12 matching lines @@
 #include "chrome/browser/metrics/variations/variations_registry_syncer_win.h"
 #endif
 
 class PrefService;
 class PrefRegistrySimple;
 
 namespace user_prefs {
 class PrefRegistrySyncable;
 }
 
+namespace metrics {
+class MetricsStateManager;
+}
+
 namespace chrome_variations {
 
 class VariationsSeed;
 
 // Used to setup field trials based on stored variations seed data, and fetch
 // new seed data from the variations server.
 class VariationsService
     : public net::URLFetcherDelegate,
       public ResourceRequestAllowedNotifier::Observer {
  public:
@@ skipping 31 matching lines @@
 
   // 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);
 
-  // Factory method for creating a VariationsService.
-  static VariationsService* Create(PrefService* local_state);
+  // Factory method for creating a VariationsService. Does not take ownership of
+  // |state_manager|. Caller should ensure that |state_manager| is valid for the
+  // lifetime of this class.
+  static scoped_ptr<VariationsService> Create(
+      PrefService* local_state,
+      metrics::MetricsStateManager* state_manager);
 
   // Set the PrefService responsible for getting policy-related preferences,
   // such as the restrict parameter.
   void set_policy_pref_service(PrefService* service) {
     DCHECK(service);
     policy_pref_service_ = service;
   }
 
  protected:
   // Starts the fetching process once, where |OnURLFetchComplete| is called with
   // the response.
   virtual void DoActualFetch();
 
   // Stores the seed to prefs. Set as virtual and protected so that it can be
   // overridden by tests.
   virtual void StoreSeed(const std::string& seed_data,
                          const std::string& seed_signature,
                          const base::Time& date_fetched);
 
   // This constructor exists for injecting a mock notifier. It is meant for
-  // testing only. This instance will take ownership of |notifier|.
+  // testing only. This instance will take ownership of |notifier|. Does not
+  // take ownership of |state_manager|. Caller should ensure that
+  // |state_manager| is valid for the lifetime of this class.
   VariationsService(ResourceRequestAllowedNotifier* notifier,
-                    PrefService* local_state);
+                    PrefService* local_state,
+                    metrics::MetricsStateManager* state_manager);
 
  private:
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, DoNotFetchIfOffline);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, DoNotFetchIfOnlineToOnline);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, FetchOnReconnect);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, LoadSeed);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, StoreSeed);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedStoredWhenOKStatus);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedNotStoredWhenNonOKStatus);
   FRIEND_TEST_ALL_PREFIXES(VariationsServiceTest, SeedDateUpdatedOn304Status);
 
-  // Creates the VariationsService with the given |local_state| prefs service.
+  // Creates the VariationsService with the given |local_state| prefs service
+  // and |state_manager|. Does not take ownership of |state_manager|. Caller
+  // should ensure that |state_manager| is valid for the lifetime of this class.
   // Use the |Create| factory method to create a VariationsService.
-  explicit VariationsService(PrefService* local_state);
+  VariationsService(PrefService* local_state,
+                    metrics::MetricsStateManager* state_manager);
 
   // Checks if prerequisites for fetching the Variations seed are met, and if
   // so, performs the actual fetch using |DoActualFetch|.
   void FetchVariationsSeed();
 
   // net::URLFetcherDelegate implementation:
   virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE;
 
   // ResourceRequestAllowedNotifier::Observer implementation:
   virtual void OnResourceRequestsAllowed() OVERRIDE;
 
   // Record the time of the most recent successful fetch.
   void RecordLastFetchTime();
 
   // The pref service used to store persist the variations seed.
   PrefService* local_state_;
 
+  // Used for instantiating entropy providers for variations seed simulation.
+  // Weak pointer.
+  metrics::MetricsStateManager* state_manager_;
+
   // Used to obtain policy-related preferences. Depending on the platform, will
   // either be Local State or Profile prefs.
   PrefService* policy_pref_service_;
 
   VariationsSeedStore seed_store_;
 
   // 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_;
@@ skipping 24 matching lines @@
   // Helper that handles synchronizing Variations with the Registry.
   VariationsRegistrySyncer registry_syncer_;
 #endif
 
   DISALLOW_COPY_AND_ASSIGN(VariationsService);
 };
 
 }  // namespace chrome_variations
 
 #endif  // CHROME_BROWSER_METRICS_VARIATIONS_VARIATIONS_SERVICE_H_