OLD | NEW |
---|---|
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "content/browser/loader/test_resource_handler.h" | 5 #include "content/browser/loader/test_resource_handler.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "content/browser/loader/resource_controller.h" | 8 #include "content/browser/loader/resource_controller.h" |
9 #include "content/public/common/resource_response.h" | 9 #include "content/public/common/resource_response.h" |
10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
(...skipping 114 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
125 bool TestResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf, | 125 bool TestResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf, |
126 int* buf_size, | 126 int* buf_size, |
127 int min_size) { | 127 int min_size) { |
128 EXPECT_FALSE(canceled_); | 128 EXPECT_FALSE(canceled_); |
129 EXPECT_FALSE(expect_on_data_downloaded_); | 129 EXPECT_FALSE(expect_on_data_downloaded_); |
130 EXPECT_EQ(0, on_response_completed_called_); | 130 EXPECT_EQ(0, on_response_completed_called_); |
131 ScopedCallDepthTracker call_depth_tracker(&call_depth_); | 131 ScopedCallDepthTracker call_depth_tracker(&call_depth_); |
132 | 132 |
133 ++on_will_read_called_; | 133 ++on_will_read_called_; |
134 | 134 |
135 *buf = buffer_; | 135 if (!on_will_read_result_) { |
136 *buf_size = buffer_size_; | |
137 memset(buffer_->data(), '\0', buffer_size_); | |
138 if (!on_will_read_result_) | |
139 canceled_ = true; | 136 canceled_ = true; |
137 } else { | |
138 *buf = buffer_; | |
139 *buf_size = buffer_size_; | |
140 memset(buffer_->data(), '\0', buffer_size_); | |
mmenke
2017/01/10 19:20:34
This change is to meet MockResourceLoader's expect
| |
141 } | |
142 | |
140 return on_will_read_result_; | 143 return on_will_read_result_; |
141 } | 144 } |
142 | 145 |
143 bool TestResourceHandler::OnReadCompleted(int bytes_read, bool* defer) { | 146 bool TestResourceHandler::OnReadCompleted(int bytes_read, bool* defer) { |
144 EXPECT_FALSE(canceled_); | 147 EXPECT_FALSE(canceled_); |
145 EXPECT_FALSE(expect_on_data_downloaded_); | 148 EXPECT_FALSE(expect_on_data_downloaded_); |
146 EXPECT_EQ(1, on_will_start_called_); | 149 EXPECT_EQ(1, on_will_start_called_); |
147 EXPECT_EQ(1, on_response_started_called_); | 150 EXPECT_EQ(1, on_response_started_called_); |
148 EXPECT_EQ(0, on_response_completed_called_); | 151 EXPECT_EQ(0, on_response_completed_called_); |
149 EXPECT_EQ(0, on_read_eof_); | 152 EXPECT_EQ(0, on_read_eof_); |
(...skipping 71 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
221 void TestResourceHandler::WaitUntilDeferred() { | 224 void TestResourceHandler::WaitUntilDeferred() { |
222 deferred_run_loop_->Run(); | 225 deferred_run_loop_->Run(); |
223 deferred_run_loop_.reset(new base::RunLoop()); | 226 deferred_run_loop_.reset(new base::RunLoop()); |
224 } | 227 } |
225 | 228 |
226 void TestResourceHandler::WaitUntilResponseComplete() { | 229 void TestResourceHandler::WaitUntilResponseComplete() { |
227 response_complete_run_loop_.Run(); | 230 response_complete_run_loop_.Run(); |
228 } | 231 } |
229 | 232 |
230 } // namespace content | 233 } // namespace content |
OLD | NEW |