Platform-specific prune state storage.

chrome/browser/safe_browsing/incident_reporting/incident_reporting_service_unittest.cc

 // Copyright 2014 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 "chrome/browser/safe_browsing/incident_reporting/incident_reporting_service.h"
 
 #include <map>
 #include <string>
 
 #include "base/bind.h"
@@ skipping 12 matching lines @@
 #include "chrome/browser/safe_browsing/incident_reporting/tracked_preference_incident.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/safe_browsing/csd.pb.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_pref_service_syncable.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/testing_profile_manager.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+#if defined(OS_WIN)
+#include "base/test/test_reg_util_win.h"
+#endif
+
 // A test fixture that sets up a test task runner and makes it the thread's
 // runner. The fixture implements a fake envrionment data collector and a fake
 // report uploader.
 class IncidentReportingServiceTest : public testing::Test {
  protected:
   // An IRS class that allows a test harness to provide a fake environment
   // collector and report uploader via callbacks.
   class TestIncidentReportingService
       : public safe_browsing::IncidentReportingService {
    public:
@@ skipping 130 matching lines @@
         on_delayed_analysis_action_(ON_DELAYED_ANALYSIS_NO_ACTION),
         upload_result_(safe_browsing::IncidentReportUploader::UPLOAD_SUCCESS),
         environment_collected_(),
         download_finder_created_(),
         download_finder_destroyed_(),
         uploader_destroyed_(),
         delayed_analysis_ran_() {}
 
   void SetUp() override {
     testing::Test::SetUp();
+#if defined(OS_WIN)
+    // Redirect HKCU so that the platform state store used by the test doesn't
+    // collide with existing Chrome installs or other tests running in parallel.
+    registry_override_manager_.OverrideRegistry(HKEY_CURRENT_USER);
+#endif
     ASSERT_TRUE(profile_manager_.SetUp());
   }
 
   // Sets the action to be taken by the test fixture when the service creates a
   // LastDownloadFinder.
   void SetCreateDownloadFinderAction(OnCreateDownloadFinderAction action) {
     on_create_download_finder_action_ = action;
   }
 
   // Creates and returns a profile (owned by the profile manager) with or
@@ skipping 271 matching lines @@
 
   void OnDownloadFinderDestroyed() { download_finder_destroyed_ = true; }
   void OnUploaderDestroyed() { uploader_destroyed_ = true; }
 
   void OnDelayedAnalysis(scoped_ptr<safe_browsing::IncidentReceiver> receiver) {
     delayed_analysis_ran_ = true;
     if (on_delayed_analysis_action_ == ON_DELAYED_ANALYSIS_ADD_INCIDENT)
       receiver->AddIncidentForProcess(MakeTestIncident(nullptr));
   }
 
+#if defined(OS_WIN)
+  registry_util::RegistryOverrideManager registry_override_manager_;
+#endif
+
   // A mapping of profile name to its corresponding properties.
   std::map<std::string, ProfileProperties> profile_properties_;
 };
 
 // static
 base::LazyInstance<base::ThreadLocalPointer<
     IncidentReportingServiceTest::TestIncidentReportingService> >::Leaky
     IncidentReportingServiceTest::TestIncidentReportingService::test_instance_ =
         LAZY_INSTANCE_INITIALIZER;
 
@@ skipping 581 matching lines @@
   ASSERT_FALSE(new_state->HasKey(omnibox_type));
   // But other data must be untouched.
   ASSERT_TRUE(new_state->HasKey(preference_type));
 }
 
 // Parallel uploads
 // Shutdown during processing
 // environment colection taking longer than incident delay timer
 // environment colection taking longer than incident delay timer, and then
 // another incident arriving