[net/http auth] Use strings to identify authentication schemes.

net/http/http_auth_handler_mock.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.
 
 #include "net/http/http_auth_handler_mock.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string_util.h"
 #include "base/thread_task_runner_handle.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_auth_challenge_tokenizer.h"
 #include "net/http/http_request_info.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace net {
 
 HttpAuthHandlerMock::HttpAuthHandlerMock()
-  : resolve_(RESOLVE_INIT),
-    generate_async_(false),
-    generate_rv_(OK),
-    auth_token_(NULL),
-    first_round_(true),
-    connection_based_(false),
-    allows_default_credentials_(false),
-    allows_explicit_credentials_(true),
-    weak_factory_(this) {
-}
+    : generate_rv_(OK), weak_factory_(this) {}
 
 HttpAuthHandlerMock::~HttpAuthHandlerMock() {
 }
 
 void HttpAuthHandlerMock::SetResolveExpectation(Resolve resolve) {
   EXPECT_EQ(RESOLVE_INIT, resolve_);
   resolve_ = resolve;
 }
 
 bool HttpAuthHandlerMock::NeedsCanonicalName() {
@@ skipping 33 matching lines @@
   return rv;
 }
 
 void HttpAuthHandlerMock::SetGenerateExpectation(bool async, int rv) {
   generate_async_ = async;
   generate_rv_ = rv;
 }
 
 HttpAuth::AuthorizationResult HttpAuthHandlerMock::HandleAnotherChallenge(
     HttpAuthChallengeTokenizer* challenge) {
-  // If we receive an empty challenge for a connection based scheme, or a second
-  // challenge for a non connection based scheme, assume it's a rejection.
-  if (!is_connection_based() || challenge->base64_param().empty())
+  // If we receive a second challenge for a regular scheme, assume it's a
+  // rejection. Receiving an empty second challenge when expecting multiple
+  // rounds is also considered a rejection.
+  if (!expect_multiple_challenges() || challenge->base64_param().empty())
     return HttpAuth::AUTHORIZATION_RESULT_REJECT;
-  if (!base::LowerCaseEqualsASCII(challenge->scheme(), "mock"))
+  if (!challenge->SchemeIs("mock"))
     return HttpAuth::AUTHORIZATION_RESULT_INVALID;
   return HttpAuth::AUTHORIZATION_RESULT_ACCEPT;
 }
 
 bool HttpAuthHandlerMock::NeedsIdentity() {
   return first_round_;
 }
 
 bool HttpAuthHandlerMock::AllowsDefaultCredentials() {
   return allows_default_credentials_;
 }
 
 bool HttpAuthHandlerMock::AllowsExplicitCredentials() {
   return allows_explicit_credentials_;
 }
 
 bool HttpAuthHandlerMock::Init(HttpAuthChallengeTokenizer* challenge) {
-  auth_scheme_ = HttpAuth::AUTH_SCHEME_MOCK;
-  score_ = 1;
-  properties_ = connection_based_ ? IS_CONNECTION_BASED : 0;
+  auth_scheme_ = "mock";
   return true;
 }
 
 int HttpAuthHandlerMock::GenerateAuthTokenImpl(
     const AuthCredentials* credentials,
     const HttpRequestInfo* request,
     const CompletionCallback& callback,
     std::string* auth_token) {
   first_round_ = false;
   request_url_ = request->url;
@@ skipping 39 matching lines @@
 }
 
 HttpAuthHandlerMock::Factory::~Factory() {
 }
 
 void HttpAuthHandlerMock::Factory::AddMockHandler(
     HttpAuthHandler* handler, HttpAuth::Target target) {
   handlers_[target].push_back(handler);
 }
 
+bool HttpAuthHandlerMock::Factory::HaveAuthHandlers(
+    HttpAuth::Target target) const {
+  return !handlers_[target].empty();
+}
+
 int HttpAuthHandlerMock::Factory::CreateAuthHandler(
     HttpAuthChallengeTokenizer* challenge,
     HttpAuth::Target target,
     const GURL& origin,
     CreateReason reason,
     int nonce_count,
     const BoundNetLog& net_log,
     scoped_ptr<HttpAuthHandler>* handler) {
   if (handlers_[target].empty())
     return ERR_UNEXPECTED;
   scoped_ptr<HttpAuthHandler> tmp_handler(handlers_[target][0]);
   std::vector<HttpAuthHandler*>& handlers = handlers_[target].get();
   handlers.erase(handlers.begin());
   if (do_init_from_challenge_ &&
       !tmp_handler->InitFromChallenge(challenge, target, origin, net_log))
     return ERR_INVALID_RESPONSE;
   handler->swap(tmp_handler);
   return OK;
 }
 
 }  // namespace net