OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 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 "net/url_request/url_request_context_builder.h" | 5 #include "net/url_request/url_request_context_builder.h" |
6 | 6 |
7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
8 #include "build/build_config.h" | 8 #include "build/build_config.h" |
9 #include "net/base/request_priority.h" | 9 #include "net/base/request_priority.h" |
10 #include "net/http/http_auth_handler.h" | 10 #include "net/http/http_auth_handler.h" |
11 #include "net/http/http_auth_handler_factory.h" | 11 #include "net/http/http_auth_handler_factory.h" |
12 #include "net/test/spawned_test_server/spawned_test_server.h" | 12 #include "net/test/spawned_test_server/spawned_test_server.h" |
13 #include "net/url_request/url_request.h" | 13 #include "net/url_request/url_request.h" |
14 #include "net/url_request/url_request_test_util.h" | 14 #include "net/url_request/url_request_test_util.h" |
15 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
16 #include "testing/platform_test.h" | 16 #include "testing/platform_test.h" |
| 17 #include "url/origin.h" |
17 | 18 |
18 #if defined(OS_LINUX) || defined(OS_ANDROID) | 19 #if defined(OS_LINUX) || defined(OS_ANDROID) |
19 #include "net/proxy/proxy_config.h" | 20 #include "net/proxy/proxy_config.h" |
20 #include "net/proxy/proxy_config_service_fixed.h" | 21 #include "net/proxy/proxy_config_service_fixed.h" |
21 #endif // defined(OS_LINUX) || defined(OS_ANDROID) | 22 #endif // defined(OS_LINUX) || defined(OS_ANDROID) |
22 | 23 |
23 namespace net { | 24 namespace net { |
24 | 25 |
25 namespace { | 26 namespace { |
26 | 27 |
27 class MockHttpAuthHandlerFactory : public HttpAuthHandlerFactory { | 28 class MockHttpAuthHandlerFactory : public HttpAuthHandlerFactory { |
28 public: | 29 public: |
29 explicit MockHttpAuthHandlerFactory(int return_code) : | 30 explicit MockHttpAuthHandlerFactory(int return_code) : |
30 return_code_(return_code) {} | 31 return_code_(return_code) {} |
31 ~MockHttpAuthHandlerFactory() override {} | 32 ~MockHttpAuthHandlerFactory() override {} |
32 | 33 |
33 int CreateAuthHandler(HttpAuthChallengeTokenizer* challenge, | 34 int CreateAuthHandler(HttpAuthChallengeTokenizer* challenge, |
34 HttpAuth::Target target, | 35 HttpAuth::Target target, |
35 const GURL& origin, | 36 const url::Origin& origin, |
36 CreateReason reason, | 37 CreateReason reason, |
37 int nonce_count, | 38 int nonce_count, |
38 const BoundNetLog& net_log, | 39 const BoundNetLog& net_log, |
39 scoped_ptr<HttpAuthHandler>* handler) override { | 40 scoped_ptr<HttpAuthHandler>* handler) override { |
40 handler->reset(); | 41 handler->reset(); |
41 return return_code_; | 42 return return_code_; |
42 } | 43 } |
43 | 44 |
44 private: | 45 private: |
45 int return_code_; | 46 int return_code_; |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
85 scoped_ptr<URLRequest> request( | 86 scoped_ptr<URLRequest> request( |
86 context->CreateRequest(test_server_.GetURL("echoheader?User-Agent"), | 87 context->CreateRequest(test_server_.GetURL("echoheader?User-Agent"), |
87 DEFAULT_PRIORITY, &delegate)); | 88 DEFAULT_PRIORITY, &delegate)); |
88 request->set_method("GET"); | 89 request->set_method("GET"); |
89 request->Start(); | 90 request->Start(); |
90 base::MessageLoop::current()->Run(); | 91 base::MessageLoop::current()->Run(); |
91 EXPECT_EQ("Bar", delegate.data_received()); | 92 EXPECT_EQ("Bar", delegate.data_received()); |
92 } | 93 } |
93 | 94 |
94 TEST_F(URLRequestContextBuilderTest, ExtraHttpAuthHandlerFactory) { | 95 TEST_F(URLRequestContextBuilderTest, ExtraHttpAuthHandlerFactory) { |
95 GURL gurl("www.google.com"); | 96 url::Origin origin("www.google.com"); |
96 const int kBasicReturnCode = OK; | 97 const int kBasicReturnCode = OK; |
97 MockHttpAuthHandlerFactory* mock_factory_basic = | 98 MockHttpAuthHandlerFactory* mock_factory_basic = |
98 new MockHttpAuthHandlerFactory(kBasicReturnCode); | 99 new MockHttpAuthHandlerFactory(kBasicReturnCode); |
99 scoped_ptr<HttpAuthHandler> handler; | 100 scoped_ptr<HttpAuthHandler> handler; |
100 builder_.add_http_auth_handler_factory("ExtraScheme", mock_factory_basic); | 101 builder_.add_http_auth_handler_factory("ExtraScheme", mock_factory_basic); |
101 scoped_ptr<URLRequestContext> context(builder_.Build()); | 102 scoped_ptr<URLRequestContext> context(builder_.Build()); |
102 // Verify that a handler is returned for and added scheme. | 103 // Verify that a handler is returned for and added scheme. |
103 EXPECT_EQ(kBasicReturnCode, | 104 EXPECT_EQ(kBasicReturnCode, |
104 context->http_auth_handler_factory()->CreateAuthHandlerFromString( | 105 context->http_auth_handler_factory()->CreateAuthHandlerFromString( |
105 "ExtraScheme", | 106 "ExtraScheme", HttpAuth::AUTH_SERVER, origin, BoundNetLog(), |
106 HttpAuth::AUTH_SERVER, | |
107 gurl, | |
108 BoundNetLog(), | |
109 &handler)); | 107 &handler)); |
110 // Verify that a handler isn't returned for a bogus scheme. | 108 // Verify that a handler isn't returned for a bogus scheme. |
111 EXPECT_EQ(ERR_UNSUPPORTED_AUTH_SCHEME, | 109 EXPECT_EQ( |
112 context->http_auth_handler_factory()->CreateAuthHandlerFromString( | 110 ERR_UNSUPPORTED_AUTH_SCHEME, |
113 "Bogus", HttpAuth::AUTH_SERVER, gurl, BoundNetLog(), &handler)); | 111 context->http_auth_handler_factory()->CreateAuthHandlerFromString( |
| 112 "Bogus", HttpAuth::AUTH_SERVER, origin, BoundNetLog(), &handler)); |
114 } | 113 } |
115 | 114 |
116 } // namespace | 115 } // namespace |
117 | 116 |
118 } // namespace net | 117 } // namespace net |
OLD | NEW |