Reset pending request type on network error

google_apis/gaia/gaia_oauth_client_unittest.cc

 // Copyright (c) 2012 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 <vector>
 
 #include "base/json/json_reader.h"
@@ skipping 281 matching lines @@
   EXPECT_CALL(delegate, OnGetTokensResponse(kTestRefreshToken, kTestAccessToken,
       kTestExpiresIn)).Times(1);
 
   MockOAuthFetcherFactory factory;
   factory.set_results(kDummyGetTokensResult);
 
   GaiaOAuthClient auth(GetRequestContext());
   auth.GetTokensFromAuthCode(client_info_, "auth_code", -1, &delegate);
 }
 
+TEST_F(GaiaOAuthClientTest, GetTokensAfterNetworkFailure) {
+  int response_code = net::HTTP_INTERNAL_SERVER_ERROR;
+
+  MockGaiaOAuthClientDelegate failure_delegate;
+  EXPECT_CALL(failure_delegate, OnNetworkError(response_code)).Times(1);
+
+  MockGaiaOAuthClientDelegate success_delegate;
+  EXPECT_CALL(success_delegate, OnGetTokensResponse(kTestRefreshToken,
+      kTestAccessToken, kTestExpiresIn)).Times(1);
+
+  MockOAuthFetcherFactory factory;
+  factory.set_response_code(response_code);
+  factory.set_max_failure_count(4);
+  factory.set_results(kDummyGetTokensResult);
+
+  GaiaOAuthClient auth(GetRequestContext());
+  auth.GetTokensFromAuthCode(client_info_, "auth_code", 2, &failure_delegate);
+  auth.GetTokensFromAuthCode(client_info_, "auth_code", -1, &success_delegate);
+}
+
 TEST_F(GaiaOAuthClientTest, RefreshTokenSuccess) {
   MockGaiaOAuthClientDelegate delegate;
   EXPECT_CALL(delegate, OnRefreshTokenResponse(kTestAccessToken,
       kTestExpiresIn)).Times(1);
 
   MockOAuthFetcherFactory factory;
   factory.set_results(kDummyRefreshTokenResult);
   factory.set_complete_immediately(false);
 
   GaiaOAuthClient auth(GetRequestContext());
@@ skipping 97 matching lines @@
 
   GaiaOAuthClient auth(GetRequestContext());
   auth.GetTokenHandleInfo("some_handle", 1, &delegate);
 
   std::string audience;
   ASSERT_TRUE(captured_result->GetString("audience", &audience));
   ASSERT_EQ("1234567890.apps.googleusercontent.com", audience);
 }
 
 }  // namespace gaia