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 "net/http/http_auth_controller.h" | 5 #include "net/http/http_auth_controller.h" |
6 | 6 |
7 #include "base/strings/utf_string_conversions.h" | 7 #include "base/strings/utf_string_conversions.h" |
8 #include "net/base/net_errors.h" | 8 #include "net/base/net_errors.h" |
9 #include "net/base/net_log.h" | 9 #include "net/base/net_log.h" |
10 #include "net/base/test_completion_callback.h" | 10 #include "net/base/test_completion_callback.h" |
(...skipping 110 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
121 TEST(HttpAuthControllerTest, NoExplicitCredentialsAllowed) { | 121 TEST(HttpAuthControllerTest, NoExplicitCredentialsAllowed) { |
122 // Modified mock HttpAuthHandler for this test. | 122 // Modified mock HttpAuthHandler for this test. |
123 class MockHandler : public HttpAuthHandlerMock { | 123 class MockHandler : public HttpAuthHandlerMock { |
124 public: | 124 public: |
125 MockHandler(int expected_rv, HttpAuth::Scheme scheme) | 125 MockHandler(int expected_rv, HttpAuth::Scheme scheme) |
126 : expected_scheme_(scheme) { | 126 : expected_scheme_(scheme) { |
127 SetGenerateExpectation(false, expected_rv); | 127 SetGenerateExpectation(false, expected_rv); |
128 } | 128 } |
129 | 129 |
130 protected: | 130 protected: |
131 virtual bool Init(HttpAuthChallengeTokenizer* challenge) OVERRIDE { | 131 virtual bool Init(HttpAuthChallengeTokenizer* challenge) override { |
132 HttpAuthHandlerMock::Init(challenge); | 132 HttpAuthHandlerMock::Init(challenge); |
133 set_allows_default_credentials(true); | 133 set_allows_default_credentials(true); |
134 set_allows_explicit_credentials(false); | 134 set_allows_explicit_credentials(false); |
135 set_connection_based(true); | 135 set_connection_based(true); |
136 // Pretend to be SCHEME_BASIC so we can test failover logic. | 136 // Pretend to be SCHEME_BASIC so we can test failover logic. |
137 if (challenge->scheme() == "Basic") { | 137 if (challenge->scheme() == "Basic") { |
138 auth_scheme_ = HttpAuth::AUTH_SCHEME_BASIC; | 138 auth_scheme_ = HttpAuth::AUTH_SCHEME_BASIC; |
139 --score_; // Reduce score, so we rank below Mock. | 139 --score_; // Reduce score, so we rank below Mock. |
140 set_allows_explicit_credentials(true); | 140 set_allows_explicit_credentials(true); |
141 } | 141 } |
142 EXPECT_EQ(expected_scheme_, auth_scheme_); | 142 EXPECT_EQ(expected_scheme_, auth_scheme_); |
143 return true; | 143 return true; |
144 } | 144 } |
145 | 145 |
146 virtual int GenerateAuthTokenImpl(const AuthCredentials* credentials, | 146 virtual int GenerateAuthTokenImpl(const AuthCredentials* credentials, |
147 const HttpRequestInfo* request, | 147 const HttpRequestInfo* request, |
148 const CompletionCallback& callback, | 148 const CompletionCallback& callback, |
149 std::string* auth_token) OVERRIDE { | 149 std::string* auth_token) override { |
150 int result = | 150 int result = |
151 HttpAuthHandlerMock::GenerateAuthTokenImpl(credentials, | 151 HttpAuthHandlerMock::GenerateAuthTokenImpl(credentials, |
152 request, callback, | 152 request, callback, |
153 auth_token); | 153 auth_token); |
154 EXPECT_TRUE(result != OK || | 154 EXPECT_TRUE(result != OK || |
155 !AllowsExplicitCredentials() || | 155 !AllowsExplicitCredentials() || |
156 !credentials->Empty()); | 156 !credentials->Empty()); |
157 return result; | 157 return result; |
158 } | 158 } |
159 | 159 |
(...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
230 EXPECT_TRUE(controller->HaveAuth()); | 230 EXPECT_TRUE(controller->HaveAuth()); |
231 EXPECT_TRUE(controller->IsAuthSchemeDisabled(HttpAuth::AUTH_SCHEME_MOCK)); | 231 EXPECT_TRUE(controller->IsAuthSchemeDisabled(HttpAuth::AUTH_SCHEME_MOCK)); |
232 EXPECT_FALSE(controller->IsAuthSchemeDisabled(HttpAuth::AUTH_SCHEME_BASIC)); | 232 EXPECT_FALSE(controller->IsAuthSchemeDisabled(HttpAuth::AUTH_SCHEME_BASIC)); |
233 | 233 |
234 // Should only succeed if we are using the AUTH_SCHEME_BASIC MockHandler. | 234 // Should only succeed if we are using the AUTH_SCHEME_BASIC MockHandler. |
235 EXPECT_EQ(OK, controller->MaybeGenerateAuthToken( | 235 EXPECT_EQ(OK, controller->MaybeGenerateAuthToken( |
236 &request, CompletionCallback(), dummy_log)); | 236 &request, CompletionCallback(), dummy_log)); |
237 } | 237 } |
238 | 238 |
239 } // namespace net | 239 } // namespace net |
OLD | NEW |