| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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 #ifndef NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_ | 5 #ifndef NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_ |
| 6 #define NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_ | 6 #define NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_ |
| 7 | 7 |
| 8 #include "net/http/http_transaction.h" | 8 #include "net/http/http_transaction.h" |
| 9 | 9 |
| 10 #include <stdint.h> | 10 #include <stdint.h> |
| (...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 123 | 123 |
| 124 //----------------------------------------------------------------------------- | 124 //----------------------------------------------------------------------------- |
| 125 // use this class to test completely consuming a transaction | 125 // use this class to test completely consuming a transaction |
| 126 | 126 |
| 127 class TestTransactionConsumer { | 127 class TestTransactionConsumer { |
| 128 public: | 128 public: |
| 129 TestTransactionConsumer(RequestPriority priority, | 129 TestTransactionConsumer(RequestPriority priority, |
| 130 HttpTransactionFactory* factory); | 130 HttpTransactionFactory* factory); |
| 131 virtual ~TestTransactionConsumer(); | 131 virtual ~TestTransactionConsumer(); |
| 132 | 132 |
| 133 void Start(const HttpRequestInfo* request, const BoundNetLog& net_log); | 133 void Start(const HttpRequestInfo* request, const NetLogWithSource& net_log); |
| 134 | 134 |
| 135 bool is_done() const { return state_ == DONE; } | 135 bool is_done() const { return state_ == DONE; } |
| 136 int error() const { return error_; } | 136 int error() const { return error_; } |
| 137 | 137 |
| 138 const HttpResponseInfo* response_info() const { | 138 const HttpResponseInfo* response_info() const { |
| 139 return trans_->GetResponseInfo(); | 139 return trans_->GetResponseInfo(); |
| 140 } | 140 } |
| 141 const HttpTransaction* transaction() const { return trans_.get(); } | 141 const HttpTransaction* transaction() const { return trans_.get(); } |
| 142 const std::string& content() const { return content_; } | 142 const std::string& content() const { return content_; } |
| 143 | 143 |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 178 : public HttpTransaction, | 178 : public HttpTransaction, |
| 179 public base::SupportsWeakPtr<MockNetworkTransaction> { | 179 public base::SupportsWeakPtr<MockNetworkTransaction> { |
| 180 typedef WebSocketHandshakeStreamBase::CreateHelper CreateHelper; | 180 typedef WebSocketHandshakeStreamBase::CreateHelper CreateHelper; |
| 181 | 181 |
| 182 public: | 182 public: |
| 183 MockNetworkTransaction(RequestPriority priority, MockNetworkLayer* factory); | 183 MockNetworkTransaction(RequestPriority priority, MockNetworkLayer* factory); |
| 184 ~MockNetworkTransaction() override; | 184 ~MockNetworkTransaction() override; |
| 185 | 185 |
| 186 int Start(const HttpRequestInfo* request, | 186 int Start(const HttpRequestInfo* request, |
| 187 const CompletionCallback& callback, | 187 const CompletionCallback& callback, |
| 188 const BoundNetLog& net_log) override; | 188 const NetLogWithSource& net_log) override; |
| 189 | 189 |
| 190 int RestartIgnoringLastError(const CompletionCallback& callback) override; | 190 int RestartIgnoringLastError(const CompletionCallback& callback) override; |
| 191 | 191 |
| 192 int RestartWithCertificate(X509Certificate* client_cert, | 192 int RestartWithCertificate(X509Certificate* client_cert, |
| 193 SSLPrivateKey* client_private_key, | 193 SSLPrivateKey* client_private_key, |
| 194 const CompletionCallback& callback) override; | 194 const CompletionCallback& callback) override; |
| 195 | 195 |
| 196 int RestartWithAuth(const AuthCredentials& credentials, | 196 int RestartWithAuth(const AuthCredentials& credentials, |
| 197 const CompletionCallback& callback) override; | 197 const CompletionCallback& callback) override; |
| 198 | 198 |
| (...skipping 48 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 247 // Bogus value that will be returned by GetTotalReceivedBytes() if the | 247 // Bogus value that will be returned by GetTotalReceivedBytes() if the |
| 248 // MockNetworkTransaction was started. | 248 // MockNetworkTransaction was started. |
| 249 static const int64_t kTotalReceivedBytes; | 249 static const int64_t kTotalReceivedBytes; |
| 250 // Bogus value that will be returned by GetTotalSentBytes() if the | 250 // Bogus value that will be returned by GetTotalSentBytes() if the |
| 251 // MockNetworkTransaction was started. | 251 // MockNetworkTransaction was started. |
| 252 static const int64_t kTotalSentBytes; | 252 static const int64_t kTotalSentBytes; |
| 253 | 253 |
| 254 private: | 254 private: |
| 255 int StartInternal(const HttpRequestInfo* request, | 255 int StartInternal(const HttpRequestInfo* request, |
| 256 const CompletionCallback& callback, | 256 const CompletionCallback& callback, |
| 257 const BoundNetLog& net_log); | 257 const NetLogWithSource& net_log); |
| 258 void CallbackLater(const CompletionCallback& callback, int result); | 258 void CallbackLater(const CompletionCallback& callback, int result); |
| 259 void RunCallback(const CompletionCallback& callback, int result); | 259 void RunCallback(const CompletionCallback& callback, int result); |
| 260 | 260 |
| 261 const HttpRequestInfo* request_; | 261 const HttpRequestInfo* request_; |
| 262 HttpResponseInfo response_; | 262 HttpResponseInfo response_; |
| 263 std::string data_; | 263 std::string data_; |
| 264 int64_t data_cursor_; | 264 int64_t data_cursor_; |
| 265 int64_t content_length_; | 265 int64_t content_length_; |
| 266 int test_mode_; | 266 int test_mode_; |
| 267 RequestPriority priority_; | 267 RequestPriority priority_; |
| 268 MockTransactionReadHandler read_handler_; | 268 MockTransactionReadHandler read_handler_; |
| 269 CreateHelper* websocket_handshake_stream_create_helper_; | 269 CreateHelper* websocket_handshake_stream_create_helper_; |
| 270 BeforeNetworkStartCallback before_network_start_callback_; | 270 BeforeNetworkStartCallback before_network_start_callback_; |
| 271 base::WeakPtr<MockNetworkLayer> transaction_factory_; | 271 base::WeakPtr<MockNetworkLayer> transaction_factory_; |
| 272 int64_t received_bytes_; | 272 int64_t received_bytes_; |
| 273 int64_t sent_bytes_; | 273 int64_t sent_bytes_; |
| 274 | 274 |
| 275 // NetLog ID of the fake / non-existent underlying socket used by the | 275 // NetLog ID of the fake / non-existent underlying socket used by the |
| 276 // connection. Requires Start() be passed a BoundNetLog with a real NetLog to | 276 // connection. Requires Start() be passed a NetLogWithSource with a real |
| 277 // NetLog to |
| 277 // be initialized. | 278 // be initialized. |
| 278 unsigned int socket_log_id_; | 279 unsigned int socket_log_id_; |
| 279 | 280 |
| 280 bool done_reading_called_; | 281 bool done_reading_called_; |
| 281 | 282 |
| 282 base::WeakPtrFactory<MockNetworkTransaction> weak_factory_; | 283 base::WeakPtrFactory<MockNetworkTransaction> weak_factory_; |
| 283 | 284 |
| 284 }; | 285 }; |
| 285 | 286 |
| 286 class MockNetworkLayer : public HttpTransactionFactory, | 287 class MockNetworkLayer : public HttpTransactionFactory, |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 348 | 349 |
| 349 //----------------------------------------------------------------------------- | 350 //----------------------------------------------------------------------------- |
| 350 // helpers | 351 // helpers |
| 351 | 352 |
| 352 // read the transaction completely | 353 // read the transaction completely |
| 353 int ReadTransaction(HttpTransaction* trans, std::string* result); | 354 int ReadTransaction(HttpTransaction* trans, std::string* result); |
| 354 | 355 |
| 355 } // namespace net | 356 } // namespace net |
| 356 | 357 |
| 357 #endif // NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_ | 358 #endif // NET_HTTP_HTTP_TRANSACTION_UNITTEST_H_ |
| OLD | NEW |