1// Copyright (c) 2010 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/string_util.h"
8#include "base/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_handler_mock.h"
14#include "net/http/http_request_info.h"
15#include "testing/gtest/include/gtest/gtest.h"
16
17namespace net {
18
19TEST(HttpAuthHandlerTest, NetLog) {
20  NetLog::Source source;
21  GURL origin("http://www.example.com");
22  std::string challenge = "Mock asdf";
23  string16 username = ASCIIToUTF16("user");
24  string16 password = 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        HttpAuth::ChallengeTokenizer tokenizer(
39            challenge.begin(), challenge.end());
40        HttpAuthHandlerMock mock_handler;
41        CapturingNetLog capturing_net_log(CapturingNetLog::kUnbounded);
42        BoundNetLog bound_net_log(source, &capturing_net_log);
43
44        mock_handler.InitFromChallenge(&tokenizer, target,
45                                       origin, bound_net_log);
46        mock_handler.SetGenerateExpectation(async, rv);
47        mock_handler.GenerateAuthToken(&username, &password, &request,
48                                       &test_callback, &auth_token);
49        if (async)
50          test_callback.WaitForResult();
51
52        net::CapturingNetLog::EntryList entries;
53        capturing_net_log.GetEntries(&entries);
54
55        EXPECT_EQ(2u, entries.size());
56        EXPECT_TRUE(LogContainsBeginEvent(entries, 0, event_type));
57        EXPECT_TRUE(LogContainsEndEvent(entries, 1, event_type));
58      }
59    }
60  }
61}
62
63}  // namespace net
64