| 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_handler_mock.h" | 5 #include "net/http/http_auth_handler_mock.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "base/string_util.h" | 9 #include "base/string_util.h" |
| 10 #include "net/base/net_errors.h" | 10 #include "net/base/net_errors.h" |
| 11 #include "net/http/http_request_info.h" | 11 #include "net/http/http_request_info.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
| 13 | 13 |
| 14 namespace net { | 14 namespace net { |
| 15 | 15 |
| 16 HttpAuthHandlerMock::HttpAuthHandlerMock() | 16 HttpAuthHandlerMock::HttpAuthHandlerMock() |
| 17 : resolve_(RESOLVE_INIT), user_callback_(NULL), | 17 : resolve_(RESOLVE_INIT), |
| 18 ALLOW_THIS_IN_INITIALIZER_LIST(ptr_factory_(this)), | 18 ALLOW_THIS_IN_INITIALIZER_LIST(weak_factory_(this)), |
| 19 generate_async_(false), generate_rv_(OK), | 19 generate_async_(false), |
| 20 generate_rv_(OK), |
| 20 auth_token_(NULL), | 21 auth_token_(NULL), |
| 21 first_round_(true), | 22 first_round_(true), |
| 22 connection_based_(false), | 23 connection_based_(false), |
| 23 allows_default_credentials_(false), | 24 allows_default_credentials_(false), |
| 24 allows_explicit_credentials_(true) { | 25 allows_explicit_credentials_(true) { |
| 25 } | 26 } |
| 26 | 27 |
| 27 HttpAuthHandlerMock::~HttpAuthHandlerMock() { | 28 HttpAuthHandlerMock::~HttpAuthHandlerMock() { |
| 28 } | 29 } |
| 29 | 30 |
| 30 void HttpAuthHandlerMock::SetResolveExpectation(Resolve resolve) { | 31 void HttpAuthHandlerMock::SetResolveExpectation(Resolve resolve) { |
| 31 EXPECT_EQ(RESOLVE_INIT, resolve_); | 32 EXPECT_EQ(RESOLVE_INIT, resolve_); |
| 32 resolve_ = resolve; | 33 resolve_ = resolve; |
| 33 } | 34 } |
| 34 | 35 |
| 35 bool HttpAuthHandlerMock::NeedsCanonicalName() { | 36 bool HttpAuthHandlerMock::NeedsCanonicalName() { |
| 36 switch (resolve_) { | 37 switch (resolve_) { |
| 37 case RESOLVE_SYNC: | 38 case RESOLVE_SYNC: |
| 38 case RESOLVE_ASYNC: | 39 case RESOLVE_ASYNC: |
| 39 return true; | 40 return true; |
| 40 case RESOLVE_SKIP: | 41 case RESOLVE_SKIP: |
| 41 resolve_ = RESOLVE_TESTED; | 42 resolve_ = RESOLVE_TESTED; |
| 42 return false; | 43 return false; |
| 43 default: | 44 default: |
| 44 NOTREACHED(); | 45 NOTREACHED(); |
| 45 return false; | 46 return false; |
| 46 } | 47 } |
| 47 } | 48 } |
| 48 | 49 |
| 49 int HttpAuthHandlerMock::ResolveCanonicalName(HostResolver* host_resolver, | 50 int HttpAuthHandlerMock::ResolveCanonicalName( |
| 50 OldCompletionCallback* callback) { | 51 HostResolver* host_resolver, const CompletionCallback& callback) { |
| 51 EXPECT_NE(RESOLVE_TESTED, resolve_); | 52 EXPECT_NE(RESOLVE_TESTED, resolve_); |
| 52 int rv = OK; | 53 int rv = OK; |
| 53 switch (resolve_) { | 54 switch (resolve_) { |
| 54 case RESOLVE_SYNC: | 55 case RESOLVE_SYNC: |
| 55 resolve_ = RESOLVE_TESTED; | 56 resolve_ = RESOLVE_TESTED; |
| 56 break; | 57 break; |
| 57 case RESOLVE_ASYNC: | 58 case RESOLVE_ASYNC: |
| 58 EXPECT_TRUE(user_callback_ == NULL); | 59 EXPECT_TRUE(callback_.is_null()); |
| 59 rv = ERR_IO_PENDING; | 60 rv = ERR_IO_PENDING; |
| 60 user_callback_ = callback; | 61 callback_ = callback; |
| 61 MessageLoop::current()->PostTask( | 62 MessageLoop::current()->PostTask( |
| 62 FROM_HERE, base::Bind( | 63 FROM_HERE, base::Bind( |
| 63 &HttpAuthHandlerMock::OnResolveCanonicalName, | 64 &HttpAuthHandlerMock::OnResolveCanonicalName, |
| 64 ptr_factory_.GetWeakPtr())); | 65 weak_factory_.GetWeakPtr())); |
| 65 break; | 66 break; |
| 66 default: | 67 default: |
| 67 NOTREACHED(); | 68 NOTREACHED(); |
| 68 break; | 69 break; |
| 69 } | 70 } |
| 70 return rv; | 71 return rv; |
| 71 } | 72 } |
| 72 | 73 |
| 73 void HttpAuthHandlerMock::SetGenerateExpectation(bool async, int rv) { | 74 void HttpAuthHandlerMock::SetGenerateExpectation(bool async, int rv) { |
| 74 generate_async_ = async; | 75 generate_async_ = async; |
| (...skipping 26 matching lines...) Expand all Loading... |
| 101 bool HttpAuthHandlerMock::Init(HttpAuth::ChallengeTokenizer* challenge) { | 102 bool HttpAuthHandlerMock::Init(HttpAuth::ChallengeTokenizer* challenge) { |
| 102 auth_scheme_ = HttpAuth::AUTH_SCHEME_MOCK; | 103 auth_scheme_ = HttpAuth::AUTH_SCHEME_MOCK; |
| 103 score_ = 1; | 104 score_ = 1; |
| 104 properties_ = connection_based_ ? IS_CONNECTION_BASED : 0; | 105 properties_ = connection_based_ ? IS_CONNECTION_BASED : 0; |
| 105 return true; | 106 return true; |
| 106 } | 107 } |
| 107 | 108 |
| 108 int HttpAuthHandlerMock::GenerateAuthTokenImpl( | 109 int HttpAuthHandlerMock::GenerateAuthTokenImpl( |
| 109 const AuthCredentials* credentials, | 110 const AuthCredentials* credentials, |
| 110 const HttpRequestInfo* request, | 111 const HttpRequestInfo* request, |
| 111 OldCompletionCallback* callback, | 112 const CompletionCallback& callback, |
| 112 std::string* auth_token) { | 113 std::string* auth_token) { |
| 113 first_round_ = false; | 114 first_round_ = false; |
| 114 request_url_ = request->url; | 115 request_url_ = request->url; |
| 115 if (generate_async_) { | 116 if (generate_async_) { |
| 116 EXPECT_TRUE(user_callback_ == NULL); | 117 EXPECT_TRUE(callback_.is_null()); |
| 117 EXPECT_TRUE(auth_token_ == NULL); | 118 EXPECT_TRUE(auth_token_ == NULL); |
| 118 user_callback_ = callback; | 119 callback_ = callback; |
| 119 auth_token_ = auth_token; | 120 auth_token_ = auth_token; |
| 120 MessageLoop::current()->PostTask( | 121 MessageLoop::current()->PostTask( |
| 121 FROM_HERE, base::Bind( | 122 FROM_HERE, base::Bind( |
| 122 &HttpAuthHandlerMock::OnGenerateAuthToken, | 123 &HttpAuthHandlerMock::OnGenerateAuthToken, |
| 123 ptr_factory_.GetWeakPtr())); | 124 weak_factory_.GetWeakPtr())); |
| 124 return ERR_IO_PENDING; | 125 return ERR_IO_PENDING; |
| 125 } else { | 126 } else { |
| 126 if (generate_rv_ == OK) | 127 if (generate_rv_ == OK) |
| 127 *auth_token = "auth_token"; | 128 *auth_token = "auth_token"; |
| 128 return generate_rv_; | 129 return generate_rv_; |
| 129 } | 130 } |
| 130 } | 131 } |
| 131 | 132 |
| 132 void HttpAuthHandlerMock::OnResolveCanonicalName() { | 133 void HttpAuthHandlerMock::OnResolveCanonicalName() { |
| 133 EXPECT_EQ(RESOLVE_ASYNC, resolve_); | 134 EXPECT_EQ(RESOLVE_ASYNC, resolve_); |
| 134 EXPECT_TRUE(user_callback_ != NULL); | 135 EXPECT_TRUE(!callback_.is_null()); |
| 135 resolve_ = RESOLVE_TESTED; | 136 resolve_ = RESOLVE_TESTED; |
| 136 OldCompletionCallback* callback = user_callback_; | 137 CompletionCallback callback = callback_; |
| 137 user_callback_ = NULL; | 138 callback_.Reset(); |
| 138 callback->Run(OK); | 139 callback.Run(OK); |
| 139 } | 140 } |
| 140 | 141 |
| 141 void HttpAuthHandlerMock::OnGenerateAuthToken() { | 142 void HttpAuthHandlerMock::OnGenerateAuthToken() { |
| 142 EXPECT_TRUE(generate_async_); | 143 EXPECT_TRUE(generate_async_); |
| 143 EXPECT_TRUE(user_callback_ != NULL); | 144 EXPECT_TRUE(!callback_.is_null()); |
| 144 if (generate_rv_ == OK) | 145 if (generate_rv_ == OK) |
| 145 *auth_token_ = "auth_token"; | 146 *auth_token_ = "auth_token"; |
| 146 auth_token_ = NULL; | 147 auth_token_ = NULL; |
| 147 OldCompletionCallback* callback = user_callback_; | 148 CompletionCallback callback = callback_; |
| 148 user_callback_ = NULL; | 149 callback_.Reset(); |
| 149 callback->Run(generate_rv_); | 150 callback.Run(generate_rv_); |
| 150 } | 151 } |
| 151 | 152 |
| 152 HttpAuthHandlerMock::Factory::Factory() | 153 HttpAuthHandlerMock::Factory::Factory() |
| 153 : do_init_from_challenge_(false) { | 154 : do_init_from_challenge_(false) { |
| 154 // TODO(cbentzel): Default do_init_from_challenge_ to true. | 155 // TODO(cbentzel): Default do_init_from_challenge_ to true. |
| 155 } | 156 } |
| 156 | 157 |
| 157 HttpAuthHandlerMock::Factory::~Factory() { | 158 HttpAuthHandlerMock::Factory::~Factory() { |
| 158 } | 159 } |
| 159 | 160 |
| (...skipping 16 matching lines...) Expand all Loading... |
| 176 std::vector<HttpAuthHandler*>& handlers = handlers_[target].get(); | 177 std::vector<HttpAuthHandler*>& handlers = handlers_[target].get(); |
| 177 handlers.erase(handlers.begin()); | 178 handlers.erase(handlers.begin()); |
| 178 if (do_init_from_challenge_ && | 179 if (do_init_from_challenge_ && |
| 179 !tmp_handler->InitFromChallenge(challenge, target, origin, net_log)) | 180 !tmp_handler->InitFromChallenge(challenge, target, origin, net_log)) |
| 180 return ERR_INVALID_RESPONSE; | 181 return ERR_INVALID_RESPONSE; |
| 181 handler->swap(tmp_handler); | 182 handler->swap(tmp_handler); |
| 182 return OK; | 183 return OK; |
| 183 } | 184 } |
| 184 | 185 |
| 185 } // namespace net | 186 } // namespace net |
| OLD | NEW |