net/http/http_auth_handler_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "net/http/http_auth_handler.h"
   6
   7 #include "base/strings/string_util.h"
   8 #include "base/strings/utf_string_conversions.h"
   9 #include "net/base/capturing_net_log.h"
  10 #include "net/base/net_errors.h"
  11 #include "net/base/net_log_unittest.h"
  12 #include "net/base/test_completion_callback.h"
  13 #include "net/http/http_auth_challenge_tokenizer.h"
  14 #include "net/http/http_auth_handler_mock.h"
  15 #include "net/http/http_request_info.h"
  16 #include "testing/gtest/include/gtest/gtest.h"
  17
  18 namespace net {
  19
  20 TEST(HttpAuthHandlerTest, NetLog) {
  21   GURL origin("http://www.example.com");
  22   std::string challenge = "Mock asdf";
  23   AuthCredentials credentials(base::ASCIIToUTF16("user"),
  24                               base::ASCIIToUTF16("pass"));
  25   std::string auth_token;
  26   HttpRequestInfo request;
  27
  28   for (int i = 0; i < 2; ++i) {
  29     bool async = (i == 0);
  30     for (int j = 0; j < 2; ++j) {
  31       int rv = (j == 0) ? OK : ERR_UNEXPECTED;
  32       for (int k = 0; k < 2; ++k) {
  33         TestCompletionCallback test_callback;
  34         HttpAuth::Target target =
  35             (k == 0) ? HttpAuth::AUTH_PROXY : HttpAuth::AUTH_SERVER;
  36         NetLog::EventType event_type =
  37             (k == 0) ? NetLog::TYPE_AUTH_PROXY : NetLog::TYPE_AUTH_SERVER;
  38         HttpAuthChallengeTokenizer tokenizer(
  39             challenge.begin(), challenge.end());
  40         HttpAuthHandlerMock mock_handler;
  41         CapturingNetLog capturing_net_log;
  42         BoundNetLog bound_net_log(BoundNetLog::Make(&capturing_net_log,
  43                                                     net::NetLog::SOURCE_NONE));
  44
  45         mock_handler.InitFromChallenge(&tokenizer, target,
  46                                        origin, bound_net_log);
  47         mock_handler.SetGenerateExpectation(async, rv);
  48         mock_handler.GenerateAuthToken(&credentials, &request,
  49                                        test_callback.callback(), &auth_token);
  50         if (async)
  51           test_callback.WaitForResult();
  52
  53         CapturingNetLog::CapturedEntryList entries;
  54         capturing_net_log.GetEntries(&entries);
  55
  56         EXPECT_EQ(2u, entries.size());
  57         EXPECT_TRUE(LogContainsBeginEvent(entries, 0, event_type));
  58         EXPECT_TRUE(LogContainsEndEvent(entries, 1, event_type));
  59       }
  60     }
  61   }
  62 }
  63
  64 }  // namespace net