OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <set> | 5 #include <set> |
6 #include <string> | 6 #include <string> |
7 | 7 |
8 #include "base/memory/ref_counted.h" | 8 #include "base/memory/ref_counted.h" |
9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
10 #include "base/strings/string_util.h" | 10 #include "base/strings/string_util.h" |
(...skipping 11 matching lines...) Expand all Loading... |
22 #include "net/http/mock_allow_url_security_manager.h" | 22 #include "net/http/mock_allow_url_security_manager.h" |
23 #include "testing/gtest/include/gtest/gtest.h" | 23 #include "testing/gtest/include/gtest/gtest.h" |
24 | 24 |
25 namespace net { | 25 namespace net { |
26 | 26 |
27 namespace { | 27 namespace { |
28 | 28 |
29 HttpAuthHandlerMock* CreateMockHandler(bool expect_multiple_challenges) { | 29 HttpAuthHandlerMock* CreateMockHandler(bool expect_multiple_challenges) { |
30 HttpAuthHandlerMock* auth_handler = new HttpAuthHandlerMock(); | 30 HttpAuthHandlerMock* auth_handler = new HttpAuthHandlerMock(); |
31 auth_handler->set_expect_multiple_challenges(expect_multiple_challenges); | 31 auth_handler->set_expect_multiple_challenges(expect_multiple_challenges); |
32 std::string challenge_text = "Basic"; | 32 std::string challenge_text = "Mock"; |
33 HttpAuthChallengeTokenizer challenge(challenge_text.begin(), | 33 HttpAuthChallengeTokenizer challenge(challenge_text.begin(), |
34 challenge_text.end()); | 34 challenge_text.end()); |
35 GURL origin("www.example.com"); | 35 GURL origin("www.example.com"); |
36 EXPECT_TRUE(auth_handler->InitFromChallenge(&challenge, | 36 EXPECT_TRUE(auth_handler->InitFromChallenge(&challenge, HttpAuth::AUTH_SERVER, |
37 HttpAuth::AUTH_SERVER, | 37 origin, BoundNetLog())); |
38 origin, | |
39 BoundNetLog())); | |
40 return auth_handler; | 38 return auth_handler; |
41 } | 39 } |
42 | 40 |
43 HttpResponseHeaders* HeadersFromResponseText(const std::string& response) { | 41 HttpResponseHeaders* HeadersFromResponseText(const std::string& response) { |
44 return new HttpResponseHeaders( | 42 return new HttpResponseHeaders( |
45 HttpUtil::AssembleRawHeaders(response.c_str(), response.length())); | 43 HttpUtil::AssembleRawHeaders(response.c_str(), response.length())); |
46 } | 44 } |
47 | 45 |
48 HttpAuth::AuthorizationResult HandleChallengeResponse( | 46 HttpAuth::AuthorizationResult HandleChallengeResponse( |
49 bool expect_multiple_challenges, | 47 bool expect_multiple_challenges, |
50 const std::string& headers_text, | 48 const std::string& headers_text, |
51 std::string* challenge_used) { | 49 std::string* challenge_used) { |
52 scoped_ptr<HttpAuthHandlerMock> mock_handler( | 50 scoped_ptr<HttpAuthHandlerMock> mock_handler( |
53 CreateMockHandler(expect_multiple_challenges)); | 51 CreateMockHandler(expect_multiple_challenges)); |
54 HttpAuthSchemeSet disabled_schemes; | 52 HttpAuthSchemeSet disabled_schemes; |
55 scoped_refptr<HttpResponseHeaders> headers( | 53 scoped_refptr<HttpResponseHeaders> headers( |
56 HeadersFromResponseText(headers_text)); | 54 HeadersFromResponseText(headers_text)); |
57 return HttpAuth::HandleChallengeResponse( | 55 return HttpAuth::HandleChallengeResponse(mock_handler.get(), headers.get(), |
58 mock_handler.get(), | 56 HttpAuth::AUTH_SERVER, |
59 headers.get(), | 57 disabled_schemes, challenge_used); |
60 HttpAuth::AUTH_SERVER, | |
61 disabled_schemes, | |
62 challenge_used); | |
63 } | 58 } |
64 | 59 |
65 } // namespace | 60 } // namespace |
66 | 61 |
67 TEST(HttpAuthTest, ChooseBestChallenge) { | 62 TEST(HttpAuthTest, ChooseBestChallenge) { |
68 static const struct { | 63 static const struct { |
69 const char* headers; | 64 const char* headers; |
70 const char* challenge_scheme; | 65 const char* challenge_scheme; |
71 const char* challenge_realm; | 66 const char* challenge_realm; |
72 } tests[] = { | 67 } tests[] = { |
(...skipping 186 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
259 std::string name; | 254 std::string name; |
260 | 255 |
261 name = HttpAuth::GetAuthorizationHeaderName(HttpAuth::AUTH_SERVER); | 256 name = HttpAuth::GetAuthorizationHeaderName(HttpAuth::AUTH_SERVER); |
262 EXPECT_STREQ("Authorization", name.c_str()); | 257 EXPECT_STREQ("Authorization", name.c_str()); |
263 | 258 |
264 name = HttpAuth::GetAuthorizationHeaderName(HttpAuth::AUTH_PROXY); | 259 name = HttpAuth::GetAuthorizationHeaderName(HttpAuth::AUTH_PROXY); |
265 EXPECT_STREQ("Proxy-Authorization", name.c_str()); | 260 EXPECT_STREQ("Proxy-Authorization", name.c_str()); |
266 } | 261 } |
267 | 262 |
268 } // namespace net | 263 } // namespace net |
OLD | NEW |