Forcibly clear worker ref counts on shutdown.

chrome/browser/safe_browsing/incident_reporting/state_store_unittest.cc

 // Copyright 2015 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/state_store.h"
 
 #include <stdint.h>
 
 #include "base/files/scoped_temp_dir.h"
 #include "base/json/json_file_value_serializer.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/test_simple_task_runner.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/values.h"
 #include "build/build_config.h"
 #include "chrome/browser/prefs/browser_prefs.h"
 #include "chrome/browser/safe_browsing/incident_reporting/incident.h"
 #include "chrome/browser/safe_browsing/incident_reporting/platform_state_store.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/testing_profile_manager.h"
 #include "components/pref_registry/pref_registry_syncable.h"
 #include "components/syncable_prefs/pref_service_syncable.h"
 #include "components/syncable_prefs/pref_service_syncable_factory.h"
+#include "content/public/test/test_browser_thread_bundle.h"
 #include "extensions/browser/quota_service.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #if defined(OS_WIN)
 #include "base/test/test_reg_util_win.h"
 #endif
 
 namespace safe_browsing {
 
 #if defined(OS_WIN)
@@ skipping 28 matching lines @@
  protected:
   struct TestData {
     IncidentType type;
     const char* key;
     uint32_t digest;
   };
 
   StateStoreTest()
       : profile_(nullptr),
         task_runner_(new base::TestSimpleTaskRunner()),
-        thread_task_runner_handle_(task_runner_),
         profile_manager_(TestingBrowserProcess::GetGlobal()) {}
 
   void SetUp() override {
     PlatformStateStoreTestBase::SetUp();
+    base::MessageLoop::current()->SetTaskRunner(task_runner_);
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
     ASSERT_TRUE(profile_manager_.SetUp());
     CreateProfile();
   }
 
   void DeleteProfile() {
     if (profile_) {
       profile_ = nullptr;
       profile_manager_.DeleteTestingProfile(kProfileName_);
     }
@@ skipping 21 matching lines @@
         new user_prefs::PrefRegistrySyncable();
     chrome::RegisterUserProfilePrefs(pref_registry);
     profile_ = profile_manager_.CreateTestingProfile(
         kProfileName_, factory.CreateSyncable(pref_registry),
         base::UTF8ToUTF16(kProfileName_), 0, std::string(),
         TestingProfile::TestingFactories());
   }
 
   static const char kProfileName_[];
   static const TestData kTestData_[];
+  content::TestBrowserThreadBundle thread_bundle_;
   TestingProfile* profile_;
   scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
 
  private:
   base::FilePath GetPrefsPath() {
     return temp_dir_.GetPath().AppendASCII("prefs");
   }
 
   extensions::QuotaService::ScopedDisablePurgeForTesting
       disable_purge_for_testing_;
   base::ScopedTempDir temp_dir_;
-  base::ThreadTaskRunnerHandle thread_task_runner_handle_;
   TestingProfileManager profile_manager_;
 
   DISALLOW_COPY_AND_ASSIGN(StateStoreTest);
 };
 
 // static
 const char StateStoreTest::kProfileName_[] = "test_profile";
 const StateStoreTest::TestData StateStoreTest::kTestData_[] = {
     {IncidentType::TRACKED_PREFERENCE, "tp_one", 1},
     {IncidentType::TRACKED_PREFERENCE, "tp_two", 2},
@@ skipping 129 matching lines @@
     // Verify that the state survived.
     ASSERT_TRUE(state_store.HasBeenReported(data.type, data.key, data.digest));
 #else
     // Verify that the state did not survive.
     ASSERT_FALSE(state_store.HasBeenReported(data.type, data.key, data.digest));
 #endif
   }
 }
 
 }  // namespace safe_browsing