OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/safe_browsing/download_feedback.h" | 5 #include "chrome/browser/safe_browsing/download_feedback.h" |
6 | 6 |
7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
8 #include "base/file_util.h" | 8 #include "base/file_util.h" |
9 #include "base/files/scoped_temp_dir.h" | 9 #include "base/files/scoped_temp_dir.h" |
10 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
(...skipping 151 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
162 ClientDownloadResponse::DANGEROUS_HOST); | 162 ClientDownloadResponse::DANGEROUS_HOST); |
163 std::string ping_request( | 163 std::string ping_request( |
164 expected_report_metadata.download_request().SerializeAsString()); | 164 expected_report_metadata.download_request().SerializeAsString()); |
165 std::string ping_response( | 165 std::string ping_response( |
166 expected_report_metadata.download_response().SerializeAsString()); | 166 expected_report_metadata.download_response().SerializeAsString()); |
167 | 167 |
168 const char kTestFeedbackURL[] = "https://example.com/test/upload"; | 168 const char kTestFeedbackURL[] = "https://example.com/test/upload"; |
169 CommandLine::ForCurrentProcess()->AppendSwitchASCII( | 169 CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
170 switches::kSbDownloadFeedbackURL, kTestFeedbackURL); | 170 switches::kSbDownloadFeedbackURL, kTestFeedbackURL); |
171 | 171 |
172 DownloadFeedback* feedback = DownloadFeedback::Create( | 172 DownloadFeedback* feedback = |
173 url_request_context_getter_, | 173 DownloadFeedback::Create(url_request_context_getter_.get(), |
174 file_task_runner_, | 174 file_task_runner_.get(), |
175 upload_file_path_, | 175 upload_file_path_, |
176 ping_request, | 176 ping_request, |
177 ping_response); | 177 ping_response); |
178 EXPECT_FALSE(uploader()); | 178 EXPECT_FALSE(uploader()); |
179 | 179 |
180 feedback->Start(base::Bind(&DownloadFeedbackTest::FinishCallback, | 180 feedback->Start(base::Bind(&DownloadFeedbackTest::FinishCallback, |
181 base::Unretained(this), | 181 base::Unretained(this), |
182 feedback)); | 182 feedback)); |
183 ASSERT_TRUE(uploader()); | 183 ASSERT_TRUE(uploader()); |
184 EXPECT_FALSE(feedback_finish_called_); | 184 EXPECT_FALSE(feedback_finish_called_); |
185 EXPECT_TRUE(uploader()->start_called_); | 185 EXPECT_TRUE(uploader()->start_called_); |
186 | 186 |
187 EXPECT_EQ(url_request_context_getter_, | 187 EXPECT_EQ(url_request_context_getter_, |
(...skipping 21 matching lines...) Expand all Loading... |
209 upload_file_data_.size()); | 209 upload_file_data_.size()); |
210 expected_report_metadata.mutable_download_request()->mutable_digests( | 210 expected_report_metadata.mutable_download_request()->mutable_digests( |
211 )->set_sha1("hi"); | 211 )->set_sha1("hi"); |
212 expected_report_metadata.mutable_download_response()->set_verdict( | 212 expected_report_metadata.mutable_download_response()->set_verdict( |
213 ClientDownloadResponse::DANGEROUS_HOST); | 213 ClientDownloadResponse::DANGEROUS_HOST); |
214 std::string ping_request( | 214 std::string ping_request( |
215 expected_report_metadata.download_request().SerializeAsString()); | 215 expected_report_metadata.download_request().SerializeAsString()); |
216 std::string ping_response( | 216 std::string ping_response( |
217 expected_report_metadata.download_response().SerializeAsString()); | 217 expected_report_metadata.download_response().SerializeAsString()); |
218 | 218 |
219 DownloadFeedback* feedback = DownloadFeedback::Create( | 219 DownloadFeedback* feedback = |
220 url_request_context_getter_, | 220 DownloadFeedback::Create(url_request_context_getter_.get(), |
221 file_task_runner_, | 221 file_task_runner_.get(), |
222 upload_file_path_, | 222 upload_file_path_, |
223 ping_request, | 223 ping_request, |
224 ping_response); | 224 ping_response); |
225 EXPECT_FALSE(uploader()); | 225 EXPECT_FALSE(uploader()); |
226 | 226 |
227 feedback->Start(base::Bind(&DownloadFeedbackTest::FinishCallback, | 227 feedback->Start(base::Bind(&DownloadFeedbackTest::FinishCallback, |
228 base::Unretained(this), | 228 base::Unretained(this), |
229 feedback)); | 229 feedback)); |
230 ASSERT_TRUE(uploader()); | 230 ASSERT_TRUE(uploader()); |
231 EXPECT_FALSE(feedback_finish_called_); | 231 EXPECT_FALSE(feedback_finish_called_); |
232 EXPECT_TRUE(uploader()->start_called_); | 232 EXPECT_TRUE(uploader()->start_called_); |
233 EXPECT_TRUE(file_util::PathExists(upload_file_path_)); | 233 EXPECT_TRUE(file_util::PathExists(upload_file_path_)); |
234 | 234 |
235 delete feedback; | 235 delete feedback; |
236 EXPECT_FALSE(feedback_finish_called_); | 236 EXPECT_FALSE(feedback_finish_called_); |
237 | 237 |
238 base::RunLoop().RunUntilIdle(); | 238 base::RunLoop().RunUntilIdle(); |
239 EXPECT_FALSE(file_util::PathExists(upload_file_path_)); | 239 EXPECT_FALSE(file_util::PathExists(upload_file_path_)); |
240 } | 240 } |
241 | 241 |
242 } // namespace safe_browsing | 242 } // namespace safe_browsing |
OLD | NEW |