Forcibly clear worker ref counts on shutdown.

chrome/browser/chromeos/login/saml/saml_offline_signin_limiter_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/chromeos/login/saml/saml_offline_signin_limiter.h"
 
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/test/simple_test_clock.h"
 #include "base/test/test_simple_task_runner.h"
-#include "base/threading/thread_task_runner_handle.h"
 #include "base/time/clock.h"
 #include "chrome/browser/chromeos/login/saml/saml_offline_signin_limiter_factory.h"
 #include "chrome/browser/chromeos/login/users/mock_user_manager.h"
 #include "chrome/browser/chromeos/login/users/scoped_user_manager_enabler.h"
 #include "chrome/browser/profiles/profile.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/prefs/testing_pref_service.h"
+#include "content/public/test/test_browser_thread_bundle.h"
 #include "extensions/browser/quota_service.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::Mock;
 using testing::Return;
 using testing::Sequence;
 using testing::_;
 
 namespace chromeos {
@@ skipping 13 matching lines @@
 
   void DestroyLimiter();
   void CreateLimiter();
 
   void SetUpUserManager();
 
   const AccountId test_account_id_ = AccountId::FromUserEmail(kTestUser);
 
   TestingPrefServiceSimple* GetTestingLocalState();
 
+  content::TestBrowserThreadBundle thread_bundle_;
   extensions::QuotaService::ScopedDisablePurgeForTesting
       disable_purge_for_testing_;
 
   scoped_refptr<base::TestSimpleTaskRunner> runner_;
-  base::ThreadTaskRunnerHandle runner_handle_;
 
   MockUserManager* user_manager_;  // Not owned.
   ScopedUserManagerEnabler user_manager_enabler_;
 
   std::unique_ptr<TestingProfile> profile_;
   base::SimpleTestClock clock_;
 
   SAMLOfflineSigninLimiter* limiter_;  // Owned.
 
   TestingPrefServiceSimple testing_local_state_;
 
   DISALLOW_COPY_AND_ASSIGN(SAMLOfflineSigninLimiterTest);
 };
 
 SAMLOfflineSigninLimiterTest::SAMLOfflineSigninLimiterTest()
     : runner_(new base::TestSimpleTaskRunner),
-      runner_handle_(runner_),
       user_manager_(new MockUserManager),
       user_manager_enabler_(user_manager_),
       limiter_(NULL) {}
 
 SAMLOfflineSigninLimiterTest::~SAMLOfflineSigninLimiterTest() {
   DestroyLimiter();
   Mock::VerifyAndClearExpectations(user_manager_);
   EXPECT_CALL(*user_manager_, Shutdown()).Times(1);
   EXPECT_CALL(*user_manager_, RemoveSessionStateObserver(_)).Times(1);
   profile_.reset();
@@ skipping 12 matching lines @@
   DestroyLimiter();
   limiter_ = new SAMLOfflineSigninLimiter(profile_.get(), &clock_);
 }
 
 void SAMLOfflineSigninLimiterTest::SetUpUserManager() {
   EXPECT_CALL(*user_manager_, GetLocalState())
       .WillRepeatedly(Return(GetTestingLocalState()));
 }
 
 void SAMLOfflineSigninLimiterTest::SetUp() {
+  base::MessageLoop::current()->SetTaskRunner(runner_);
   profile_.reset(new TestingProfile);
 
   SAMLOfflineSigninLimiterFactory::SetClockForTesting(&clock_);
   user_manager_->AddUser(test_account_id_);
   profile_->set_profile_name(kTestUser);
   clock_.Advance(base::TimeDelta::FromHours(1));
 
   user_manager_->RegisterPrefs(GetTestingLocalState()->registry());
   SetUpUserManager();
 }
@@ skipping 532 matching lines @@
   EXPECT_CALL(*user_manager_, SaveForceOnlineSignin(test_account_id_, true))
       .Times(1);
   limiter_->SignedIn(UserContext::AUTH_FLOW_OFFLINE);
 
   const base::Time last_gaia_signin_time = base::Time::FromInternalValue(
       prefs->GetInt64(prefs::kSAMLLastGAIASignInTime));
   EXPECT_EQ(gaia_signin_time, last_gaia_signin_time);
 }
 
 } //  namespace chromeos