Add a scoper object for URLFetcher::Factory

chrome/common/net/gaia/gaia_oauth_client_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.
 //
 // A complete set of unit tests for GaiaOAuthClient.
 
 #include <string>
 
 #include "base/message_loop.h"
 #include "base/string_number_conversions.h"
 #include "base/string_util.h"
 #include "chrome/common/net/gaia/gaia_oauth_client.h"
 #include "chrome/common/net/http_return.h"
 #include "chrome/test/testing_profile.h"
-#include "content/common/test_url_fetcher_factory.h"
 #include "content/common/url_fetcher.h"
+#include "content/test/test_url_fetcher_factory.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/net_errors.h"
 #include "net/url_request/url_request_status.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using ::testing::_;
 
 namespace {
 // Responds as though OAuth returned from the server.
@@ skipping 35 matching lines @@
 
  private:
   int response_code_;
   int max_failure_count_;
   int current_failure_count_;
   GURL url_;
   std::string results_;
   DISALLOW_COPY_AND_ASSIGN(MockOAuthFetcher);
 };
 
-class MockOAuthFetcherFactory : public URLFetcher::Factory {
+class MockOAuthFetcherFactory : public URLFetcher::Factory,
+                                public ScopedURLFetcherFactory {
  public:
   MockOAuthFetcherFactory()
-      : response_code_(RC_REQUEST_OK) {}
+      : ScopedURLFetcherFactory(ALLOW_THIS_IN_INITIALIZER_LIST(this)),
+        response_code_(RC_REQUEST_OK) {
+  }
   ~MockOAuthFetcherFactory() {}
   virtual URLFetcher* CreateURLFetcher(
       int id,
       const GURL& url,
       URLFetcher::RequestType request_type,
       URLFetcher::Delegate* d) {
     return new MockOAuthFetcher(
         response_code_,
         max_failure_count_,
         url,
@@ skipping 59 matching lines @@
 TEST_F(GaiaOAuthClientTest, NetworkFailure) {
   int response_code = RC_INTERNAL_SERVER_ERROR;
 
   MockGaiaOAuthClientDelegate delegate;
   EXPECT_CALL(delegate, OnNetworkError(response_code))
       .Times(1);
 
   TestingProfile profile;
 
   MockOAuthFetcherFactory factory;
-  URLFetcher::set_factory(&factory);
   factory.set_response_code(response_code);
   factory.set_max_failure_count(4);
 
   OAuthClientInfo client_info;
   client_info.client_id = "test_client_id";
   client_info.client_secret = "test_client_secret";
   GaiaOAuthClient auth(kGaiaOAuth2Url,
                        profile_.GetRequestContext());
   auth.GetTokensFromAuthCode(client_info, "auth_code", 2, &delegate);
-  URLFetcher::set_factory(NULL);
 }
 
 TEST_F(GaiaOAuthClientTest, NetworkFailureRecover) {
   int response_code = RC_INTERNAL_SERVER_ERROR;
 
   MockGaiaOAuthClientDelegate delegate;
   EXPECT_CALL(delegate, OnGetTokensResponse(kTestRefreshToken, kTestAccessToken,
       kTestExpiresIn)).Times(1);
 
   TestingProfile profile;
 
   MockOAuthFetcherFactory factory;
-  URLFetcher::set_factory(&factory);
   factory.set_response_code(response_code);
   factory.set_max_failure_count(4);
   factory.set_results(kDummyGetTokensResult);
 
   OAuthClientInfo client_info;
   client_info.client_id = "test_client_id";
   client_info.client_secret = "test_client_secret";
   GaiaOAuthClient auth(kGaiaOAuth2Url,
                        profile_.GetRequestContext());
   auth.GetTokensFromAuthCode(client_info, "auth_code", -1, &delegate);
-  URLFetcher::set_factory(NULL);
 }
 
 TEST_F(GaiaOAuthClientTest, OAuthFailure) {
   int response_code = RC_BAD_REQUEST;
 
   MockGaiaOAuthClientDelegate delegate;
   EXPECT_CALL(delegate, OnOAuthError()).Times(1);
 
   TestingProfile profile;
 
   MockOAuthFetcherFactory factory;
-  URLFetcher::set_factory(&factory);
   factory.set_response_code(response_code);
   factory.set_max_failure_count(-1);
   factory.set_results(kDummyGetTokensResult);
 
   OAuthClientInfo client_info;
   client_info.client_id = "test_client_id";
   client_info.client_secret = "test_client_secret";
   GaiaOAuthClient auth(kGaiaOAuth2Url,
                        profile_.GetRequestContext());
   auth.GetTokensFromAuthCode(client_info, "auth_code", -1, &delegate);
-  URLFetcher::set_factory(NULL);
 }
 
 
 TEST_F(GaiaOAuthClientTest, GetTokensSuccess) {
   MockGaiaOAuthClientDelegate delegate;
   EXPECT_CALL(delegate, OnGetTokensResponse(kTestRefreshToken, kTestAccessToken,
       kTestExpiresIn)).Times(1);
 
   TestingProfile profile;
 
   MockOAuthFetcherFactory factory;
-  URLFetcher::set_factory(&factory);
   factory.set_results(kDummyGetTokensResult);
 
   OAuthClientInfo client_info;
   client_info.client_id = "test_client_id";
   client_info.client_secret = "test_client_secret";
   GaiaOAuthClient auth(kGaiaOAuth2Url,
                        profile_.GetRequestContext());
   auth.GetTokensFromAuthCode(client_info, "auth_code", -1, &delegate);
-  URLFetcher::set_factory(NULL);
 }
 
 TEST_F(GaiaOAuthClientTest, RefreshTokenSuccess) {
   MockGaiaOAuthClientDelegate delegate;
   EXPECT_CALL(delegate, OnRefreshTokenResponse(kTestAccessToken,
       kTestExpiresIn)).Times(1);
 
   TestingProfile profile;
 
   MockOAuthFetcherFactory factory;
-  URLFetcher::set_factory(&factory);
   factory.set_results(kDummyRefreshTokenResult);
 
   OAuthClientInfo client_info;
   client_info.client_id = "test_client_id";
   client_info.client_secret = "test_client_secret";
   GaiaOAuthClient auth(kGaiaOAuth2Url,
                        profile_.GetRequestContext());
   auth.GetTokensFromAuthCode(client_info, "auth_code", -1, &delegate);
-  URLFetcher::set_factory(NULL);
 }
 }  // namespace gaia