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 |