Forcibly clear worker ref counts on shutdown.

chrome/browser/notifications/extension_welcome_notification_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/notifications/extension_welcome_notification.h"
 
 #include <stdint.h>
 
 #include <memory>
 #include <string>
 
 #include "base/macros.h"
 #include "base/message_loop/message_loop.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 "chrome/browser/notifications/notification.h"
 #include "chrome/browser/prefs/pref_service_syncable_util.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_browser_process.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/pref_registry/pref_registry_syncable.h"
 #include "components/prefs/pref_service.h"
 #include "components/sync/api/fake_sync_change_processor.h"
 #include "components/sync/api/sync_error_factory_mock.h"
 #include "components/syncable_prefs/testing_pref_service_syncable.h"
+#include "content/public/test/test_browser_thread_bundle.h"
 #include "extensions/browser/quota_service.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/message_center/fake_message_center.h"
 #include "ui/message_center/notification.h"
 
 class MockMessageCenter : public message_center::FakeMessageCenter {
  public:
   MockMessageCenter()
       : add_notification_calls_(0),
         remove_notification_calls_(0),
@@ skipping 92 matching lines @@
 class ExtensionWelcomeNotificationTest : public testing::Test {
  protected:
   ExtensionWelcomeNotificationTest() {
     scoped_refptr<user_prefs::PrefRegistrySyncable> pref_registry(
         new user_prefs::PrefRegistrySyncable());
     ExtensionWelcomeNotification::RegisterProfilePrefs(pref_registry.get());
   }
 
   void SetUp() override {
     task_runner_ = new base::TestSimpleTaskRunner();
-    thread_task_runner_handle_.reset(
-        new base::ThreadTaskRunnerHandle(task_runner_));
+    base::MessageLoop::current()->SetTaskRunner(task_runner_);
     profile_.reset(new TestingProfile());
     delegate_ = new WelcomeNotificationDelegate();
     welcome_notification_.reset(
         ExtensionWelcomeNotification::Create(profile_.get(), delegate_));
   }
 
   void TearDown() override {
     delegate_ = NULL;
     welcome_notification_.reset();
     profile_.reset();
     TestingBrowserProcess::DeleteInstance();
-    thread_task_runner_handle_.reset();
     task_runner_ = NULL;
   }
 
   void StartPreferenceSyncing() const {
     PrefServiceSyncableFromProfile(profile_.get())
         ->GetSyncableService(syncer::PREFERENCES)
         ->MergeDataAndStartSyncing(syncer::PREFERENCES, syncer::SyncDataList(),
                                    std::unique_ptr<syncer::SyncChangeProcessor>(
                                        new syncer::FakeSyncChangeProcessor),
                                    std::unique_ptr<syncer::SyncErrorFactory>(
@@ skipping 64 matching lines @@
     rich_notification_data.priority = 0;
     Notification notification(
         message_center::NOTIFICATION_TYPE_BASE_FORMAT,
         base::UTF8ToUTF16("Title"), base::UTF8ToUTF16("Body"), gfx::Image(),
         notifier_id, base::UTF8ToUTF16("Source"), GURL("http://tests.url"),
         notification_id, rich_notification_data,
         new TestNotificationDelegate("TestNotification"));
     welcome_notification_->ShowWelcomeNotificationIfNecessary(notification);
   }
 
+  content::TestBrowserThreadBundle thread_bundle_;
   extensions::QuotaService::ScopedDisablePurgeForTesting
       disable_purge_for_testing_;
   scoped_refptr<base::TestSimpleTaskRunner> task_runner_;
-  std::unique_ptr<base::ThreadTaskRunnerHandle> thread_task_runner_handle_;
   std::unique_ptr<TestingProfile> profile_;
   // Weak Ref owned by welcome_notification_
   WelcomeNotificationDelegate* delegate_;
   std::unique_ptr<ExtensionWelcomeNotification> welcome_notification_;
 
   DISALLOW_COPY_AND_ASSIGN(ExtensionWelcomeNotificationTest);
 };
 
 // Show a regular notification. Expect that WelcomeNotification will
 // not show a welcome notification.
@@ skipping 279 matching lines @@
 
   EXPECT_TRUE(task_runner()->GetPendingTasks().empty());
   EXPECT_EQ(message_center()->add_notification_calls(), 0);
   EXPECT_EQ(message_center()->remove_notification_calls(), 0);
   EXPECT_EQ(message_center()->notifications_with_shown_as_popup(), 0);
   EXPECT_FALSE(GetBooleanPref(prefs::kWelcomeNotificationDismissed));
   EXPECT_TRUE(GetBooleanPref(prefs::kWelcomeNotificationDismissedLocal));
   EXPECT_TRUE(GetBooleanPref(prefs::kWelcomeNotificationPreviouslyPoppedUp));
   EXPECT_EQ(GetInt64Pref(prefs::kWelcomeNotificationExpirationTimestamp), 1);
 }