[net/http auth] Make HttpAuthHandler challenge handling asynchronous.

net/http/http_auth_handler_basic_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.
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "net/base/net_errors.h"
 #include "net/base/test_completion_callback.h"
 #include "net/http/http_auth_challenge_tokenizer.h"
 #include "net/http/http_auth_handler_basic.h"
 #include "net/http/http_request_info.h"
+#include "net/http/http_response_info.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace net {
 
 TEST(HttpAuthHandlerBasicTest, GenerateAuthToken) {
   static const struct {
     const char* username;
     const char* password;
     const char* expected_credentials;
   } tests[] = {
     { "foo", "bar", "Basic Zm9vOmJhcg==" },
     // Empty username
     { "", "foobar", "Basic OmZvb2Jhcg==" },
     // Empty password
     { "anon", "", "Basic YW5vbjo=" },
     // Empty username and empty password.
     { "", "", "Basic Og==" },
   };
   GURL origin("http://www.example.com");
   HttpAuthHandlerBasic::Factory factory;
   for (size_t i = 0; i < arraysize(tests); ++i) {
     std::string challenge = "Basic realm=\"Atlantis\"";
     scoped_ptr<HttpAuthHandler> basic =
         factory.CreateAuthHandlerForScheme("basic");
     ASSERT_TRUE(basic);
     HttpAuthChallengeTokenizer tokenizer(challenge.begin(), challenge.end());
-    EXPECT_EQ(OK, basic->HandleInitialChallenge(
-                      tokenizer, HttpAuth::AUTH_SERVER, origin, BoundNetLog()));
+    HttpResponseInfo response_info;
+    TestCompletionCallback callback;
+    int rv = basic->HandleInitialChallenge(tokenizer, response_info,
+                                           HttpAuth::AUTH_SERVER, origin,
+                                           BoundNetLog(), callback.callback());
+    EXPECT_EQ(OK, callback.GetResult(rv));
     AuthCredentials credentials(base::ASCIIToUTF16(tests[i].username),
                                 base::ASCIIToUTF16(tests[i].password));
     HttpRequestInfo request_info;
     std::string auth_token;
-    TestCompletionCallback callback;
-    int rv = basic->GenerateAuthToken(&credentials, request_info,
-                                      callback.callback(), &auth_token);
-    EXPECT_EQ(OK, rv);
+    rv = basic->GenerateAuthToken(&credentials, request_info,
+                                  callback.callback(), &auth_token);
+    EXPECT_EQ(OK, callback.GetResult(rv));
     EXPECT_STREQ(tests[i].expected_credentials, auth_token.c_str());
   }
 }
 
 TEST(HttpAuthHandlerBasicTest, HandleAnotherChallenge) {
   static const struct {
     const char* challenge;
     HttpAuth::AuthorizationResult expected_rv;
   } tests[] = {
     // The handler is initialized using this challenge.  The first
@@ skipping 26 matching lines @@
     }
   };
 
   GURL origin("http://www.example.com");
   HttpAuthHandlerBasic::Factory factory;
   scoped_ptr<HttpAuthHandler> basic =
       factory.CreateAuthHandlerForScheme("basic");
   std::string initial_challenge(tests[0].challenge);
   HttpAuthChallengeTokenizer tokenizer(initial_challenge.begin(),
                                        initial_challenge.end());
-  EXPECT_EQ(OK, basic->HandleInitialChallenge(tokenizer, HttpAuth::AUTH_SERVER,
-                                              origin, BoundNetLog()));
+  HttpResponseInfo response_info;
+  TestCompletionCallback callback;
+  int rv = basic->HandleInitialChallenge(tokenizer, response_info,
+                                         HttpAuth::AUTH_SERVER, origin,
+                                         BoundNetLog(), callback.callback());
+  EXPECT_EQ(OK, callback.GetResult(rv));
 
   for (size_t i = 0; i < arraysize(tests); ++i) {
     std::string challenge(tests[i].challenge);
     HttpAuthChallengeTokenizer tok(challenge.begin(),
                                    challenge.end());
     EXPECT_EQ(tests[i].expected_rv, basic->HandleAnotherChallenge(tok));
   }
 }
 
 TEST(HttpAuthHandlerBasicTest, HandleInitialChallenge) {
@@ skipping 72 matching lines @@
       "foo-\xC3\xA5",
     },
   };
   HttpAuthHandlerBasic::Factory factory;
   GURL origin("http://www.example.com");
   for (size_t i = 0; i < arraysize(tests); ++i) {
     std::string challenge = tests[i].challenge;
     scoped_ptr<HttpAuthHandler> basic =
         factory.CreateAuthHandlerForScheme("basic");
     HttpAuthChallengeTokenizer tokenizer(challenge.begin(), challenge.end());
-    int rv = basic->HandleInitialChallenge(tokenizer, HttpAuth::AUTH_SERVER,
-                                           origin, BoundNetLog());
-    EXPECT_EQ(tests[i].expected_rv, rv);
+    HttpResponseInfo response_info;
+    TestCompletionCallback callback;
+    int rv = basic->HandleInitialChallenge(tokenizer, response_info,
+                                           HttpAuth::AUTH_SERVER, origin,
+                                           BoundNetLog(), callback.callback());
+    EXPECT_EQ(tests[i].expected_rv, callback.GetResult(rv));
     if (rv == OK)
       EXPECT_EQ(tests[i].expected_realm, basic->realm());
   }
 }
 
 TEST(HttpAuthHandlerBasicTest, CreateAuthHandlerForScheme) {
   HttpAuthHandlerBasic::Factory basic_factory;
 
   EXPECT_TRUE(basic_factory.CreateAuthHandlerForScheme("basic"));
   EXPECT_FALSE(basic_factory.CreateAuthHandlerForScheme("bogus"));
 }
 
 TEST(HttpAuthHandlerBasicTest, CreateAndInitPreemptiveAuthHandler_Valid) {
   HttpAuthHandlerBasic::Factory digest_factory;
   HttpAuthCache auth_cache;
   std::string challenge("basic realm=\"Foo\"");
-  HttpAuthChallengeTokenizer tokenizer(challenge.begin(), challenge.end());
 
   HttpAuthCache::Entry* entry =
       auth_cache.Add(GURL("http://example.com/foo").GetOrigin(), "foo", "basic",
                      challenge, AuthCredentials(), "/foo");
   EXPECT_TRUE(digest_factory.CreateAndInitPreemptiveAuthHandler(
-      entry, tokenizer, HttpAuth::AUTH_SERVER, BoundNetLog()));
+      entry, HttpAuth::AUTH_SERVER, BoundNetLog()));
 }
 
 TEST(HttpAuthHandlerBasicTest, CreateAndInitPreemptiveAuthHandler_Invalid) {
   HttpAuthHandlerBasic::Factory digest_factory;
   HttpAuthCache auth_cache;
   std::string challenge("digest realm=\"foo\"");
-  HttpAuthChallengeTokenizer tokenizer(challenge.begin(), challenge.end());
 
   HttpAuthCache::Entry* entry =
       auth_cache.Add(GURL("http://example.com").GetOrigin(), "bar", "digest",
                      challenge, AuthCredentials(), "/bar");
   EXPECT_FALSE(digest_factory.CreateAndInitPreemptiveAuthHandler(
-      entry, tokenizer, HttpAuth::AUTH_SERVER, BoundNetLog()));
+      entry, HttpAuth::AUTH_SERVER, BoundNetLog()));
 }
 
 }  // namespace net