[net/auth] Discard current handler token generation fails.

net/http/http_auth_handler_mock.h

 // 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.
 
 #ifndef NET_HTTP_HTTP_AUTH_HANDLER_MOCK_H_
 #define NET_HTTP_HTTP_AUTH_HANDLER_MOCK_H_
 
 #include <memory>
+#include <ostream>
 #include <string>
 #include <vector>
 
 #include "base/memory/weak_ptr.h"
 #include "net/http/http_auth_handler.h"
 #include "net/http/http_auth_handler_factory.h"
 #include "url/gurl.h"
 
 namespace net {
 
 class HostResolver;
 
 // MockAuthHandler is used in tests to reliably trigger edge cases.
 class HttpAuthHandlerMock : public HttpAuthHandler {
  public:
+  enum class State {
+    WAIT_FOR_INIT,
+    WAIT_FOR_CHALLENGE,
+    WAIT_FOR_GENERATE_AUTH_TOKEN,
+    TOKEN_PENDING,
+    DONE
+  };
+
   enum Resolve {
     RESOLVE_INIT,
     RESOLVE_SKIP,
     RESOLVE_SYNC,
     RESOLVE_ASYNC,
     RESOLVE_TESTED,
   };
 
   // The Factory class returns handlers in the order they were added via
   // AddMockHandler.
@@ skipping 47 matching lines @@
   }
 
   void set_allows_explicit_credentials(bool allows_explicit_credentials) {
     allows_explicit_credentials_ = allows_explicit_credentials;
   }
 
   const GURL& request_url() const {
     return request_url_;
   }
 
+  State state() const { return state_; }
+
   // HttpAuthHandler:
   HttpAuth::AuthorizationResult HandleAnotherChallenge(
       HttpAuthChallengeTokenizer* challenge) override;
   bool NeedsIdentity() override;
   bool AllowsDefaultCredentials() override;
   bool AllowsExplicitCredentials() override;
 
  protected:
   bool Init(HttpAuthChallengeTokenizer* challenge,
             const SSLInfo& ssl_info) override;
 
   int GenerateAuthTokenImpl(const AuthCredentials* credentials,
                             const HttpRequestInfo* request,
                             const CompletionCallback& callback,
                             std::string* auth_token) override;
 
  private:
   void OnResolveCanonicalName();
 
   void OnGenerateAuthToken();
 
+  State state_;
   Resolve resolve_;
   CompletionCallback callback_;
   bool generate_async_;
   int generate_rv_;
   std::string* auth_token_;
   bool first_round_;
   bool connection_based_;
   bool allows_default_credentials_;
   bool allows_explicit_credentials_;
   GURL request_url_;
   base::WeakPtrFactory<HttpAuthHandlerMock> weak_factory_;
 };
 
+void PrintTo(const HttpAuthHandlerMock::State& state, ::std::ostream* os);
+
 }  // namespace net
 
 #endif  // NET_HTTP_HTTP_AUTH_HANDLER_MOCK_H_