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_url_loader_client.h" | 5 #include "content/browser/loader/test_url_loader_client.h" |
6 | 6 |
7 #include "base/memory/ref_counted.h" | 7 #include "base/memory/ref_counted.h" |
8 #include "base/run_loop.h" | 8 #include "base/run_loop.h" |
9 #include "testing/gtest/include/gtest/gtest.h" | 9 #include "testing/gtest/include/gtest/gtest.h" |
10 | 10 |
11 namespace content { | 11 namespace content { |
12 | 12 |
13 TestURLLoaderClient::TestURLLoaderClient() : binding_(this) {} | 13 TestURLLoaderClient::TestURLLoaderClient() : binding_(this) {} |
14 TestURLLoaderClient::~TestURLLoaderClient() {} | 14 TestURLLoaderClient::~TestURLLoaderClient() {} |
15 | 15 |
16 void TestURLLoaderClient::OnReceiveResponse( | 16 void TestURLLoaderClient::OnReceiveResponse( |
17 const ResourceResponseHead& response_head, | 17 const ResourceResponseHead& response_head, |
18 mojom::DownloadedTempFilePtr downloaded_file) { | 18 mojom::DownloadedTempFilePtr downloaded_file) { |
| 19 EXPECT_FALSE(has_received_response_); |
| 20 EXPECT_FALSE(has_received_cached_metadata_); |
| 21 EXPECT_FALSE(has_received_completion_); |
19 has_received_response_ = true; | 22 has_received_response_ = true; |
20 response_head_ = response_head; | 23 response_head_ = response_head; |
21 if (quit_closure_for_on_receive_response_) | 24 if (quit_closure_for_on_receive_response_) |
22 quit_closure_for_on_receive_response_.Run(); | 25 quit_closure_for_on_receive_response_.Run(); |
23 } | 26 } |
24 | 27 |
25 void TestURLLoaderClient::OnReceiveRedirect( | 28 void TestURLLoaderClient::OnReceiveRedirect( |
26 const net::RedirectInfo& redirect_info, | 29 const net::RedirectInfo& redirect_info, |
27 const ResourceResponseHead& response_head) { | 30 const ResourceResponseHead& response_head) { |
| 31 EXPECT_FALSE(has_received_cached_metadata_); |
28 EXPECT_FALSE(response_body_.is_valid()); | 32 EXPECT_FALSE(response_body_.is_valid()); |
29 EXPECT_FALSE(has_received_response_); | 33 EXPECT_FALSE(has_received_response_); |
30 // Use ClearHasReceivedRedirect to accept more redirects. | 34 // Use ClearHasReceivedRedirect to accept more redirects. |
31 EXPECT_FALSE(has_received_redirect_); | 35 EXPECT_FALSE(has_received_redirect_); |
| 36 EXPECT_FALSE(has_received_completion_); |
32 has_received_redirect_ = true; | 37 has_received_redirect_ = true; |
33 redirect_info_ = redirect_info; | 38 redirect_info_ = redirect_info; |
34 response_head_ = response_head; | 39 response_head_ = response_head; |
35 if (quit_closure_for_on_receive_redirect_) | 40 if (quit_closure_for_on_receive_redirect_) |
36 quit_closure_for_on_receive_redirect_.Run(); | 41 quit_closure_for_on_receive_redirect_.Run(); |
37 } | 42 } |
38 | 43 |
39 void TestURLLoaderClient::OnDataDownloaded(int64_t data_length, | 44 void TestURLLoaderClient::OnDataDownloaded(int64_t data_length, |
40 int64_t encoded_data_length) { | 45 int64_t encoded_data_length) { |
| 46 EXPECT_TRUE(has_received_response_); |
| 47 EXPECT_FALSE(has_received_completion_); |
41 has_data_downloaded_ = true; | 48 has_data_downloaded_ = true; |
42 download_data_length_ += data_length; | 49 download_data_length_ += data_length; |
43 encoded_download_data_length_ += encoded_data_length; | 50 encoded_download_data_length_ += encoded_data_length; |
44 if (quit_closure_for_on_data_downloaded_) | 51 if (quit_closure_for_on_data_downloaded_) |
45 quit_closure_for_on_data_downloaded_.Run(); | 52 quit_closure_for_on_data_downloaded_.Run(); |
46 } | 53 } |
47 | 54 |
48 void TestURLLoaderClient::OnReceiveCachedMetadata( | 55 void TestURLLoaderClient::OnReceiveCachedMetadata( |
49 const std::vector<uint8_t>& data) { | 56 const std::vector<uint8_t>& data) { |
50 EXPECT_FALSE(has_received_cached_metadata_); | 57 EXPECT_FALSE(has_received_cached_metadata_); |
51 EXPECT_TRUE(has_received_response_); | 58 EXPECT_TRUE(has_received_response_); |
52 EXPECT_FALSE(has_received_completion_); | 59 EXPECT_FALSE(has_received_completion_); |
53 has_received_cached_metadata_ = true; | 60 has_received_cached_metadata_ = true; |
54 cached_metadata_ = | 61 cached_metadata_ = |
55 std::string(reinterpret_cast<const char*>(data.data()), data.size()); | 62 std::string(reinterpret_cast<const char*>(data.data()), data.size()); |
56 if (quit_closure_for_on_receive_cached_metadata_) | 63 if (quit_closure_for_on_receive_cached_metadata_) |
57 quit_closure_for_on_receive_cached_metadata_.Run(); | 64 quit_closure_for_on_receive_cached_metadata_.Run(); |
58 } | 65 } |
59 | 66 |
60 void TestURLLoaderClient::OnTransferSizeUpdated(int32_t transfer_size_diff) { | 67 void TestURLLoaderClient::OnTransferSizeUpdated(int32_t transfer_size_diff) { |
| 68 EXPECT_TRUE(has_received_response_); |
| 69 EXPECT_FALSE(has_received_completion_); |
61 EXPECT_GT(transfer_size_diff, 0); | 70 EXPECT_GT(transfer_size_diff, 0); |
62 body_transfer_size_ += transfer_size_diff; | 71 body_transfer_size_ += transfer_size_diff; |
63 } | 72 } |
64 | 73 |
65 void TestURLLoaderClient::OnStartLoadingResponseBody( | 74 void TestURLLoaderClient::OnStartLoadingResponseBody( |
66 mojo::ScopedDataPipeConsumerHandle body) { | 75 mojo::ScopedDataPipeConsumerHandle body) { |
| 76 EXPECT_TRUE(has_received_response_); |
| 77 EXPECT_FALSE(has_received_completion_); |
67 response_body_ = std::move(body); | 78 response_body_ = std::move(body); |
68 if (quit_closure_for_on_start_loading_response_body_) | 79 if (quit_closure_for_on_start_loading_response_body_) |
69 quit_closure_for_on_start_loading_response_body_.Run(); | 80 quit_closure_for_on_start_loading_response_body_.Run(); |
70 } | 81 } |
71 | 82 |
72 void TestURLLoaderClient::OnComplete( | 83 void TestURLLoaderClient::OnComplete( |
73 const ResourceRequestCompletionStatus& status) { | 84 const ResourceRequestCompletionStatus& status) { |
| 85 EXPECT_FALSE(has_received_completion_); |
74 has_received_completion_ = true; | 86 has_received_completion_ = true; |
75 completion_status_ = status; | 87 completion_status_ = status; |
76 if (quit_closure_for_on_complete_) | 88 if (quit_closure_for_on_complete_) |
77 quit_closure_for_on_complete_.Run(); | 89 quit_closure_for_on_complete_.Run(); |
78 } | 90 } |
79 | 91 |
80 void TestURLLoaderClient::ClearHasReceivedRedirect() { | 92 void TestURLLoaderClient::ClearHasReceivedRedirect() { |
81 has_received_redirect_ = false; | 93 has_received_redirect_ = false; |
82 } | 94 } |
83 | 95 |
(...skipping 58 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
142 void TestURLLoaderClient::RunUntilComplete() { | 154 void TestURLLoaderClient::RunUntilComplete() { |
143 if (has_received_completion_) | 155 if (has_received_completion_) |
144 return; | 156 return; |
145 base::RunLoop run_loop; | 157 base::RunLoop run_loop; |
146 quit_closure_for_on_complete_ = run_loop.QuitClosure(); | 158 quit_closure_for_on_complete_ = run_loop.QuitClosure(); |
147 run_loop.Run(); | 159 run_loop.Run(); |
148 quit_closure_for_on_complete_.Reset(); | 160 quit_closure_for_on_complete_.Reset(); |
149 } | 161 } |
150 | 162 |
151 } // namespace content | 163 } // namespace content |
OLD | NEW |