Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "content/browser/loader/test_resource_handler.h" | |
| 6 | |
| 7 #include "base/logging.h" | |
| 8 #include "net/url_request/url_request_status.h" | |
| 9 #include "testing/gtest/include/gtest/gtest.h" | |
| 10 | |
| 11 namespace content { | |
| 12 | |
| 13 TestResourceHandler::TestResourceHandler(net::URLRequestStatus* request_status, | |
| 14 std::string* body) | |
| 15 : ResourceHandler(nullptr), request_status_(request_status), body_(body) { | |
| 16 SetBufferSize(2048); | |
| 17 } | |
| 18 | |
| 19 TestResourceHandler::TestResourceHandler() | |
| 20 : TestResourceHandler(nullptr, nullptr) {} | |
| 21 | |
| 22 TestResourceHandler::~TestResourceHandler() {} | |
| 23 | |
| 24 void TestResourceHandler::SetController(ResourceController* controller) {} | |
| 25 | |
| 26 bool TestResourceHandler::OnRequestRedirected( | |
| 27 const net::RedirectInfo& redirect_info, | |
| 28 ResourceResponse* response, | |
| 29 bool* defer) { | |
| 30 NOTREACHED(); | |
|
Charlie Harrison
2016/10/26 14:05:19
Can you NOTREACHED() << "Redirects are not support
mmenke
2016/10/26 15:27:29
Added the text, kept NOTREACHED(). Experimenting
| |
| 31 return false; | |
| 32 } | |
| 33 | |
| 34 bool TestResourceHandler::OnResponseStarted(ResourceResponse* response, | |
| 35 bool* defer) { | |
| 36 EXPECT_EQ(1, on_will_start_called_); | |
| 37 EXPECT_EQ(0, on_response_started_called_); | |
| 38 EXPECT_EQ(0, on_response_completed_called_); | |
| 39 ++on_response_started_called_; | |
| 40 | |
| 41 if (!on_response_started_result_) | |
| 42 return false; | |
| 43 *defer = defer_on_response_started_; | |
| 44 defer_on_response_started_ = false; | |
| 45 return true; | |
| 46 } | |
| 47 | |
| 48 bool TestResourceHandler::OnWillStart(const GURL& url, bool* defer) { | |
|
Charlie Harrison
2016/10/26 14:05:19
include gurl.h
mmenke
2016/10/26 15:27:29
I don't think it's needed - we never dereference i
| |
| 49 EXPECT_EQ(0, on_response_started_called_); | |
| 50 EXPECT_EQ(0, on_will_start_called_); | |
| 51 EXPECT_EQ(0, on_response_completed_called_); | |
| 52 ++on_will_start_called_; | |
| 53 | |
| 54 if (!on_will_start_result_) | |
| 55 return false; | |
| 56 | |
| 57 *defer = defer_on_will_start_; | |
| 58 return true; | |
| 59 } | |
| 60 | |
| 61 bool TestResourceHandler::OnWillRead(scoped_refptr<net::IOBuffer>* buf, | |
| 62 int* buf_size, | |
| 63 int min_size) { | |
| 64 EXPECT_EQ(0, on_response_completed_called_); | |
| 65 ++on_will_read_called_; | |
| 66 | |
| 67 *buf = buffer_; | |
| 68 *buf_size = buffer_size_; | |
| 69 memset(buffer_->data(), '\0', buffer_size_); | |
| 70 return on_will_read_result_; | |
| 71 } | |
| 72 | |
| 73 bool TestResourceHandler::OnReadCompleted(int bytes_read, bool* defer) { | |
| 74 EXPECT_EQ(1, on_will_start_called_); | |
| 75 EXPECT_EQ(1, on_response_started_called_); | |
| 76 EXPECT_EQ(0, on_response_completed_called_); | |
| 77 ++on_read_completed_called_; | |
| 78 | |
| 79 EXPECT_LE(static_cast<size_t>(bytes_read), buffer_size_); | |
| 80 if (body_) | |
| 81 body_->append(buffer_->data(), bytes_read); | |
| 82 if (!on_read_completed_result_) | |
| 83 return false; | |
| 84 *defer = defer_on_read_completed_; | |
| 85 defer_on_read_completed_ = false; | |
| 86 return true; | |
| 87 } | |
| 88 | |
| 89 void TestResourceHandler::OnResponseCompleted( | |
| 90 const net::URLRequestStatus& status, | |
| 91 bool* defer) { | |
| 92 EXPECT_EQ(0, on_response_completed_called_); | |
| 93 ++on_response_completed_called_; | |
| 94 | |
| 95 if (request_status_) | |
| 96 *request_status_ = status; | |
| 97 *defer = defer_on_response_completed_; | |
| 98 defer_on_response_completed_ = false; | |
| 99 } | |
| 100 | |
| 101 void TestResourceHandler::OnDataDownloaded(int bytes_downloaded) { | |
| 102 NOTREACHED(); | |
|
Charlie Harrison
2016/10/26 14:05:19
ditto
mmenke
2016/10/26 15:27:29
Done.
| |
| 103 } | |
| 104 | |
| 105 void TestResourceHandler::SetBufferSize(int buffer_size) { | |
| 106 buffer_ = new net::IOBuffer(buffer_size); | |
| 107 buffer_size_ = buffer_size; | |
| 108 memset(buffer_->data(), '\0', buffer_size); | |
| 109 } | |
| 110 | |
| 111 } // namespace content | |
| OLD | NEW |