Add a scoper object for URLFetcher::Factory

chrome/browser/net/gaia/token_service_unittest.cc

 // Copyright (c) 2011 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.
 //
 // This file defines a unit test for the profile's token service.
 
 #include "chrome/browser/net/gaia/token_service_unittest.h"
 
 #include "base/command_line.h"
 #include "base/synchronization/waitable_event.h"
 #include "chrome/browser/password_manager/encryptor.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/net/gaia/gaia_auth_fetcher_unittest.h"
 #include "chrome/common/net/gaia/gaia_constants.h"
-#include "content/common/test_url_fetcher_factory.h"
+#include "content/test/test_url_fetcher_factory.h"
 
 TokenAvailableTracker::TokenAvailableTracker() {}
 
 TokenAvailableTracker::~TokenAvailableTracker() {}
 
 void TokenAvailableTracker::Observe(int type,
                                     const NotificationSource& source,
                                     const NotificationDetails& details) {
   TestNotificationTracker::Observe(type, source, details);
   if (type == chrome::NOTIFICATION_TOKEN_AVAILABLE) {
@@ skipping 37 matching lines @@
   profile_.reset(new TestingProfile());
   profile_->CreateWebDataService(false);
   WaitForDBLoadCompletion();
 
   success_tracker_.ListenFor(chrome::NOTIFICATION_TOKEN_AVAILABLE,
                              Source<TokenService>(&service_));
   failure_tracker_.ListenFor(chrome::NOTIFICATION_TOKEN_REQUEST_FAILED,
                              Source<TokenService>(&service_));
 
   service_.Initialize("test", profile_.get());
-
-  URLFetcher::set_factory(NULL);
 }
 
 void TokenServiceTestHarness::TearDown() {
   // You have to destroy the profile before the db_thread_ stops.
   if (profile_.get()) {
     profile_.reset(NULL);
   }
 
   db_thread_.Stop();
   MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask);
@@ skipping 98 matching lines @@
   EXPECT_TRUE(service_.HasLsid());
 
   service_.ResetCredentialsInMemory();
 
   EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kSyncService));
   EXPECT_FALSE(service_.HasLsid());
 }
 
 TEST_F(TokenServiceTest, ResetComplex) {
   TestURLFetcherFactory factory;
-  URLFetcher::set_factory(&factory);
   service_.StartFetchingTokens();
   // You have to call delegates by hand with the test fetcher,
   // Let's pretend only one returned.
 
   service_.OnIssueAuthTokenSuccess(GaiaConstants::kSyncService, "eraseme");
   EXPECT_TRUE(service_.HasTokenForService(GaiaConstants::kSyncService));
   EXPECT_EQ(service_.GetTokenForService(GaiaConstants::kSyncService),
             "eraseme");
   EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kTalkService));
 
   service_.ResetCredentialsInMemory();
   EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kSyncService));
   EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kTalkService));
   EXPECT_FALSE(service_.HasLsid());
 
   // Now start using it again.
   service_.UpdateCredentials(credentials_);
   EXPECT_TRUE(service_.HasLsid());
   service_.StartFetchingTokens();
 
   service_.OnIssueAuthTokenSuccess(GaiaConstants::kSyncService, "token");
   service_.OnIssueAuthTokenSuccess(GaiaConstants::kTalkService, "token2");
 
   EXPECT_EQ(service_.GetTokenForService(GaiaConstants::kSyncService), "token");
   EXPECT_EQ(service_.GetTokenForService(GaiaConstants::kTalkService), "token2");
 }
 
 TEST_F(TokenServiceTest, FullIntegration) {
-  MockFactory<MockFetcher> factory;
   std::string result = "SID=sid\nLSID=lsid\nAuth=auth\n";
-  factory.set_results(result);
-  URLFetcher::set_factory(&factory);
-  EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kSyncService));
-  EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kTalkService));
-  service_.StartFetchingTokens();
-  URLFetcher::set_factory(NULL);
+
+  {
+    MockFactory<MockFetcher> factory;
+    factory.set_results(result);
+    EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kSyncService));
+    EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kTalkService));
+    service_.StartFetchingTokens();
+  }
 
   EXPECT_TRUE(service_.HasTokenForService(GaiaConstants::kSyncService));
   EXPECT_TRUE(service_.HasTokenForService(GaiaConstants::kTalkService));
   // Gaia returns the entire result as the token so while this is a shared
   // result with ClientLogin, it doesn't matter, we should still get it back.
   EXPECT_EQ(service_.GetTokenForService(GaiaConstants::kSyncService), result);
   EXPECT_EQ(service_.GetTokenForService(GaiaConstants::kTalkService), result);
 
   service_.ResetCredentialsInMemory();
   EXPECT_FALSE(service_.HasTokenForService(GaiaConstants::kSyncService));
@@ skipping 120 matching lines @@
 
     *CommandLine::ForCurrentProcess() = original_cl;
   }
 };
 
 TEST_F(TokenServiceCommandLineTest, TestValueOverride) {
   EXPECT_TRUE(service_.HasTokenForService("my_service"));
   EXPECT_EQ("my_value", service_.GetTokenForService("my_service"));
 }
 #endif   // ifndef NDEBUG