| Index: net/http/http_transaction_test_util.h
|
| diff --git a/net/http/http_transaction_test_util.h b/net/http/http_transaction_test_util.h
|
| index 0c311df4cde55c3e646206f104f9e57673039611..865d7f99fefbe297ecbbcfb715428174016b0c43 100644
|
| --- a/net/http/http_transaction_test_util.h
|
| +++ b/net/http/http_transaction_test_util.h
|
| @@ -52,10 +52,14 @@ enum {
|
| TEST_MODE_SLOW_READ = 1 << 5
|
| };
|
|
|
| -typedef void (*MockTransactionHandler)(const HttpRequestInfo* request,
|
| - std::string* response_status,
|
| - std::string* response_headers,
|
| - std::string* response_data);
|
| +using MockTransactionReadHandler = int (*)(int64 content_length,
|
| + int64 offset,
|
| + IOBuffer* buf,
|
| + int buf_len);
|
| +using MockTransactionHandler = void (*)(const HttpRequestInfo* request,
|
| + std::string* response_status,
|
| + std::string* response_headers,
|
| + std::string* response_data);
|
|
|
| struct MockTransaction {
|
| const char* url;
|
| @@ -71,6 +75,7 @@ struct MockTransaction {
|
| const char* data;
|
| int test_mode;
|
| MockTransactionHandler handler;
|
| + MockTransactionReadHandler read_handler;
|
| scoped_refptr<X509Certificate> cert;
|
| CertStatus cert_status;
|
| int ssl_connection_status;
|
| @@ -251,10 +256,13 @@ class MockNetworkTransaction
|
| const HttpRequestInfo* request_;
|
| HttpResponseInfo response_;
|
| std::string data_;
|
| - int data_cursor_;
|
| + int64 data_cursor_;
|
| + int64 content_length_;
|
| int test_mode_;
|
| RequestPriority priority_;
|
| + MockTransactionReadHandler read_handler_;
|
| CreateHelper* websocket_handshake_stream_create_helper_;
|
| + BeforeNetworkStartCallback before_network_start_callback_;
|
| base::WeakPtr<MockNetworkLayer> transaction_factory_;
|
| int64 received_bytes_;
|
| int64_t sent_bytes_;
|
| @@ -280,6 +288,10 @@ class MockNetworkLayer : public HttpTransactionFactory,
|
| void TransactionDoneReading();
|
| void TransactionStopCaching();
|
|
|
| + // Resets the transaction count. Can be called after test setup in order to
|
| + // make test expectations independent of how test setup is performed.
|
| + void ResetTransactionCount();
|
| +
|
| // Returns the last priority passed to CreateTransaction, or
|
| // DEFAULT_PRIORITY if it hasn't been called yet.
|
| RequestPriority last_create_transaction_priority() const {
|
|
|