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/http/http_response_body_drainer.h" | 5 #include "net/http/http_response_body_drainer.h" |
6 | 6 |
7 #include <cstring> | 7 #include <cstring> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/compiler_specific.h" | 10 #include "base/compiler_specific.h" |
(...skipping 62 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
73 is_sync_(false), | 73 is_sync_(false), |
74 is_last_chunk_zero_size_(false), | 74 is_last_chunk_zero_size_(false), |
75 is_complete_(false), | 75 is_complete_(false), |
76 weak_factory_(this) {} | 76 weak_factory_(this) {} |
77 virtual ~MockHttpStream() {} | 77 virtual ~MockHttpStream() {} |
78 | 78 |
79 // HttpStream implementation. | 79 // HttpStream implementation. |
80 virtual int InitializeStream(const HttpRequestInfo* request_info, | 80 virtual int InitializeStream(const HttpRequestInfo* request_info, |
81 RequestPriority priority, | 81 RequestPriority priority, |
82 const BoundNetLog& net_log, | 82 const BoundNetLog& net_log, |
83 const CompletionCallback& callback) OVERRIDE { | 83 const CompletionCallback& callback) override { |
84 return ERR_UNEXPECTED; | 84 return ERR_UNEXPECTED; |
85 } | 85 } |
86 virtual int SendRequest(const HttpRequestHeaders& request_headers, | 86 virtual int SendRequest(const HttpRequestHeaders& request_headers, |
87 HttpResponseInfo* response, | 87 HttpResponseInfo* response, |
88 const CompletionCallback& callback) OVERRIDE { | 88 const CompletionCallback& callback) override { |
89 return ERR_UNEXPECTED; | 89 return ERR_UNEXPECTED; |
90 } | 90 } |
91 virtual UploadProgress GetUploadProgress() const OVERRIDE { | 91 virtual UploadProgress GetUploadProgress() const override { |
92 return UploadProgress(); | 92 return UploadProgress(); |
93 } | 93 } |
94 virtual int ReadResponseHeaders(const CompletionCallback& callback) OVERRIDE { | 94 virtual int ReadResponseHeaders(const CompletionCallback& callback) override { |
95 return ERR_UNEXPECTED; | 95 return ERR_UNEXPECTED; |
96 } | 96 } |
97 | 97 |
98 virtual bool CanFindEndOfResponse() const OVERRIDE { return true; } | 98 virtual bool CanFindEndOfResponse() const override { return true; } |
99 virtual bool IsConnectionReused() const OVERRIDE { return false; } | 99 virtual bool IsConnectionReused() const override { return false; } |
100 virtual void SetConnectionReused() OVERRIDE {} | 100 virtual void SetConnectionReused() override {} |
101 virtual bool IsConnectionReusable() const OVERRIDE { return false; } | 101 virtual bool IsConnectionReusable() const override { return false; } |
102 virtual int64 GetTotalReceivedBytes() const OVERRIDE { return 0; } | 102 virtual int64 GetTotalReceivedBytes() const override { return 0; } |
103 virtual void GetSSLInfo(SSLInfo* ssl_info) OVERRIDE {} | 103 virtual void GetSSLInfo(SSLInfo* ssl_info) override {} |
104 virtual void GetSSLCertRequestInfo( | 104 virtual void GetSSLCertRequestInfo( |
105 SSLCertRequestInfo* cert_request_info) OVERRIDE {} | 105 SSLCertRequestInfo* cert_request_info) override {} |
106 | 106 |
107 // Mocked API | 107 // Mocked API |
108 virtual int ReadResponseBody(IOBuffer* buf, int buf_len, | 108 virtual int ReadResponseBody(IOBuffer* buf, int buf_len, |
109 const CompletionCallback& callback) OVERRIDE; | 109 const CompletionCallback& callback) override; |
110 virtual void Close(bool not_reusable) OVERRIDE { | 110 virtual void Close(bool not_reusable) override { |
111 CHECK(!closed_); | 111 CHECK(!closed_); |
112 closed_ = true; | 112 closed_ = true; |
113 result_waiter_->set_result(not_reusable); | 113 result_waiter_->set_result(not_reusable); |
114 } | 114 } |
115 | 115 |
116 virtual HttpStream* RenewStreamForAuth() OVERRIDE { | 116 virtual HttpStream* RenewStreamForAuth() override { |
117 return NULL; | 117 return NULL; |
118 } | 118 } |
119 | 119 |
120 virtual bool IsResponseBodyComplete() const OVERRIDE { return is_complete_; } | 120 virtual bool IsResponseBodyComplete() const override { return is_complete_; } |
121 | 121 |
122 virtual bool IsSpdyHttpStream() const OVERRIDE { return false; } | 122 virtual bool IsSpdyHttpStream() const override { return false; } |
123 | 123 |
124 virtual bool GetLoadTimingInfo( | 124 virtual bool GetLoadTimingInfo( |
125 LoadTimingInfo* load_timing_info) const OVERRIDE { return false; } | 125 LoadTimingInfo* load_timing_info) const override { return false; } |
126 | 126 |
127 virtual void Drain(HttpNetworkSession*) OVERRIDE {} | 127 virtual void Drain(HttpNetworkSession*) override {} |
128 | 128 |
129 virtual void SetPriority(RequestPriority priority) OVERRIDE {} | 129 virtual void SetPriority(RequestPriority priority) override {} |
130 | 130 |
131 // Methods to tweak/observer mock behavior: | 131 // Methods to tweak/observer mock behavior: |
132 void set_stall_reads_forever() { stall_reads_forever_ = true; } | 132 void set_stall_reads_forever() { stall_reads_forever_ = true; } |
133 | 133 |
134 void set_num_chunks(int num_chunks) { num_chunks_ = num_chunks; } | 134 void set_num_chunks(int num_chunks) { num_chunks_ = num_chunks; } |
135 | 135 |
136 void set_sync() { is_sync_ = true; } | 136 void set_sync() { is_sync_ = true; } |
137 | 137 |
138 void set_is_last_chunk_zero_size() { is_last_chunk_zero_size_ = true; } | 138 void set_is_last_chunk_zero_size() { is_last_chunk_zero_size_ = true; } |
139 | 139 |
(...skipping 162 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
302 too_many_chunks += 1; // Now it's too large. | 302 too_many_chunks += 1; // Now it's too large. |
303 | 303 |
304 mock_stream_->set_num_chunks(too_many_chunks); | 304 mock_stream_->set_num_chunks(too_many_chunks); |
305 drainer_->Start(session_.get()); | 305 drainer_->Start(session_.get()); |
306 EXPECT_TRUE(result_waiter_.WaitForResult()); | 306 EXPECT_TRUE(result_waiter_.WaitForResult()); |
307 } | 307 } |
308 | 308 |
309 } // namespace | 309 } // namespace |
310 | 310 |
311 } // namespace net | 311 } // namespace net |
OLD | NEW |